Add if-statement node

1 files changed, 12 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/BackendMethodTests.java b/src/test/java/org/zwobble/couscous/tests/BackendMethodTests.java
index f1af3a9..f77140b 100644
--- a/src/test/java/org/zwobble/couscous/tests/BackendMethodTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/BackendMethodTests.java
@@ -9,11 +9,13 @@ import org.zwobble.couscous.ast.MethodNode;
 import org.zwobble.couscous.values.PrimitiveValue;
 import org.zwobble.couscous.values.PrimitiveValues;
 import org.zwobble.couscous.values.StringValue;
+
 import static java.util.Arrays.asList;
 import static org.junit.Assert.assertEquals;
 import static org.zwobble.couscous.ast.AssignmentNode.assign;
 import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement;
 import static org.zwobble.couscous.ast.FormalArgumentNode.formalArg;
+import static org.zwobble.couscous.ast.IfStatementNode.ifStatement;
 import static org.zwobble.couscous.ast.LiteralNode.literal;
 import static org.zwobble.couscous.ast.LocalVariableDeclarationNode.localVariableDeclaration;
 import static org.zwobble.couscous.ast.MethodNode.staticMethod;
@@ -85,6 +87,16 @@ public abstract class BackendMethodTests {
         assertEquals(value("[updated value]"), result);
     }
     
+    @Test
+    public void whenConditionIsTrueIfStatementExecutesTrueBranch() {
+        MethodNode.Builder method = staticMethod("hello")
+            .statement(ifStatement(
+                literal(true),
+                asList(returns(literal("[true]"))),
+                asList(returns(literal("[false]")))));
+        assertEquals(value("[true]"), runMethod(method));
+    }
+    
     protected PrimitiveValue runMethod(MethodNode.Builder methodBuilder, PrimitiveValue... arguments) {
         MethodNode method = methodBuilder.build();
         String className = "com.example.Program";