diff --git a/src/gui_2.py b/src/gui_2.py index bd6f2a5..835777a 100644 --- a/src/gui_2.py +++ b/src/gui_2.py @@ -1950,6 +1950,10 @@ class App: if t: t['status'] = 'blocked' self._push_mma_state_update() + def _cb_kill_ticket(self, ticket_id: str) -> None: + if self.controller and self.controller.engine: + self.controller.engine.kill_worker(ticket_id) + def _reorder_ticket(self, src_idx: int, dst_idx: int) -> None: if src_idx == dst_idx: return new_tickets = list(self.active_tickets) @@ -1997,12 +2001,13 @@ class App: self.bulk_block() # Table flags = imgui.TableFlags_.borders | imgui.TableFlags_.row_bg | imgui.TableFlags_.resizable | imgui.TableFlags_.scroll_y - if imgui.begin_table("ticket_queue_table", 5, flags, imgui.ImVec2(0, 300)): + if imgui.begin_table("ticket_queue_table", 6, flags, imgui.ImVec2(0, 300)): imgui.table_setup_column("Select", imgui.TableColumnFlags_.width_fixed, 40) imgui.table_setup_column("ID", imgui.TableColumnFlags_.width_fixed, 80) imgui.table_setup_column("Priority", imgui.TableColumnFlags_.width_fixed, 100) imgui.table_setup_column("Status", imgui.TableColumnFlags_.width_fixed, 100) imgui.table_setup_column("Description", imgui.TableColumnFlags_.width_stretch) + imgui.table_setup_column("Actions", imgui.TableColumnFlags_.width_fixed, 80) imgui.table_headers_row() for i, t in enumerate(self.active_tickets): @@ -2061,6 +2066,13 @@ class App: imgui.table_next_column() imgui.text(t.get('description', '')) + # Actions - Kill button for in_progress tickets + imgui.table_next_column() + status = t.get('status', 'todo') + if status == 'in_progress': + if imgui.button(f"Kill##{tid}"): + self._cb_kill_ticket(tid) + imgui.end_table() def _render_mma_dashboard(self) -> None: diff --git a/tests/test_gui_kill_button.py b/tests/test_gui_kill_button.py new file mode 100644 index 0000000..e68c353 --- /dev/null +++ b/tests/test_gui_kill_button.py @@ -0,0 +1,39 @@ +import pytest +from unittest.mock import MagicMock, patch + +def test_gui_has_kill_button_method(): + from src.gui_2 import App + assert hasattr(App, '_cb_kill_ticket'), "App must have _cb_kill_ticket method" + +def test_render_ticket_queue_table_columns(): + with patch("src.gui_2.imgui") as mock_imgui: + mock_imgui.begin_table.return_value = True + mock_imgui.table_setup_column = MagicMock() + mock_imgui.table_headers_row = MagicMock() + mock_imgui.table_next_row = MagicMock() + mock_imgui.table_next_column = MagicMock() + mock_imgui.button.return_value = False + mock_imgui.checkbox = MagicMock(return_value=(False, False)) + mock_imgui.selectable = MagicMock(return_value=(False, False)) + mock_imgui.begin_drag_drop_source = MagicMock(return_value=False) + mock_imgui.begin_drag_drop_target = MagicMock(return_value=False) + mock_imgui.text = MagicMock() + mock_imgui.end_table = MagicMock() + mock_imgui.begin_combo.return_value = False + mock_imgui.ComboFlags_.height_small = 0 + mock_imgui.push_style_color = MagicMock() + mock_imgui.pop_style_color = MagicMock() + mock_imgui.same_line = MagicMock() + + from src.gui_2 import App + app = App.__new__(App) + app.active_track = MagicMock() + app.active_tickets = [{"id": "T-001", "priority": "medium", "status": "in_progress", "description": "Test task"}] + app.ui_selected_tickets = set() + app.ui_selected_ticket_id = None + app.controller = MagicMock() + app._push_mma_state_update = MagicMock() + app._cb_kill_ticket = MagicMock() + app._render_ticket_queue() + columns_called = [call[0][0] for call in mock_imgui.table_setup_column.call_args_list] + assert "Actions" in columns_called, f"Expected Actions column, got: {columns_called}"