Merge pull request #42 from cyberthirst/fix/varinfo-compare

fix test for rw analysis for AnAssign nodes
charles-cooper · Aug 7, 2024 · 0bd0709 · 0bd0709
2 parents 562183a + d31ab50
commit 0bd0709
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/tests/unit/ast/test_ast_dict.py b/tests/unit/ast/test_ast_dict.py
@@ -1255,7 +1255,13 @@ def qux2():
         {
             "annotation": {"ast_type": "Name", "id": "uint256"},
             "ast_type": "AnnAssign",
-            "target": {"ast_type": "Name", "id": "x"},
+            "target": {
+                "ast_type": "Name",
+                "id": "x",
+                "variable_reads": [
+                    {"name": "x", "decl_node": {"node_id": 15, "source_id": 0}, "access_path": []}
+                ],
+            },
             "value": {
                 "ast_type": "Attribute",
                 "attr": "counter",
@@ -1300,7 +1306,13 @@ def qux2():
         {
             "annotation": {"ast_type": "Name", "id": "uint256"},
             "ast_type": "AnnAssign",
-            "target": {"ast_type": "Name", "id": "x"},
+            "target": {
+                "ast_type": "Name",
+                "id": "x",
+                "variable_reads": [
+                    {"name": "x", "decl_node": {"node_id": 35, "source_id": 0}, "access_path": []}
+                ],
+            },
             "value": {
                 "ast_type": "Attribute",
                 "attr": "counter",
@@ -1317,7 +1329,13 @@ def qux2():
         {
             "annotation": {"ast_type": "Name", "id": "uint256"},
             "ast_type": "AnnAssign",
-            "target": {"ast_type": "Name", "id": "y"},
+            "target": {
+                "ast_type": "Name",
+                "id": "y",
+                "variable_reads": [
+                    {"name": "y", "decl_node": {"node_id": 44, "source_id": 0}, "access_path": []}
+                ],
+            },
             "value": {
                 "ast_type": "Attribute",
                 "attr": "counter",