1 files changed, 27 insertions, 2 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 da0f26f..0f2313a 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
@@ -79,8 +79,8 @@ public class JavaWriterTests {
         var java = new JavaInterfaceDeclaration(
             "Shape",
             List.of(
-                new JavaTypeRef(List.of("abc", "def"), "One"),
-                new JavaTypeRef(List.of("abc", "def"), "Two")
+                new JavaTypeRef(List.of("abc", "def"), "One", List.of()),
+                new JavaTypeRef(List.of("abc", "def"), "Two", List.of())
             )
         );
 
@@ -92,6 +92,31 @@ public class JavaWriterTests {
         ));
     }
 
+    @Test
+    public void typeRefWithoutArgsHasNoAngleBrackets() throws IOException {
+        var java = new JavaTypeRef(List.of("abc", "def"), "One", List.of());
+
+        var string = write(writer -> writer.writeTypeRef(java));
+
+        assertThat(string, equalTo("abc.def.One"));
+    }
+
+    @Test
+    public void typeRefWithArgsHasAngleBrackets() throws IOException {
+        var java = new JavaTypeRef(
+            List.of("abc", "def"),
+            "One",
+            List.of(
+                new JavaTypeRef(List.of(), "Two", List.of()),
+                new JavaTypeRef(List.of(), "Three", List.of())
+            )
+        );
+
+        var string = write(writer -> writer.writeTypeRef(java));
+
+        assertThat(string, equalTo("abc.def.One<Two, Three>"));
+    }
+
     private String write(Write write) throws IOException {
         var stringWriter = new StringWriter();
         var javaWriter = new JavaWriter(new CodeWriter(stringWriter));