feat(api): Expand API hooks with drag and right_click actions

This commit is contained in:
2026-05-12 19:06:54 -04:00
parent d92361b025
commit a3b117dabd
4 changed files with 72 additions and 2 deletions
+12
View File
@@ -190,6 +190,18 @@ class ApiHookClient:
""" """
return self.set_value(item, value) return self.set_value(item, value)
def drag(self, src_item: str, dst_item: str) -> dict[str, Any]:
"""
Simulates a drag and drop operation.
"""
return self.push_event("drag", {"src_item": src_item, "dst_item": dst_item})
def right_click(self, item: str) -> dict[str, Any]:
"""
Simulates a right-click on an item.
"""
return self.push_event("right_click", {"item": item})
def get_gui_state(self) -> dict[str, Any]: def get_gui_state(self) -> dict[str, Any]:
""" """
Returns the full GUI state available via the hook API. Returns the full GUI state available via the hook API.
+13
View File
@@ -682,6 +682,8 @@ class AppController:
'btn_rebuild_rag_index': self._rebuild_rag_index, 'btn_rebuild_rag_index': self._rebuild_rag_index,
'btn_clear_summary_cache': self._handle_clear_summary_cache, 'btn_clear_summary_cache': self._handle_clear_summary_cache,
} }
self._drag_actions: dict[str, Callable[..., Any]] = {}
self._right_clickable_actions: dict[str, Callable[..., Any]] = {}
self._predefined_callbacks: dict[str, Callable[..., Any]] = { self._predefined_callbacks: dict[str, Callable[..., Any]] = {
'_test_callback_func_write_to_file': self._test_callback_func_write_to_file, '_test_callback_func_write_to_file': self._test_callback_func_write_to_file,
'_set_env_var': lambda k, v: os.environ.update({k: v}), '_set_env_var': lambda k, v: os.environ.update({k: v}),
@@ -910,6 +912,17 @@ class AppController:
func() func()
except Exception: except Exception:
func() func()
elif action == "drag":
src_item = task.get("src_item")
dst_item = task.get("dst_item")
if src_item in self._drag_actions:
func = self._drag_actions[src_item]
func(dst_item=dst_item)
elif action == "right_click":
item = task.get("item")
if item in self._right_clickable_actions:
func = self._right_clickable_actions[item]
func()
elif action == "select_list_item": elif action == "select_list_item":
item = task.get("listbox", task.get("item")) item = task.get("listbox", task.get("item"))
value = task.get("item_value", task.get("value")) value = task.get("item_value", task.get("value"))
+26 -1
View File
@@ -81,4 +81,29 @@ def test_get_node_status() -> None:
} }
status = client.get_node_status("T1") status = client.get_node_status("T1")
assert status["status"] == "todo" assert status["status"] == "todo"
mock_make.assert_any_call('GET', '/api/mma/node/T1') mock_make.assert_any_call('GET', '/api/mma/node/T1')
def test_drag_success() -> None:
"""Test that drag correctly sends a POST request to the /api/gui endpoint."""
client = ApiHookClient()
with patch.object(client, '_make_request') as mock_make:
mock_make.return_value = {"status": "queued"}
res = client.drag("src_id", "dst_id")
assert res["status"] == "queued"
mock_make.assert_any_call('POST', '/api/gui', data={
"action": "drag",
"src_item": "src_id",
"dst_item": "dst_id"
})
def test_right_click_success() -> None:
"""Test that right_click correctly sends a POST request to the /api/gui endpoint."""
client = ApiHookClient()
with patch.object(client, '_make_request') as mock_make:
mock_make.return_value = {"status": "queued"}
res = client.right_click("item_id")
assert res["status"] == "queued"
mock_make.assert_any_call('POST', '/api/gui', data={
"action": "right_click",
"item": "item_id"
})
+21 -1
View File
@@ -1,6 +1,6 @@
from typing import Generator from typing import Generator
import pytest import pytest
from unittest.mock import patch from unittest.mock import patch, Mock
from src import ai_client from src import ai_client
from src.gui_2 import App from src.gui_2 import App
@@ -46,3 +46,23 @@ def test_gcli_path_updates_adapter(app_instance: App) -> None:
app_instance.controller._process_pending_gui_tasks() app_instance.controller._process_pending_gui_tasks()
assert ai_client._gemini_cli_adapter is not None assert ai_client._gemini_cli_adapter is not None
assert ai_client._gemini_cli_adapter.binary_path == '/new/path/to/gemini' assert ai_client._gemini_cli_adapter.binary_path == '/new/path/to/gemini'
def test_process_pending_gui_tasks_drag(app_instance: App) -> None:
"""Test that the drag action is correctly processed and dispatches to the registered callback."""
mock_callback = Mock()
app_instance.controller._drag_actions["src_id"] = mock_callback
app_instance.controller._pending_gui_tasks = [
{"action": "drag", "src_item": "src_id", "dst_item": "dst_id"}
]
app_instance.controller._process_pending_gui_tasks()
mock_callback.assert_called_once_with(dst_item="dst_id")
def test_process_pending_gui_tasks_right_click(app_instance: App) -> None:
"""Test that the right_click action is correctly processed and dispatches to the registered callback."""
mock_callback = Mock()
app_instance.controller._right_clickable_actions["item_id"] = mock_callback
app_instance.controller._pending_gui_tasks = [
{"action": "right_click", "item": "item_id"}
]
app_instance.controller._process_pending_gui_tasks()
mock_callback.assert_called_once()