feat(gui): Add kill button per worker in ticket queue table
This commit is contained in:
14
src/gui_2.py
14
src/gui_2.py
@@ -1950,6 +1950,10 @@ class App:
|
|||||||
if t: t['status'] = 'blocked'
|
if t: t['status'] = 'blocked'
|
||||||
self._push_mma_state_update()
|
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:
|
def _reorder_ticket(self, src_idx: int, dst_idx: int) -> None:
|
||||||
if src_idx == dst_idx: return
|
if src_idx == dst_idx: return
|
||||||
new_tickets = list(self.active_tickets)
|
new_tickets = list(self.active_tickets)
|
||||||
@@ -1997,12 +2001,13 @@ class App:
|
|||||||
self.bulk_block()
|
self.bulk_block()
|
||||||
# Table
|
# Table
|
||||||
flags = imgui.TableFlags_.borders | imgui.TableFlags_.row_bg | imgui.TableFlags_.resizable | imgui.TableFlags_.scroll_y
|
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("Select", imgui.TableColumnFlags_.width_fixed, 40)
|
||||||
imgui.table_setup_column("ID", imgui.TableColumnFlags_.width_fixed, 80)
|
imgui.table_setup_column("ID", imgui.TableColumnFlags_.width_fixed, 80)
|
||||||
imgui.table_setup_column("Priority", imgui.TableColumnFlags_.width_fixed, 100)
|
imgui.table_setup_column("Priority", imgui.TableColumnFlags_.width_fixed, 100)
|
||||||
imgui.table_setup_column("Status", 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("Description", imgui.TableColumnFlags_.width_stretch)
|
||||||
|
imgui.table_setup_column("Actions", imgui.TableColumnFlags_.width_fixed, 80)
|
||||||
imgui.table_headers_row()
|
imgui.table_headers_row()
|
||||||
|
|
||||||
for i, t in enumerate(self.active_tickets):
|
for i, t in enumerate(self.active_tickets):
|
||||||
@@ -2061,6 +2066,13 @@ class App:
|
|||||||
imgui.table_next_column()
|
imgui.table_next_column()
|
||||||
imgui.text(t.get('description', ''))
|
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()
|
imgui.end_table()
|
||||||
|
|
||||||
def _render_mma_dashboard(self) -> None:
|
def _render_mma_dashboard(self) -> None:
|
||||||
|
|||||||
39
tests/test_gui_kill_button.py
Normal file
39
tests/test_gui_kill_button.py
Normal 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}"
|
||||||
Reference in New Issue
Block a user