Check row type is struct

1 files changed, 26 insertions, 1 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerTableDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerTableDefinitionTests.java
index 7c08688..a4e1cf4 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerTableDefinitionTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerTableDefinitionTests.java
@@ -4,10 +4,13 @@ import org.junit.jupiter.api.Test;
 import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedTableDefinitionNode;
 import org.zwobble.hobgoblin.compiler.ast.untyped.UntypedTypeLevelReferenceNode;
 import org.zwobble.hobgoblin.compiler.sources.NullSource;
+import org.zwobble.hobgoblin.compiler.typechecker.errors.TableRowTypeMustBeStructError;
 import org.zwobble.hobgoblin.compiler.types.NamespaceName;
+import org.zwobble.hobgoblin.compiler.types.SimpleNativeType;
 import org.zwobble.hobgoblin.compiler.types.SimpleStructType;
 import org.zwobble.hobgoblin.compiler.types.Types;
 
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.zwobble.hobgoblin.compiler.ast.typed.TypedTableDefinitionNodeMatcher.isTypedTableDefinitionNode;
 import static org.zwobble.hobgoblin.compiler.typechecker.TypeChecker.typeCheckDatabaseSchemaStatement;
 import static org.zwobble.precisely.AssertThat.assertThat;
@@ -15,7 +18,7 @@ import static org.zwobble.precisely.Matchers.equalTo;
 
 public class TypeCheckerTableDefinitionTests {
     @Test
-    public void databaseSchemaStatementsAreTypeChecked() {
+    public void minimalTableDefinition() {
         var untyped = UntypedTableDefinitionNode.arbitrary()
             .withName("authors")
             .withRowType(UntypedTypeLevelReferenceNode.arbitrary()
@@ -35,4 +38,26 @@ public class TypeCheckerTableDefinitionTests {
             .withRowType(equalTo(authorType))
         );
     }
+
+    @Test
+    public void whenRowTypeIsNotStructTypeThenErrorIsThrown() {
+        var untyped = UntypedTableDefinitionNode.arbitrary()
+            .withName("authors")
+            .withRowType(UntypedTypeLevelReferenceNode.arbitrary()
+                .withName("Author")
+                .build()
+            )
+            .build();
+        var globalContext = TypeCheckerContextArb.globalContext();
+        var namespaceContext = globalContext.enterNamespace(NamespaceName.of("App"));
+        var authorType = new SimpleNativeType(NamespaceName.of("App"), "Author");
+        namespaceContext.declare("Author", Types.metaType(authorType), NullSource.INSTANCE);
+
+        var error = assertThrows(
+            TableRowTypeMustBeStructError.class,
+            () -> typeCheckDatabaseSchemaStatement(untyped, namespaceContext)
+        );
+
+        assertThat(error.rowType(), equalTo(authorType));
+    }
 }