Move declare and set test into generic test suite

1 files changed, 13 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/BackendTests.java b/src/test/java/org/zwobble/couscous/tests/BackendTests.java
index f9566d7..dc13469 100644
--- a/src/test/java/org/zwobble/couscous/tests/BackendTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/BackendTests.java
@@ -74,6 +74,19 @@ public abstract class BackendTests {
         
         assertEquals(value("[initial value]"), result);
     }
+    
+    @Test
+    public void canDeclareVariableAndThenAssignValues() {
+        val localVariableDeclaration = localVariableDeclaration(
+            42, "x", StringValue.REF, LiteralNode.literal("[initial value]"));
+        val method = staticMethod("hello")
+            .statement(localVariableDeclaration)
+            .statement(new ExpressionStatementNode(new AssignmentNode(reference(localVariableDeclaration), LiteralNode.literal("[updated value]"))))
+            .statement(new ReturnNode(reference(localVariableDeclaration)));
+        val result = runMethod(method);
+        
+        assertEquals(value("[updated value]"), result);
+    }
 
     protected PrimitiveValue runMethod(MethodNode.MethodNodeBuilder methodBuilder, PrimitiveValue... arguments) {
         val method = methodBuilder.build();