Parse equality

2 files changed, 28 insertions, 0 deletions

diff --git a/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java b/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java
index ff249a02..bad7b9f7 100644
--- a/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java
+++ b/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java
@@ -38,6 +38,13 @@ public class UntypedNodeMatchers {
         return UntypedCallNodeMatcher.empty();
     }
 
+    public static Matcher<UntypedExpressionNode> isUntypedEqualsNode(
+        Matcher<UntypedExpressionNode> left,
+        Matcher<UntypedExpressionNode> right
+    ) {
+        return isUntypedBinaryOperationNode(UntypedBinaryOperator.EQUALS, left, right);
+    }
+
     public static Matcher<UntypedFunctionStatementNode> isUntypedExpressionStatementNode(
         Matcher<UntypedExpressionNode> expression
     ) {
diff --git a/src/test/java/org/shedlang/compiler/parser/ParserEqualsTests.java b/src/test/java/org/shedlang/compiler/parser/ParserEqualsTests.java
new file mode 100644
index 00000000..e2615d95
--- /dev/null
+++ b/src/test/java/org/shedlang/compiler/parser/ParserEqualsTests.java
@@ -0,0 +1,21 @@
+package org.shedlang.compiler.parser;
+
+import org.junit.jupiter.api.Test;
+
+import static org.shedlang.compiler.ast.untyped.UntypedNodeMatchers.*;
+import static org.shedlang.compiler.parser.ParserTesting.parseString;
+import static org.zwobble.precisely.AssertThat.assertThat;
+
+public class ParserEqualsTests {
+    @Test
+    public void canParseEqualityOfTwoPrimaryExpressions() {
+        var source = "1 == 2";
+
+        var node = parseString(source, Parser::parseTopLevelExpression);
+
+        assertThat(node, isUntypedEqualsNode(
+            isUntypedIntLiteralNode(1),
+            isUntypedIntLiteralNode(2)
+        ));
+    }
+}