Store enum variants in TypesInfo

2 files changed, 7 insertions, 1 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java
index f8a8fb4..ac85b7d 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEnumDefinitionTests.java
@@ -57,5 +57,11 @@ public class TypeCheckerEnumDefinitionTests {
                 )
             )
         ));
+        var typesInfo = globalContext.toTypesInfo();
+        var enumType = new EnumType(NamespaceName.of("a", "b"), "X");
+        assertThat(typesInfo.enumVariants(enumType), isSequence(
+            equalTo(new EnumVariant("a")),
+            equalTo(new EnumVariant("b"))
+        ));
     }
 }
diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java
index 60c2b9a..57c40f5 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerSumDefinitionTests.java
@@ -76,7 +76,7 @@ public class TypeCheckerSumDefinitionTests {
         var typesInfo = globalContext.toTypesInfo();
         assertThat(typesInfo.variantOf(rectangleType), isSequence(equalTo(sumType)));
         assertThat(typesInfo.variantOf(triangleType), isSequence(equalTo(sumType)));
-        assertThat(typesInfo.variants(sumType), isSequence(equalTo(rectangleType), equalTo(triangleType)));
+        assertThat(typesInfo.sumVariants(sumType), isSequence(equalTo(rectangleType), equalTo(triangleType)));
     }
 
     @Test