"""Tests for Iteration 2: Code Entity Fidelity."""
import pytest
from pathlib import Path
@pytest.fixture
def temp_project(tmp_path):
"""Create a temporary project directory."""
return tmp_path
@pytest.fixture
def nested_python(temp_project):
"""Create a Python file with nested classes."""
code = '''
class User:
"""User model."""
class Profile:
"""Nested profile."""
def update(self):
pass
def save(self):
pass
def helper():
pass
'''
py_file = temp_project / "models.py"
py_file.write_text(code)
return py_file
class TestQualifiedNames:
"""Test qualified name computation."""
def test_qualified_name_nested_class_method(self, temp_project, nested_python):
"""Nested method has fully qualified name."""
from daem0nmcp.code_indexer import TreeSitterIndexer
indexer = TreeSitterIndexer()
if not indexer.available:
pytest.skip("tree-sitter not available")
entities = list(indexer.index_file(nested_python, temp_project))
update_method = next((e for e in entities if e['name'] == 'update'), None)
assert update_method is not None
assert 'qualified_name' in update_method
assert 'User' in update_method['qualified_name']
assert 'Profile' in update_method['qualified_name']
def test_qualified_name_top_level_function(self, temp_project, nested_python):
"""Top-level function has module-prefixed name."""
from daem0nmcp.code_indexer import TreeSitterIndexer
indexer = TreeSitterIndexer()
if not indexer.available:
pytest.skip("tree-sitter not available")
entities = list(indexer.index_file(nested_python, temp_project))
helper_func = next((e for e in entities if e['name'] == 'helper'), None)
assert helper_func is not None
assert 'qualified_name' in helper_func
assert 'models' in helper_func['qualified_name']
class TestStableEntityIDs:
"""Test entity ID stability across line changes."""
def test_entity_id_stable_after_line_change(self, temp_project):
"""Adding lines should NOT change entity IDs."""
from daem0nmcp.code_indexer import TreeSitterIndexer
indexer = TreeSitterIndexer()
if not indexer.available:
pytest.skip("tree-sitter not available")
py_file = temp_project / "service.py"
py_file.write_text('class UserService:\n def authenticate(self): pass')
entities1 = list(indexer.index_file(py_file, temp_project))
# Add lines before (shifts line numbers)
py_file.write_text('# comment\n# another\nclass UserService:\n def authenticate(self): pass')
entities2 = list(indexer.index_file(py_file, temp_project))
# IDs should be the same
for e1 in entities1:
matching = [e2 for e2 in entities2 if e2['name'] == e1['name'] and e2['entity_type'] == e1['entity_type']]
assert len(matching) == 1, f"No match for {e1['name']}"
assert matching[0]['id'] == e1['id'], f"ID changed for {e1['name']}"
class TestImportExtraction:
"""Test import extraction from AST."""
def test_python_imports_extracted(self, temp_project):
"""Python imports should be extracted."""
from daem0nmcp.code_indexer import TreeSitterIndexer
indexer = TreeSitterIndexer()
if not indexer.available:
pytest.skip("tree-sitter not available")
py_file = temp_project / "app.py"
py_file.write_text('import os\nfrom pathlib import Path\ndef main(): pass')
entities = list(indexer.index_file(py_file, temp_project))
main_func = next((e for e in entities if e['name'] == 'main'), None)
assert main_func is not None
imports = main_func.get('imports', [])
assert len(imports) > 0, "No imports extracted"
assert 'os' in imports or any('os' in i for i in imports)
