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.
69 lines
3.3 KiB
Python
69 lines
3.3 KiB
Python
from datetime import datetime
|
|
|
|
# Import the real models
|
|
from src.mma import TrackState, Ticket
|
|
from src.models import Metadata
|
|
# Import the persistence functions from project_manager
|
|
from src.project_manager import save_track_state, load_track_state
|
|
|
|
def test_track_state_persistence(tmp_path) -> None:
|
|
"""
|
|
|
|
|
|
Tests saving and loading a TrackState object to/from a TOML file.
|
|
1. Create a TrackState object with sample metadata, discussion, and tasks.
|
|
2. Call save_track_state('test_track', state, base_dir).
|
|
3. Verify that base_dir/conductor/tracks/test_track/state.toml exists.
|
|
4. Call load_track_state('test_track', base_dir) and verify it returns an identical TrackState object.
|
|
"""
|
|
base_dir = tmp_path
|
|
track_id = "test-track-999" # Metadata internal ID
|
|
track_folder_name = "test_track" # Folder name used in persistence
|
|
# 1. Create a TrackState object with sample data
|
|
metadata = Metadata(
|
|
id=track_id,
|
|
name="Test Track",
|
|
status="in_progress",
|
|
created_at=datetime(2023, 1, 1, 12, 0, 0),
|
|
updated_at=datetime(2023, 1, 2, 13, 0, 0)
|
|
)
|
|
discussion = [
|
|
{"role": "User", "content": "Hello", "ts": datetime(2023, 1, 1, 12, 0, 0)},
|
|
{"role": "AI", "content": "Hi there!", "ts": datetime(2023, 1, 1, 12, 0, 5)}
|
|
]
|
|
tasks = [
|
|
Ticket(id="task-1", description="First task", status="completed", assigned_to="worker-1"),
|
|
Ticket(id="task-2", description="Second task", status="todo", assigned_to="worker-2")
|
|
]
|
|
original_state = TrackState(
|
|
metadata=metadata,
|
|
discussion=discussion,
|
|
tasks=tasks
|
|
)
|
|
# 2. Call save_track_state('test_track', state, base_dir)
|
|
save_track_state(track_folder_name, original_state, base_dir)
|
|
# 3. Verify that base_dir/conductor/tracks/test_track/state.toml exists
|
|
state_file_path = base_dir / "conductor" / "tracks" / track_folder_name / "state.toml"
|
|
assert state_file_path.exists(), f"State file should exist at {state_file_path}"
|
|
# 4. Call load_track_state('test_track', base_dir) and verify it returns an identical TrackState object
|
|
loaded_state = load_track_state(track_folder_name, base_dir)
|
|
assert loaded_state is not None, "load_track_state returned None"
|
|
# Verify equality
|
|
assert loaded_state.metadata.id == original_state.metadata.id
|
|
assert loaded_state.metadata.name == original_state.metadata.name
|
|
assert loaded_state.metadata.status == original_state.metadata.status
|
|
assert loaded_state.metadata.created_at == original_state.metadata.created_at
|
|
assert loaded_state.metadata.updated_at == original_state.metadata.updated_at
|
|
assert len(loaded_state.tasks) == len(original_state.tasks)
|
|
for i in range(len(original_state.tasks)):
|
|
assert loaded_state.tasks[i].id == original_state.tasks[i].id
|
|
assert loaded_state.tasks[i].description == original_state.tasks[i].description
|
|
assert loaded_state.tasks[i].status == original_state.tasks[i].status
|
|
assert loaded_state.tasks[i].assigned_to == original_state.tasks[i].assigned_to
|
|
assert len(loaded_state.discussion) == len(original_state.discussion)
|
|
for i in range(len(original_state.discussion)):
|
|
assert loaded_state.discussion[i]["role"] == original_state.discussion[i]["role"]
|
|
assert loaded_state.discussion[i]["content"] == original_state.discussion[i]["content"]
|
|
assert loaded_state.discussion[i]["ts"] == original_state.discussion[i]["ts"]
|
|
# Final check: deep equality of dataclasses
|
|
assert loaded_state == original_state |