Add assignment expr in exprStatement (#359)

summer-ji-eng · web-flow · commit a3f3c5bc9ec1 · 2020-10-02T03:25:19.000-07:00
diff --git a/src/main/java/com/google/api/generator/engine/ast/ExprStatement.java b/src/main/java/com/google/api/generator/engine/ast/ExprStatement.java
@@ -52,6 +52,7 @@ public ExprStatement build() {
             (expr instanceof MethodInvocationExpr)
                 || (expr instanceof ReferenceConstructorExpr)
                 || (expr instanceof AssignmentExpr)
+                || (expr instanceof AssignmentOperationExpr)
                 || (expr instanceof ThrowExpr)
                 || (expr instanceof ReturnExpr)
                 || (expr instanceof UnaryOperationExpr
diff --git a/src/test/java/com/google/api/generator/engine/ast/ExprStatementTest.java b/src/test/java/com/google/api/generator/engine/ast/ExprStatementTest.java
@@ -79,6 +79,15 @@ public void validExprStatement_unaryOperation() {
                 Variable.builder().setType(TypeNode.INT).setName("i").build())));
   }
 
+  @Test
+  public void validExprStatement_assignmentOperationExpr() {
+    VariableExpr lhsExpr =
+        VariableExpr.withVariable(Variable.builder().setName("i").setType(TypeNode.INT).build());
+    ValueExpr rhsExpr =
+        ValueExpr.withValue(PrimitiveValue.builder().setValue("5").setType(TypeNode.INT).build());
+    assertValidExprStatement(AssignmentOperationExpr.multiplyAssignmentWithExprs(lhsExpr, rhsExpr));
+  }
+
   @Test
   public void invalidExprStatement_variable() {
     Variable variable = Variable.builder().setType(TypeNode.INT).setName("libraryClient").build();
