Read relation method from table member access

1 files changed, 18 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEvalExpressionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEvalExpressionTests.java
index 43c5bb3..cae195a 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEvalExpressionTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/typechecker/TypeCheckerEvalExpressionTests.java
@@ -45,6 +45,24 @@ public class TypeCheckerEvalExpressionTests {
         assertThat(typed, equalTo(table));
     }
 
+    @Test
+    public void memberAccessOnTableCanReadRelationMethod() {
+        var untyped = UntypedMemberAccessNode.arbitrary()
+            .withReceiver(UntypedReferenceNode.arbitrary().withName("authors"))
+            .withMemberName("where")
+            .build();
+        var globalContext = TypeCheckerContextArb.globalContext();
+        var namespaceName = NamespaceName.of("App");
+        var namespaceContext = globalContext.enterNamespace(namespaceName);
+        var databaseSchema = new DatabaseSchema(namespaceName, "blogApp");
+        var table = new Table(databaseSchema, "authors");
+        namespaceContext.declare("authors", Types.table(table), NullSource.INSTANCE);
+
+        var typed = TypeChecker.evalExpression(untyped, namespaceContext);
+
+        assertThat(typed, equalTo(new RelationMethod(table, "where")));
+    }
+
     // == Number literal ==
 
     @Test