Parse namespace body

1 files changed, 26 insertions, 1 deletions

diff --git a/src/test/java/org/shedlang/compiler/parser/ParserNamespaceTests.java b/src/test/java/org/shedlang/compiler/parser/ParserNamespaceTests.java
index 87ebb7e8..15666dc7 100644
--- a/src/test/java/org/shedlang/compiler/parser/ParserNamespaceTests.java
+++ b/src/test/java/org/shedlang/compiler/parser/ParserNamespaceTests.java
@@ -3,9 +3,10 @@ package org.shedlang.compiler.parser;
 import org.junit.jupiter.api.Test;
 import org.shedlang.compiler.ast.UntypedNamespaceNode;
 
+import static org.shedlang.compiler.ast.UntypedNodeMatchers.isUntypedFunctionDefinitionNode;
 import static org.shedlang.compiler.parser.ParserTesting.parseString;
 import static org.zwobble.precisely.AssertThat.assertThat;
-import static org.zwobble.precisely.Matchers.instanceOf;
+import static org.zwobble.precisely.Matchers.*;
 
 public class ParserNamespaceTests {
     @Test
@@ -19,4 +20,28 @@ public class ParserNamespaceTests {
 
         assertThat(node, instanceOf(UntypedNamespaceNode.class));
     }
+
+    @Test
+    public void canParseNamespaceWithStatements() {
+        var source = """
+            fun f() -> Unit {
+            }
+            
+            fun g() -> Unit {
+            }
+            """;
+
+        var node = parseString(
+            source,
+            Parser::parseNamespace
+        );
+
+        assertThat(node, instanceOf(
+            UntypedNamespaceNode.class,
+            has("body", x -> x.body(), isSequence(
+                isUntypedFunctionDefinitionNode().withName("f"),
+                isUntypedFunctionDefinitionNode().withName("g")
+            ))
+        ));
+    }
 }