Support array types in JavaTypeRef

1 files changed, 24 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 9f6214d..79e8aa9 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
@@ -818,6 +818,30 @@ public class JavaWriterTests {
         assertThat(string, equalTo("abc.def.One.Two"));
     }
 
+    @Test
+    public void typeRefForArrayUsesSquareBrackets() throws IOException {
+        var java = JavaTypeRef.array(JavaTypeRef.topLevel(
+            JavaPackageName.of("abc", "def"),
+            JavaIdentifier.of("One")
+        ));
+
+        var string = write(writer -> writer.writeTypeRef(java));
+
+        assertThat(string, equalTo("abc.def.One[]"));
+    }
+
+    @Test
+    public void typeRefForMultiDimensionalArrayUsesMultipleSquareBrackets() throws IOException {
+        var java = JavaTypeRef.array(JavaTypeRef.array(JavaTypeRef.array(JavaTypeRef.topLevel(
+            JavaPackageName.of("abc", "def"),
+            JavaIdentifier.of("One")
+        ))));
+
+        var string = write(writer -> writer.writeTypeRef(java));
+
+        assertThat(string, equalTo("abc.def.One[][][]"));
+    }
+
     // == Visibility ==
 
     // == Statements ==