118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
from __future__ import annotations
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from inbox_service import InboxService
|
|
|
|
|
|
NANOBOT_REPO_DIR = Path("/home/kacper/nanobot")
|
|
|
|
|
|
class InboxServiceTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self._tmpdir = tempfile.TemporaryDirectory()
|
|
self.inbox_dir = Path(self._tmpdir.name) / "inbox"
|
|
self.service = InboxService(repo_dir=NANOBOT_REPO_DIR, inbox_dir=self.inbox_dir)
|
|
|
|
def tearDown(self) -> None:
|
|
self._tmpdir.cleanup()
|
|
|
|
def test_capture_list_and_dismiss_item(self) -> None:
|
|
captured = self.service.capture_item(
|
|
title="",
|
|
text="Pack chargers for Japan",
|
|
kind="task",
|
|
source="web-ui",
|
|
due="2026-04-10",
|
|
body="Remember the USB-C one too.",
|
|
session_id="web-123",
|
|
tags=["travel", "#Japan"],
|
|
confidence=0.75,
|
|
)
|
|
|
|
self.assertEqual(captured["title"], "Pack chargers for Japan")
|
|
self.assertEqual(captured["kind"], "task")
|
|
self.assertEqual(captured["source"], "web-ui")
|
|
self.assertEqual(captured["suggested_due"], "2026-04-10")
|
|
self.assertEqual(captured["metadata"]["source_session"], "web-123")
|
|
self.assertIn("## Raw Capture", captured["body"])
|
|
self.assertIn("#travel", captured["tags"])
|
|
self.assertIn("#japan", captured["tags"])
|
|
|
|
listed = self.service.list_items(
|
|
status_filter=None,
|
|
kind_filter=None,
|
|
include_closed=False,
|
|
limit=None,
|
|
tags=[],
|
|
)
|
|
self.assertEqual(len(listed), 1)
|
|
self.assertEqual(listed[0]["path"], captured["path"])
|
|
|
|
dismissed = self.service.dismiss_item(captured["path"])
|
|
self.assertEqual(dismissed["status"], "dismissed")
|
|
self.assertEqual(self.service.open_items(), [])
|
|
|
|
|
|
class InboxServiceAcceptTaskTests(unittest.IsolatedAsyncioTestCase):
|
|
def setUp(self) -> None:
|
|
self._tmpdir = tempfile.TemporaryDirectory()
|
|
self.inbox_dir = Path(self._tmpdir.name) / "inbox"
|
|
self.service = InboxService(repo_dir=NANOBOT_REPO_DIR, inbox_dir=self.inbox_dir)
|
|
self.item = self.service.capture_item(
|
|
title="Find a good PWA install guide",
|
|
text="Find a good PWA install guide",
|
|
kind="task",
|
|
source="web-ui",
|
|
due="",
|
|
body="Look for something practical, not generic.",
|
|
session_id="web-456",
|
|
tags=["research"],
|
|
confidence=0.9,
|
|
)
|
|
|
|
def tearDown(self) -> None:
|
|
self._tmpdir.cleanup()
|
|
|
|
async def test_accept_item_builds_grooming_prompt_with_hints(self) -> None:
|
|
captured: dict[str, Any] = {}
|
|
|
|
async def fake_run_nanobot_turn(message: str, **kwargs: Any) -> dict[str, Any]:
|
|
captured["message"] = message
|
|
captured["kwargs"] = kwargs
|
|
return {"status": "ok", "reply": "accepted"}
|
|
|
|
def fake_notification_to_event(payload: dict[str, Any]) -> dict[str, Any] | None:
|
|
return payload
|
|
|
|
result = await self.service.accept_item_as_task(
|
|
item=self.item["path"],
|
|
lane="committed",
|
|
title="Install the PWA correctly",
|
|
due="2026-04-12",
|
|
body_text="Prefer something with screenshots.",
|
|
tags=["#pwa", "#research"],
|
|
run_nanobot_turn=fake_run_nanobot_turn,
|
|
notification_to_event=fake_notification_to_event,
|
|
)
|
|
|
|
self.assertEqual(result["status"], "ok")
|
|
message = captured["message"]
|
|
self.assertIn("Use the inbox_board tool and the task_helper_card tool.", message)
|
|
self.assertIn(f"Inbox item path: {self.item['path']}", message)
|
|
self.assertIn("Title hint from UI: Install the PWA correctly", message)
|
|
self.assertIn("Description hint from UI: Prefer something with screenshots.", message)
|
|
self.assertIn("Due hint from UI: 2026-04-12", message)
|
|
self.assertIn("Tag hint from UI: #pwa, #research", message)
|
|
self.assertIn("Preferred destination lane: committed", message)
|
|
|
|
kwargs = captured["kwargs"]
|
|
self.assertEqual(kwargs["chat_id"], "inbox-groomer")
|
|
self.assertEqual(kwargs["timeout_seconds"], 90.0)
|
|
self.assertEqual(kwargs["metadata"]["source"], "web-inbox-card")
|
|
self.assertEqual(kwargs["metadata"]["ui_action"], "accept_task")
|
|
self.assertEqual(kwargs["metadata"]["inbox_item"], self.item["path"])
|
|
self.assertIs(kwargs["notification_to_event"], fake_notification_to_event)
|