Add Java return statement node

1 files changed, 50 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriterTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriterTests.java
index b6ad5d5..a19f5a2 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriterTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/output/lang/java/JavaWriterTests.java
@@ -243,6 +243,56 @@ public class JavaWriterTests {
     }
 
     @Test
+    public void methodWithBody() throws IOException {
+        var java = new JavaMethodDeclaration(
+            "a",
+            JavaTypeRef.INT,
+            List.of(),
+            new JavaBlock(List.of(
+                new JavaReturn(new JavaIntegerLiteral(42))
+            ))
+        );
+
+        var string = write(writer -> writer.writeMethodDeclaration(java));
+
+        assertThat(string, equalTo("""
+            public int a() {
+                return 42;
+            }"""
+        ));
+    }
+
+    @Test
+    public void block() throws IOException {
+        var java = new JavaBlock(List.of(
+            new JavaReturn(new JavaIntegerLiteral(42)),
+            new JavaReturn(new JavaIntegerLiteral(47))
+        ));
+
+        var string = write(writer -> writer.writeBlock(java));
+
+        assertThat(string, equalTo("return 42;\nreturn 47;"));
+    }
+
+    @Test
+    public void returnStatement() throws IOException {
+        var java = new JavaReturn(new JavaIntegerLiteral(42));
+
+        var string = write(writer -> writer.writeBlockStatement(java));
+
+        assertThat(string, equalTo("return 42;"));
+    }
+
+    @Test
+    public void integerLiteral() throws IOException {
+        var java = new JavaIntegerLiteral(42);
+
+        var string = write(writer -> writer.writeExpression(java));
+
+        assertThat(string, equalTo("42"));
+    }
+
+    @Test
     public void typeRefWithoutArgsHasNoAngleBrackets() throws IOException {
         var java = new JavaTypeRef(List.of("abc", "def"), "One", List.of());