Fixes chained assignments

d0c-s4vage · Jan 9, 2020 · 1899314 · 1899314
1 parent 36a03d8
commit 1899314
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/pfp/interp.py b/pfp/interp.py
@@ -2070,6 +2070,7 @@ def assign_op(x, y):
             if node.op not in switch:
                 raise errors.UnsupportedAssignmentOperator(node.coord, node.op)
             switch[node.op](field, value)
+        return field
 
     def _handle_func_def(self, node, scope, ctxt, stream):
         """Handle FuncDef nodes

diff --git a/tests/test_control_flow.py b/tests/test_control_flow.py
@@ -322,6 +322,21 @@ def test_fall_through_no_case_body3(self):
             """,
         )
 
+    def test_chained_assignment(self):
+        dom = self._test_parse_build(
+            "",
+            """
+                local int b = 10;
+                local int c;
+                local int d;
+
+                c = d = b;
+            """,
+        )
+        self.assertEqual(dom.b is dom.c, False)
+        self.assertEqual(dom.b is dom.d, False)
+        self.assertEqual(dom.c is dom.d, False)
+
 
 if __name__ == "__main__":
     unittest.main()