ed
8f11340b38
Per post_module_taxonomy_de_cruft_20260627 Phase 2 (FR7). Each
'from src.models import X' for a moved class is rewritten to
'from src.<destination> import X':
Ticket, Track, WorkerContext, TrackState, TrackMetadata,
ThinkingSegment, EMPTY_TRACK_STATE -> src.mma
ProjectContext, ProjectMeta, ProjectOutput, ProjectFiles,
ProjectScreenshots, ProjectDiscussion, EMPTY_PROJECT_CONTEXT -> src.project
FileItem, Preset, ContextPreset, ContextFileEntry,
NamedViewPreset -> src.project_files
Tool, ToolPreset -> src.tool_presets
BiasProfile -> src.tool_bias
TextEditorConfig, ExternalEditorConfig,
EMPTY_TEXT_EDITOR_CONFIG -> src.external_editor
Persona -> src.personas
WorkspaceProfile -> src.workspace_manager
MCPServerConfig, MCPConfiguration, VectorStoreConfig,
RAGConfig, load_mcp_config -> src.mcp_client
NOT touched (kept on src.models; Phase 3 or Phase 4 will move them):
GenerateRequest, ConfirmRequest, DEFAULT_TOOL_CATEGORIES, Metadata, PROVIDERS
Migration was performed by the one-time script
scripts/tier2/artifacts/post_module_taxonomy_de_cruft_20260627/migrate_imports.py
which uses a class-to-module map and re.sub() to rewrite each
'from src.models import X' line.
Total: 85 import lines rewritten across 71 files.
Note: this commit depends on the v2 SHIPPED work
(origin/tier2/module_taxonomy_refactor_20260627) being merged into
this branch NEXT. On master (without the v2 SHIPPED commits), the
destination modules do not exist and these imports would fail.
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
import pytest
|
|
import time
|
|
from pathlib import Path
|
|
from src.file_cache import ASTParser
|
|
from src.mma import Ticket, Track, WorkerContext
|
|
from src.multi_agent_conductor import run_worker_lifecycle
|
|
from src.result_types import Result
|
|
|
|
def test_targeted_extraction():
|
|
parser = ASTParser("python")
|
|
code = """
|
|
def func_a():
|
|
print("A")
|
|
func_b()
|
|
|
|
def func_b():
|
|
print("B")
|
|
func_c()
|
|
|
|
def func_c():
|
|
print("C")
|
|
|
|
def func_unrelated():
|
|
print("Unrelated")
|
|
"""
|
|
# Target func_a, should include func_b and func_c (depth 2)
|
|
result = parser.get_targeted_view(code, ["func_a"])
|
|
assert "def func_a():" in result
|
|
assert "def func_b():" in result
|
|
assert "def func_c():" in result
|
|
assert "def func_unrelated():" not in result
|
|
assert "print(" not in result # Bodies should be stripped
|
|
|
|
def test_class_targeted_extraction():
|
|
parser = ASTParser("python")
|
|
code = """
|
|
class MyClass:
|
|
def method_a(self):
|
|
self.method_b()
|
|
|
|
def method_b(self):
|
|
pass
|
|
|
|
def method_unrelated(self):
|
|
pass
|
|
"""
|
|
result = parser.get_targeted_view(code, ["MyClass.method_a"])
|
|
assert "class MyClass:" in result
|
|
assert "def method_a(self):" in result
|
|
assert "def method_b(self):" in result
|
|
assert "def method_unrelated(self):" not in result
|
|
|
|
def test_ast_caching(tmp_path):
|
|
parser = ASTParser("python")
|
|
file_path = tmp_path / "test_cache.py"
|
|
code = "def test(): pass"
|
|
file_path.write_text(code)
|
|
|
|
# First call: parses and caches
|
|
start = time.time()
|
|
view1 = parser.get_skeleton(code, path=str(file_path))
|
|
duration1 = time.time() - start
|
|
|
|
# Second call: should use cache
|
|
start = time.time()
|
|
view2 = parser.get_skeleton(code, path=str(file_path))
|
|
duration2 = time.time() - start
|
|
|
|
assert view1 == view2
|
|
# duration2 should be significantly faster, but let's just check it works
|
|
|
|
# Update file: should invalidate cache
|
|
time.sleep(0.1)
|
|
new_code = "def test_new(): pass"
|
|
file_path.write_text(new_code)
|
|
view3 = parser.get_skeleton(new_code, path=str(file_path))
|
|
assert "def test_new():" in view3
|
|
assert "def test():" not in view3
|
|
|
|
def test_performance_large_file():
|
|
parser = ASTParser("python")
|
|
# Generate a large file (approx 1000 lines)
|
|
code = "\n".join([f"def func_{i}():\n pass" for i in range(500)])
|
|
|
|
start = time.time()
|
|
parser.get_skeleton(code)
|
|
duration = time.time() - start
|
|
|
|
print(f"Large file parse duration: {duration*1000:.2f}ms")
|
|
assert duration < 0.5 # Should be well under 500ms even for first parse
|
|
|
|
def test_token_reduction_logging(capsys):
|
|
ticket = Ticket(
|
|
id="T1",
|
|
description="Test ticket",
|
|
target_file="test.py",
|
|
target_symbols=["func_a"],
|
|
context_requirements=["test.py"]
|
|
)
|
|
context = WorkerContext(ticket_id="T1", model_name="gemini-2.5-flash", messages=[])
|
|
|
|
code = "def func_a(): pass\n" + "\n".join([f"def func_{i}(): pass" for i in range(100)])
|
|
|
|
# Mock open to return our large code
|
|
with pytest.MonkeyPatch().context() as m:
|
|
m.setattr("builtins.open", lambda f, *args, **kwargs: type('obj', (object,), {'read': lambda s: code, '__enter__': lambda s: s, '__exit__': lambda s, *a: None})())
|
|
m.setattr("pathlib.Path.exists", lambda s: True)
|
|
m.setattr("src.ai_client.send", lambda **kwargs: Result(data="DONE"))
|
|
|
|
run_worker_lifecycle(ticket, context, context_files=["test.py"])
|
|
|
|
captured = capsys.readouterr()
|
|
assert "[MMA] Context pruning for T1" in captured.out
|
|
assert "reduction" in captured.out
|