mirror of
https://github.com/safishamsi/graphify.git
synced 2026-10-13 23:49:14 +00:00
Mpasha17
75e9f102bf
Keep v8's sourceless stub and resolve after the walk: rewrite to the local
id only when that id is a class defined in this file. An imported name (e.g.
pkg/widget.py doing `from .base import Widget`) or a same-id function like
`def widget()` now falls through to the cross-file rewire exactly as on v8.
Co-Authored-By: Grok Bot
(cherry picked from commit 1511543938)
81 lines
3.0 KiB
Python
81 lines
3.0 KiB
Python
"""#4334: a forward-referenced class named after its file must not swallow the file node."""
|
|
from pathlib import Path
|
|
|
|
from graphify.extract import extract
|
|
|
|
WIDGET = """from __future__ import annotations
|
|
|
|
class Helper:
|
|
def make(self, w: Widget) -> None:
|
|
w.draw()
|
|
|
|
class Widget:
|
|
def draw(self):
|
|
return 1
|
|
"""
|
|
|
|
APP = """from widget import Widget
|
|
|
|
def use(w: Widget):
|
|
w.draw()
|
|
"""
|
|
|
|
|
|
def _extract(tmp_path: Path, files: dict[str, str]):
|
|
paths = []
|
|
for name, src in files.items():
|
|
p = tmp_path / name
|
|
p.write_text(src)
|
|
paths.append(p)
|
|
result = extract(paths, cache_root=tmp_path, root=tmp_path, parallel=False)
|
|
labels = {n["id"]: n["label"] for n in result["nodes"]}
|
|
edges = {(labels.get(e["source"]), e["relation"], labels.get(e["target"])) for e in result["edges"]}
|
|
return result, labels, edges
|
|
|
|
|
|
def _check(result, labels, edges):
|
|
file_nodes = [n for n in result["nodes"] if n["label"] == "widget.py"]
|
|
assert len(file_nodes) == 1, labels
|
|
assert ("widget.py", "contains", "Widget") in edges
|
|
assert ("widget.py", "contains", "Helper") in edges
|
|
assert ("Widget", "contains", "Widget") not in edges
|
|
assert ("Widget", "contains", "Helper") not in edges
|
|
assert (".make()", "calls", ".draw()") in edges
|
|
|
|
|
|
def test_forward_ref_keeps_file_node_single_file(tmp_path):
|
|
_check(*_extract(tmp_path, {"widget.py": WIDGET}))
|
|
|
|
|
|
def test_forward_ref_keeps_file_node_with_importer(tmp_path):
|
|
result, labels, edges = _extract(tmp_path, {"widget.py": WIDGET, "app.py": APP})
|
|
_check(result, labels, edges)
|
|
assert ("app.py", "imports_from", "widget.py") in edges
|
|
assert ("app.py", "imports_from", "Widget") not in edges
|
|
|
|
|
|
def test_imported_same_name_in_subdirectory_keeps_cross_file_rewire(tmp_path):
|
|
pkg = tmp_path / "pkg"
|
|
pkg.mkdir()
|
|
result, labels, edges = _extract(tmp_path, {
|
|
"pkg/__init__.py": "",
|
|
"pkg/base.py": "class Widget:\n def draw(self):\n return 1\n",
|
|
"pkg/widget.py": "from .base import Widget\n\ndef use(w: Widget):\n w.draw()\n",
|
|
})
|
|
ids = {n["id"] for n in result["nodes"]}
|
|
assert all(e["source"] in ids and e["target"] in ids for e in result["edges"])
|
|
base_widget = next(n["id"] for n in result["nodes"]
|
|
if n["label"] == "Widget" and n["source_file"].endswith("base.py"))
|
|
use = next(n["id"] for n in result["nodes"] if n["label"] == "use()")
|
|
assert any(e["source"] == use and e["target"] == base_widget for e in result["edges"])
|
|
assert [n for n in result["nodes"] if n["label"] == "Widget"] == [
|
|
n for n in result["nodes"] if n["id"] == base_widget
|
|
]
|
|
|
|
|
|
def test_same_named_function_is_not_a_type_reference_target(tmp_path):
|
|
src = "class Helper:\n def make(self, w: Widget) -> None:\n w.draw()\n\ndef widget():\n return 1\n"
|
|
result, labels, edges = _extract(tmp_path, {"widget.py": src})
|
|
assert (".make()", "references", "widget()") not in edges
|
|
assert (".make()", "uses", "widget()") not in edges
|