From f8bcf996119e07114b191b54bc01444bd602804a Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 8 Jul 2023 21:22:43 +0100
Subject: Parse equality

---
 .../compiler/ast/untyped/UntypedNodeMatchers.java   |  7 +++++++
 .../shedlang/compiler/parser/ParserEqualsTests.java | 21 +++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 src/test/java/org/shedlang/compiler/parser/ParserEqualsTests.java

(limited to 'src/test/java')

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)
+        ));
+    }
+}
-- 
cgit v1.2.3