diff options
Diffstat (limited to 'src/test/java')
| -rw-r--r-- | src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java | 11 | ||||
| -rw-r--r-- | src/test/java/org/shedlang/compiler/parser/ParserAssignTests.java | 21 |
2 files changed, 32 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 0f9bc3a2..564d7609 100644 --- a/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java +++ b/src/test/java/org/shedlang/compiler/ast/untyped/UntypedNodeMatchers.java @@ -17,6 +17,17 @@ public class UntypedNodeMatchers { return isUntypedBinaryOperationNode(UntypedBinaryOperator.ADD, left, right); } + public static Matcher<UntypedNode> isUntypedAssignNode( + Matcher<UntypedTargetNode> target, + Matcher<UntypedExpressionNode> expression + ) { + return instanceOf( + UntypedAssignNode.class, + has("target", x -> x.target(), target), + has("expression", x -> x.expression(), expression) + ); + } + public static UntypedArgsNodeMatcher isUntypedArgsNode() { return UntypedArgsNodeMatcher.empty(); } diff --git a/src/test/java/org/shedlang/compiler/parser/ParserAssignTests.java b/src/test/java/org/shedlang/compiler/parser/ParserAssignTests.java new file mode 100644 index 00000000..ce8c68d3 --- /dev/null +++ b/src/test/java/org/shedlang/compiler/parser/ParserAssignTests.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 ParserAssignTests { + @Test + public void canParseAssignAsFunctionStatement() { + var source = "assign x := y;"; + + var node = parseString(source, Parser::parseFunctionStatement); + + assertThat(node, isUntypedAssignNode( + isUntypedTargetVariableNode("x"), + isUntypedReferenceNode("y") + )); + } +} |
