Fixed it so both origin and full symbol are represented as usages

Author: zomglings

locust/parse.py

@@ -210,6 +210,15 @@ def visit_Attribute(self, node: ast.Attribute) -> None:
             qualifications = symbols.get(cumulative_component)
             if qualifications:
                 for qualification in qualifications:
+                    origin_definition = RawDefinition(
+                        name=qualification,
+                        change_type=self.context_type.value,
+                        line=node.lineno,
+                        offset=node.col_offset,
+                        end_line=node.end_lineno,
+                        end_offset=node.end_col_offset,
+                        parent=parent,
+                    )
                     # Replace prefix with qualification in final_component
                     name = final_component.replace(
                         cumulative_component, qualification, 1
@@ -223,7 +232,8 @@ def visit_Attribute(self, node: ast.Attribute) -> None:
                         end_offset=node.end_col_offset,
                         parent=parent,
                     )
-                    self.definitions.append(definition)
+
+                    self.definitions.extend([origin_definition ,definition])
 
     def reset(self):
         self.scope = []

tests/fixtures/test_parse.json

@@ -126,6 +126,16 @@
       "total_lines": 2,
       "parent": {}
     },
+    {
+      "name": "argparse",
+      "change_type": "usage",
+      "filepath": "sample.py",
+      "revision": "0838585",
+      "line": 7,
+      "changed_lines": 1,
+      "total_lines": 1,
+      "parent": {}
+    },
     {
       "name": "argparse.ArgumentParser",
       "change_type": "usage",

tests/fixtures/test_parse_dependencies.json

@@ -311,6 +311,16 @@
       "total_lines": 1,
       "parent": {}
     },
+    {
+      "name": "json",
+      "change_type": "usage",
+      "filepath": "mod/submod/print.py",
+      "revision": "04cec01",
+      "line": 5,
+      "changed_lines": 1,
+      "total_lines": 1,
+      "parent": {}
+    },
     {
       "name": "json.dumps",
       "change_type": "usage",

tests/fixtures/test_render.json

@@ -19,6 +19,14 @@
           "total_lines": 2,
           "children": []
         },
+        {
+          "name": "argparse",
+          "type": "usage",
+          "line": 7,
+          "changed_lines": 1,
+          "total_lines": 1,
+          "children": []
+        },
         {
           "name": "argparse.ArgumentParser",
           "type": "usage",