2 files changed, 8 insertions, 1 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java b/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
index a9be381..ce8ab87 100644
--- a/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/ValueObjectTests.java
@@ -55,6 +55,7 @@ public class ValueObjectTests {
             {LocalVariableDeclarationNode.class},
             {MethodCallNode.class},
             {MethodNode.class},
+            {OperationNode.class},
             {ReturnNode.class},
             {StaticReceiver.class},
             {TernaryConditionalNode.class},
@@ -135,6 +136,8 @@ public class ValueObjectTests {
             return list("[list 1]");
         } else if (type.equals(Set.class)) {
             return Collections.singleton("[set 1]");
+        } else if (type.equals(Operator.class)) {
+            return Operator.DIVIDE;
         } else if (type.equals(Identifier.class)) {
             return TestIds.id((String)generateFirstInstance(String.class));
         } else if (type.equals(PrimitiveValue.class)) {
@@ -159,6 +162,8 @@ public class ValueObjectTests {
             return list("[list 2]");
         } else if (type.equals(Set.class)) {
             return Collections.singleton("[set 2]");
+        } else if (type.equals(Operator.class)) {
+            return Operator.MULTIPLY;
         } else if (type.equals(Identifier.class)) {
             return TestIds.id((String)generateSecondInstance(String.class));
         } else if (type.equals(PrimitiveValue.class)) {
@@ -183,6 +188,8 @@ public class ValueObjectTests {
             return list("[list]");
         } else if (type.equals(Set.class)) {
             return Collections.singleton("[set]");
+        } else if (type.equals(Operator.class)) {
+            return Operator.ADD;
         } else if (type.equals(Identifier.class)) {
             return TestIds.id((String)generateInstance(String.class));
         } else if (type.equals(PrimitiveValue.class)) {
diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
index cb1f3ac..6bc0816 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
@@ -470,7 +470,7 @@ public class JavaReaderTests {
         assertEquals(
             assign(
                 reference(argument),
-                integerOperation(operator.getMethodName(), reference(argument), literal(2))),
+                integerOperation(operator, reference(argument), literal(2))),
             statement.getExpression());
     }