164 lines
6.5 KiB
Python
164 lines
6.5 KiB
Python
|
|
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
from gui_2 import App
|
|
from models import Track
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_mock_app(mock_app: App):
|
|
mock_app._discussion_names_dirty = True
|
|
mock_app._discussion_names_cache = []
|
|
mock_app.active_track = Track(id="track-1", description="Test Track", tickets=[])
|
|
mock_app.active_tickets = []
|
|
mock_app.ui_files_base_dir = "."
|
|
mock_app.disc_roles = ["User", "AI"]
|
|
mock_app.active_discussion = "main"
|
|
mock_app.project = {"discussion": {"discussions": {"main": {"history": []}}}}
|
|
|
|
def test_add_ticket_logic(mock_app: App):
|
|
# Mock imgui calls to simulate clicking "Create" in the form
|
|
with patch('gui_2.imgui') as mock_imgui:
|
|
# Default return for any checkbox/input
|
|
mock_imgui.checkbox.side_effect = lambda label, value: (False, value)
|
|
mock_imgui.input_text.side_effect = lambda label, value, **kwargs: (False, value)
|
|
mock_imgui.input_text_multiline.side_effect = lambda label, value, *args, **kwargs: (False, value)
|
|
mock_imgui.input_int.side_effect = lambda label, value, *args, **kwargs: (False, value)
|
|
mock_imgui.begin_table.return_value = False
|
|
mock_imgui.collapsing_header.return_value = False
|
|
mock_imgui.begin_combo.return_value = False
|
|
|
|
# Simulate form state
|
|
mock_app._show_add_ticket_form = True
|
|
mock_app.ui_new_ticket_id = "T-001"
|
|
mock_app.ui_new_ticket_desc = "Test Description"
|
|
mock_app.ui_new_ticket_target = "test.py"
|
|
mock_app.ui_new_ticket_deps = "T-000"
|
|
|
|
# Configure mock_imgui.button to return True only for "Create"
|
|
def button_side_effect(label):
|
|
return label == "Create"
|
|
mock_imgui.button.side_effect = button_side_effect
|
|
# Mock other necessary imgui calls to avoid errors
|
|
mock_imgui.begin_child.return_value = True
|
|
mock_imgui.get_window_draw_list.return_value.add_rect_filled = MagicMock()
|
|
|
|
# We also need to mock _push_mma_state_update
|
|
with patch.object(mock_app, '_push_mma_state_update') as mock_push:
|
|
mock_app._render_mma_dashboard()
|
|
|
|
# Verify ticket was added
|
|
assert len(mock_app.active_tickets) == 1
|
|
t = mock_app.active_tickets[0]
|
|
assert t["id"] == "T-001"
|
|
assert t["description"] == "Test Description"
|
|
assert t["target_file"] == "test.py"
|
|
assert t["depends_on"] == ["T-000"]
|
|
assert t["status"] == "todo"
|
|
assert t["assigned_to"] == "tier3-worker"
|
|
|
|
# Verify form was closed
|
|
assert not mock_app._show_add_ticket_form
|
|
# Verify push was called
|
|
mock_push.assert_called_once()
|
|
|
|
def test_delete_ticket_logic(mock_app: App):
|
|
# Setup tickets
|
|
mock_app.active_tickets = [
|
|
{"id": "T-001", "status": "todo", "depends_on": []},
|
|
{"id": "T-002", "status": "todo", "depends_on": ["T-001"]}
|
|
]
|
|
tickets_by_id = {t['id']: t for t in mock_app.active_tickets}
|
|
children_map = {"T-001": ["T-002"]}
|
|
rendered = set()
|
|
|
|
with patch('gui_2.imgui') as mock_imgui:
|
|
# Configure mock_imgui.button to return True only for "Delete##T-001"
|
|
def button_side_effect(label):
|
|
return label == "Delete##T-001"
|
|
mock_imgui.button.side_effect = button_side_effect
|
|
mock_imgui.tree_node_ex.return_value = True
|
|
mock_imgui.get_window_draw_list.return_value.add_rect_filled = MagicMock()
|
|
|
|
with patch.object(mock_app, '_push_mma_state_update') as mock_push:
|
|
# Render T-001
|
|
mock_app._render_ticket_dag_node(mock_app.active_tickets[0], tickets_by_id, children_map, rendered)
|
|
|
|
# Verify T-001 was deleted
|
|
assert len(mock_app.active_tickets) == 1
|
|
assert mock_app.active_tickets[0]["id"] == "T-002"
|
|
# Verify dependency cleanup
|
|
assert mock_app.active_tickets[0]["depends_on"] == []
|
|
# Verify push was called
|
|
mock_push.assert_called_once()
|
|
|
|
def test_track_discussion_toggle(mock_app: App):
|
|
with (
|
|
patch('gui_2.imgui') as mock_imgui,
|
|
patch('gui_2.project_manager.load_track_history', return_value=["@2026-03-01 12:00:00\n[User]\nTrack Hello"]) as mock_load,
|
|
patch('gui_2.project_manager.str_to_entry', side_effect=lambda s, roles: {"ts": "12:00", "role": "User", "content": s.split("\n")[-1]}),
|
|
patch.object(mock_app, '_flush_disc_entries_to_project') as mock_flush,
|
|
patch.object(mock_app, '_switch_discussion') as mock_switch
|
|
):
|
|
# Track calls to ensure we only return 'changed=True' once to avoid loops
|
|
calls = {"Track Discussion": 0}
|
|
def checkbox_side_effect(label, value):
|
|
if label == "Track Discussion":
|
|
calls[label] += 1
|
|
# Only return True for 'changed' on the first call in the test
|
|
changed = (calls[label] == 1)
|
|
return changed, True
|
|
return False, value
|
|
|
|
mock_imgui.checkbox.side_effect = checkbox_side_effect
|
|
mock_imgui.begin_combo.return_value = False
|
|
mock_imgui.selectable.return_value = (False, False)
|
|
mock_imgui.button.return_value = False
|
|
mock_imgui.collapsing_header.return_value = True # For Discussions header
|
|
mock_imgui.input_text.side_effect = lambda label, value, **kwargs: (False, value)
|
|
mock_imgui.input_int.side_effect = lambda label, value, *args, **kwargs: (False, value)
|
|
mock_imgui.begin_child.return_value = True
|
|
# Mock clipper to avoid the while loop hang
|
|
mock_clipper = MagicMock()
|
|
mock_clipper.step.side_effect = [True, False]
|
|
mock_clipper.display_start = 0
|
|
mock_clipper.display_end = 0
|
|
mock_imgui.ListClipper.return_value = mock_clipper
|
|
|
|
mock_app._render_discussion_panel()
|
|
|
|
assert mock_app._track_discussion_active
|
|
mock_flush.assert_called()
|
|
mock_load.assert_called_with("track-1", ".")
|
|
assert len(mock_app.disc_entries) == 1
|
|
assert mock_app.disc_entries[0]["content"] == "Track Hello"
|
|
|
|
# Now toggle OFF
|
|
calls["Track Discussion"] = 0 # Reset for next call
|
|
def checkbox_off_side_effect(label, value):
|
|
if label == "Track Discussion":
|
|
calls[label] += 1
|
|
return (calls[label] == 1), False
|
|
return False, value
|
|
mock_imgui.checkbox.side_effect = checkbox_off_side_effect
|
|
mock_clipper.step.side_effect = [True, False] # Reset clipper
|
|
|
|
mock_app._render_discussion_panel()
|
|
|
|
assert not mock_app._track_discussion_active
|
|
mock_switch.assert_called_with(mock_app.active_discussion)
|
|
|
|
def test_push_mma_state_update(mock_app: App):
|
|
mock_app.active_tickets = [{"id": "T-001", "description": "desc", "status": "todo", "assigned_to": "tier3-worker", "depends_on": []}]
|
|
with patch('project_manager.save_track_state') as mock_save, \
|
|
patch('project_manager.load_track_state', return_value=None):
|
|
mock_app._push_mma_state_update()
|
|
|
|
assert len(mock_app.active_track.tickets) == 1
|
|
assert mock_app.active_track.tickets[0].id == "T-001"
|
|
assert mock_save.called
|
|
args, kwargs = mock_save.call_args
|
|
assert args[0] == "track-1"
|
|
state = args[1]
|
|
assert state.metadata.id == "track-1"
|
|
assert state.tasks == mock_app.active_track.tickets
|