Add equality operators to Java AST

1 files changed, 26 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 b888375..1ce78ea 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
@@ -881,6 +881,32 @@ public class JavaWriterTests {
     }
 
     @Test
+    public void binaryOperationEqualTo() throws IOException {
+        var java = new JavaBinaryOperation(
+            JavaBinaryOperator.EQUAL_TO,
+            new JavaRef(JavaIdentifier.of("x")),
+            new JavaRef(JavaIdentifier.of("y"))
+        );
+
+        var string = write(writer -> writer.writeTopLevelExpression(java));
+
+        assertThat(string, equalTo("x == y"));
+    }
+
+    @Test
+    public void binaryOperationNotEqualTo() throws IOException {
+        var java = new JavaBinaryOperation(
+            JavaBinaryOperator.NOT_EQUAL_TO,
+            new JavaRef(JavaIdentifier.of("x")),
+            new JavaRef(JavaIdentifier.of("y"))
+        );
+
+        var string = write(writer -> writer.writeTopLevelExpression(java));
+
+        assertThat(string, equalTo("x != y"));
+    }
+
+    @Test
     public void binaryOperationUnsignedLeftShift() throws IOException {
         var java = JavaBinaryOperation.arbitrary()
             .withOperator(JavaBinaryOperator.LEFT_SHIFT)