vudt15andClaude Sonnet 5
0e51356a7d
Team Hoa, EPIC R08 (UI/Application Separation) - Team Hoa scope only
(R08-T11 -> T14; R08-T01->T10 belong to Team Duy/Team Nam).
- R08-T11: ui/schedule_task_tab.py (795 lines) -> presentation/scheduling/
{kanban_board_widget,calendar_view_widget,ai_task_creator_dialog,
ai_task_import_dialog,run_history_dialog}.py + schedule_task_tab.py
shell. Kanban CRUD/drag-drop now goes through
application/scheduling/task_application_service.py (R07-T04) instead of
~30 lines of inline if/elif per drag target.
- R08-T12: ui/folder_tab.py (1587 lines, the largest of the four) ->
presentation/folder/{workspace_file_tree,document_preview_manager,
code_editor,office_document_renderer,ai_file_editor_dialog,
ai_edit_model_resolver,ai_edit_pipeline}.py + folder_tab.py shell.
Closes the R06-T05 loop: FileWorkspaceService existed since R06 with
zero production call sites (confirmed by grep); every plain-text write
(save/create/write_content) now goes through it, gaining path
containment and a Python-syntax warning the original code never had.
Pure helpers (_read_text, _is_probably_text, _pptx_available,
_split_code_block, _parse_ai_output) moved to
application/workspaces/{file_preview_helpers,ai_edit_output}.py.
- R08-T13: ui/dashboard_tab.py (437 lines) -> presentation/dashboard/
{token_usage_card_widget,usage_chart_widget,habits_widget}.py +
dashboard_tab.py shell, backed by a new
application/monitoring/dashboard_query_service.py (pricing/period/
summary queries the three widgets used to each recompute separately).
Directory-ownership note left in the checklist for Team Nam.
- R08-T14: ui/structure_graph_view.py (1035 lines) ->
presentation/graph/{graph_scene_items,graph_renderer,
graph_messages_view,graph_qa_widget}.py + structure_graph_view.py
shell. Extraction helpers (_pdf_to_markdown, _extract_file_contents)
moved to application/workspaces/graph_index_service.py (pure Python).
Renderer and Q&A panel talk only through signals
(node_selected/graph_rendered/raw_json_ready/project_changed) - neither
imports the other.
- presentation/shared/web_engine_support.py: HAS_WEB_ENGINE, previously
duplicated (folder_tab imported it FROM structure_graph_view.py) - now
one shared flag instead of one screen importing another screen's module.
All four old ui/*.py files deleted; app.py and ui/workspace_tab.py updated
to the new import paths (each god-file only had 1-2 real construction
sites, so import sites were updated directly rather than kept as a
strangler-fig shim - unlike core/tools.py at R05, which had dozens).
pytest: 377 pass (+94 vs the R07 baseline of 328; same 4 pre-existing
failures as the R05/R06 baseline, unrelated to this work).
scripts/check_imports.py: PASS. python -c "import cowork_local.app": OK.
Every new file < 400 lines (largest: graph_renderer.py, 391).
Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
"""EPIC R08-T13: DashboardQueryService — no Qt.
|
|
|
|
``core/usage_tracker.py::USAGE_DIR`` is a module-level constant (not
|
|
injectable per-call except through an explicit ``directory=`` kwarg
|
|
``load_events`` alone accepts) — this is a pre-existing testability gap the
|
|
original ``ui/dashboard_tab.py`` also had (it had zero tests before this
|
|
task). Monkeypatching the module attribute is what lets these tests write
|
|
usage events without touching the real ``~/.cowork_local/usage/``.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from datetime import date, timedelta
|
|
|
|
import pytest
|
|
|
|
from cowork_local.application.monitoring import DashboardQueryService
|
|
from cowork_local.config import AppConfig
|
|
from cowork_local.core import usage_tracker as ut
|
|
from cowork_local.state import AppContext
|
|
|
|
|
|
@pytest.fixture
|
|
def usage_dir(tmp_path, monkeypatch):
|
|
d = tmp_path / "usage"
|
|
monkeypatch.setattr(ut, "USAGE_DIR", d)
|
|
return d
|
|
|
|
|
|
@pytest.fixture
|
|
def ctx(tmp_path):
|
|
return AppContext(AppConfig.load(tmp_path / "config.json"))
|
|
|
|
|
|
def _write_event(usage_dir, day: date, **overrides):
|
|
usage_dir.mkdir(parents=True, exist_ok=True)
|
|
event = {
|
|
"ts": f"{day.isoformat()}T10:00:00", "source": "cowork", "label": "Test chat",
|
|
"provider": "anthropic", "model": "claude-sonnet-4-6",
|
|
"in": 100, "out": 50, "cache": 0, "estimated": False,
|
|
"account": "", "machine": "",
|
|
}
|
|
event.update(overrides)
|
|
path = usage_dir / f"{day.isoformat()}.jsonl"
|
|
with path.open("a", encoding="utf-8") as f:
|
|
f.write(json.dumps(event) + "\n")
|
|
|
|
|
|
def test_period_range_is_inclusive_end(usage_dir, ctx):
|
|
query = DashboardQueryService(ctx)
|
|
start, end = query.period_range("week", 0)
|
|
assert start <= end
|
|
|
|
|
|
def test_summary_aggregates_events_in_range(usage_dir, ctx):
|
|
today = date.today()
|
|
_write_event(usage_dir, today, **{"in": 100, "out": 50})
|
|
_write_event(usage_dir, today - timedelta(days=400), **{"in": 999, "out": 999}) # out of range
|
|
query = DashboardQueryService(ctx)
|
|
|
|
summary = query.summary(today, today)
|
|
|
|
assert len(summary["events"]) == 1
|
|
assert summary["stats"]["in"] == 100
|
|
assert summary["stats"]["out"] == 50
|
|
assert summary["total_cost"] >= 0
|
|
|
|
|
|
def test_summary_empty_range_has_no_events(usage_dir, ctx):
|
|
query = DashboardQueryService(ctx)
|
|
summary = query.summary(date(2020, 1, 1), date(2020, 1, 1))
|
|
assert summary["events"] == []
|
|
assert summary["stats"]["total"] == 0
|
|
|
|
|
|
def test_pricing_returns_a_dict_with_currency(usage_dir, ctx):
|
|
query = DashboardQueryService(ctx)
|
|
pricing = query.pricing()
|
|
assert "currency" in pricing
|
|
|
|
|
|
def test_chart_series_returns_points_for_the_granularity(usage_dir, ctx):
|
|
today = date.today()
|
|
_write_event(usage_dir, today)
|
|
query = DashboardQueryService(ctx)
|
|
|
|
pts = query.chart_series("week", 0, "tokens")
|
|
|
|
assert len(pts) == 7 # week view = 7 days
|
|
assert all(isinstance(p, tuple) and len(p) == 2 for p in pts)
|
|
|
|
|
|
def test_budget_status_none_when_no_budget_set(usage_dir, ctx):
|
|
query = DashboardQueryService(ctx)
|
|
assert query.budget_status() is None
|
|
|
|
|
|
def test_set_budget_then_status_reflects_it(usage_dir, ctx):
|
|
query = DashboardQueryService(ctx)
|
|
query.set_budget(100.0, "USD")
|
|
status = query.budget_status()
|
|
assert status is not None
|
|
assert status["amount_usd"] == pytest.approx(100.0)
|