Change TypedSumTypeDefinitionNode to have typeOrConstructor()

2 files changed, 8 insertions, 2 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/typed/Typed.java b/src/test/java/org/shedlang/compiler/ast/typed/Typed.java
index 58b66557..6ef8df43 100644
--- a/src/test/java/org/shedlang/compiler/ast/typed/Typed.java
+++ b/src/test/java/org/shedlang/compiler/ast/typed/Typed.java
@@ -226,7 +226,10 @@ public class Typed {
     }
 
     public static TypedSumTypeDefinitionNode sumTypeDefinition(SumType sumType) {
-        return new TypedSumTypeDefinitionNode(sumType, Sources.NULL);
+        return new TypedSumTypeDefinitionNode(
+            TypeOrConstructor.type(sumType),
+            Sources.NULL
+        );
     }
 
     public static TypedTargetNode targetIgnore() {
diff --git a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
index 65f5f24b..73aa5c2f 100644
--- a/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
+++ b/src/test/java/org/shedlang/compiler/ast/typed/TypedNodeMatchers.java
@@ -306,7 +306,10 @@ public class TypedNodeMatchers {
     ) {
         return instanceOf(
             TypedSumTypeDefinitionNode.class,
-            has("type", x -> x.type(), equalTo(type))
+            has("typeOrConstructor", x -> x.typeOrConstructor(), instanceOf(
+                TypeOrConstructor.Type.class,
+                has("value", x -> x.value(), equalTo(type))
+            ))
         );
     }