feat(gui): Add kill button per worker in ticket queue table

This commit is contained in:
2026-03-07 16:19:01 -05:00
parent 597e6b51e2
commit d74f629f47
2 changed files with 52 additions and 1 deletions

View File

@@ -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:

View File

@@ -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}"