Parse doc comment for sum definitions

1 files changed, 18 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/parser/ParserSumDefinitionTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/parser/ParserSumDefinitionTests.java
index 825511e..c512c6f 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/parser/ParserSumDefinitionTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/parser/ParserSumDefinitionTests.java
@@ -79,4 +79,22 @@ public class ParserSumDefinitionTests {
             ))
         ));
     }
+
+    @Test
+    public void sumWithDocComment() {
+        var source = """
+            /// A 2D shape.
+            sum Shape {
+            }""";
+
+        var node = parseString(
+            source,
+            Parser::parseNamespaceStatement
+        );
+
+        assertThat(node, instanceOf(
+            UntypedSumDefinitionNode.class,
+            has("docComment", UntypedSumDefinitionNode::docComment, equalTo("A 2D shape.\n"))
+        ));
+    }
 }