|
1 | 1 | /******************************************************************************* |
2 | 2 | * SysML 2 Pilot Implementation |
3 | | - * Copyright (c) 2021-2022 Model Driven Solutions, Inc. |
| 3 | + * Copyright (c) 2021-2022, 2025 Model Driven Solutions, Inc. |
4 | 4 | * |
5 | 5 | * This program is free software: you can redistribute it and/or modify |
6 | 6 | * it under the terms of the GNU Lesser General Public License as published by |
|
36 | 36 | import org.omg.sysml.lang.sysml.AttributeUsage; |
37 | 37 | import org.omg.sysml.lang.sysml.Definition; |
38 | 38 | import org.omg.sysml.lang.sysml.Element; |
| 39 | +import org.omg.sysml.lang.sysml.EnumerationDefinition; |
| 40 | +import org.omg.sysml.lang.sysml.EnumerationUsage; |
39 | 41 | import org.omg.sysml.lang.sysml.Expression; |
40 | 42 | import org.omg.sysml.lang.sysml.Feature; |
41 | 43 | import org.omg.sysml.lang.sysml.ItemUsage; |
@@ -221,4 +223,35 @@ public void testDirectedUsage() throws Exception { |
221 | 223 | assertEquals("constraint def (directedUsages)", directedFeatures.subList(0, 2), directedUsages); |
222 | 224 | } |
223 | 225 |
|
| 226 | + public final String enumeratedValueTest = |
| 227 | + "package Test {\n" |
| 228 | + + " variation item def I {\n" |
| 229 | + + " variant item i1;\n" |
| 230 | + + " variant item i2;\n" |
| 231 | + + " }" |
| 232 | + + " enum def A {\n" |
| 233 | + + " enum a1;\n" |
| 234 | + + " enum a2;\n" |
| 235 | + + " }" |
| 236 | + + "}"; |
| 237 | + |
| 238 | + @Test |
| 239 | + public void testEnumeratedValue() throws Exception { |
| 240 | + SysMLInteractive instance = getSysMLInteractiveInstance(); |
| 241 | + SysMLInteractiveResult result = instance.process(enumeratedValueTest); |
| 242 | + Element root = result.getRootElement(); |
| 243 | + List<Element> elements = ((Namespace)root).getOwnedMember(); |
| 244 | + List<Element> ownedMembers = ((Namespace)elements.get(0)).getOwnedMember(); |
| 245 | + List<Usage> variants = ((Definition)ownedMembers.get(0)).getVariant(); |
| 246 | + assertEquals("item def (variants.size)", 2, variants.size()); |
| 247 | + |
| 248 | + // Check that getting variant and enumeratedValue work for EnumerationDefinitions, |
| 249 | + // even though EnumerationDefinition::enumeratedValue redefines Definition::variant. |
| 250 | + |
| 251 | + variants = ((Definition)ownedMembers.get(1)).getVariant(); |
| 252 | + List<EnumerationUsage> enumeratedValues = ((EnumerationDefinition)ownedMembers.get(1)).getEnumeratedValue(); |
| 253 | + assertEquals("enum def (variants.size)", 2, variants.size()); |
| 254 | + assertEquals("enuim def (enumeratedValues.size)", 2, enumeratedValues.size()); |
| 255 | + assertEquals("enum def (enumeratedValues)", variants, enumeratedValues); |
| 256 | + } |
224 | 257 | } |
0 commit comments