feat(ui): Implement Track Browser and progress visualization in MMA Dashboard

This commit is contained in:
2026-02-27 22:49:03 -05:00
parent a97eb2a222
commit 2b1cfbb34d
6 changed files with 315 additions and 3 deletions

View File

@@ -0,0 +1,66 @@
import pytest
from unittest.mock import patch, MagicMock
from gui_2 import App
@pytest.fixture
def app_instance():
# We patch the dependencies of App.__init__ to avoid side effects
with (
patch('gui_2.load_config', return_value={'ai': {}, 'projects': {}}),
patch('gui_2.save_config'),
patch('gui_2.project_manager') as mock_pm,
patch('gui_2.session_logger'),
patch('gui_2.immapp.run'),
patch.object(App, '_load_active_project'),
patch.object(App, '_fetch_models'),
patch.object(App, '_load_fonts'),
patch.object(App, '_post_init')
):
app = App()
# Ensure project and ui_files_base_dir are set for _refresh_from_project
app.project = {}
app.ui_files_base_dir = "."
# Return the app and the mock_pm for use in tests
yield app, mock_pm
def test_mma_dashboard_refresh(app_instance):
app, mock_pm = app_instance
# 1. Define mock tracks
mock_tracks = [
MagicMock(id="track_1", description="Track 1"),
MagicMock(id="track_2", description="Track 2")
]
# 2. Patch get_all_tracks to return our mock list
mock_pm.get_all_tracks.return_value = mock_tracks
# 3. Call _refresh_from_project
app._refresh_from_project()
# 4. Verify that app.tracks contains the mock tracks
assert hasattr(app, 'tracks'), "App instance should have a 'tracks' attribute"
assert app.tracks == mock_tracks
assert len(app.tracks) == 2
assert app.tracks[0].id == "track_1"
assert app.tracks[1].id == "track_2"
# Verify get_all_tracks was called with the correct base_dir
mock_pm.get_all_tracks.assert_called_with(app.ui_files_base_dir)
def test_mma_dashboard_initialization_refresh(app_instance):
"""
Checks that _refresh_from_project is called during initialization if
_load_active_project is NOT mocked to skip it (but here it IS mocked in fixture).
This test verifies that calling it manually works as expected for initialization scenarios.
"""
app, mock_pm = app_instance
mock_tracks = [MagicMock(id="init_track", description="Initial Track")]
mock_pm.get_all_tracks.return_value = mock_tracks
# Simulate the refresh that would happen during a project load
app._refresh_from_project()
assert app.tracks == mock_tracks
assert app.tracks[0].id == "init_track"

View File

@@ -0,0 +1,94 @@
import pytest
import json
from pathlib import Path
from project_manager import get_all_tracks, save_track_state
from models import TrackState, Metadata, Ticket
from datetime import datetime
def test_get_all_tracks_empty(tmp_path):
# conductor/tracks directory doesn't exist
assert get_all_tracks(tmp_path) == []
def test_get_all_tracks_with_state(tmp_path):
tracks_dir = tmp_path / "conductor" / "tracks"
tracks_dir.mkdir(parents=True)
track_id = "test_track_1"
track_dir = tracks_dir / track_id
track_dir.mkdir()
# Create TrackState
metadata = Metadata(id=track_id, name="Test Track 1", status="in_progress",
created_at=datetime.now(), updated_at=datetime.now())
tasks = [
Ticket(id="task1", description="desc1", status="completed", assigned_to="user"),
Ticket(id="task2", description="desc2", status="todo", assigned_to="user")
]
state = TrackState(metadata=metadata, discussion=[], tasks=tasks)
save_track_state(track_id, state, tmp_path)
tracks = get_all_tracks(tmp_path)
assert len(tracks) == 1
track = tracks[0]
assert track["id"] == track_id
assert track["title"] == "Test Track 1"
assert track["status"] == "in_progress"
assert track["complete"] == 1
assert track["total"] == 2
assert track["progress"] == 0.5
def test_get_all_tracks_with_metadata_json(tmp_path):
tracks_dir = tmp_path / "conductor" / "tracks"
tracks_dir.mkdir(parents=True)
track_id = "test_track_2"
track_dir = tracks_dir / track_id
track_dir.mkdir()
metadata = {
"id": track_id,
"title": "Test Track 2",
"status": "planned"
}
with open(track_dir / "metadata.json", "w") as f:
json.dump(metadata, f)
# Create plan.md to test parsing
plan_content = """
# Plan
- [x] Task: Task 1
- [ ] Task: Task 2
- [~] Task: Task 3
"""
with open(track_dir / "plan.md", "w") as f:
f.write(plan_content)
tracks = get_all_tracks(tmp_path)
assert len(tracks) == 1
track = tracks[0]
assert track["id"] == track_id
assert track["title"] == "Test Track 2"
assert track["status"] == "planned"
assert track["complete"] == 1
assert track["total"] == 3
assert pytest.approx(track["progress"]) == 0.333333
def test_get_all_tracks_malformed(tmp_path):
tracks_dir = tmp_path / "conductor" / "tracks"
tracks_dir.mkdir(parents=True)
track_id = "malformed_track"
track_dir = tracks_dir / track_id
track_dir.mkdir()
# Malformed metadata.json
with open(track_dir / "metadata.json", "w") as f:
f.write("{ invalid json }")
tracks = get_all_tracks(tmp_path)
assert len(tracks) == 1
track = tracks[0]
assert track["id"] == track_id
assert track["status"] == "unknown"
assert track["complete"] == 0
assert track["total"] == 0