151 lines
5.7 KiB
Python
151 lines
5.7 KiB
Python
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from core.models import (
|
|
AIRequest,
|
|
ChatSession,
|
|
ConversationEvent,
|
|
MemoryItem,
|
|
Person,
|
|
PersonIdentifier,
|
|
User,
|
|
WorkspaceConversation,
|
|
)
|
|
|
|
|
|
class SystemDiagnosticsAPITests(TestCase):
|
|
def setUp(self):
|
|
self.user = User.objects.create_superuser(
|
|
username="sys-diag-admin",
|
|
email="sys-diag@example.com",
|
|
password="pw",
|
|
)
|
|
person = Person.objects.create(user=self.user, name="System Diagnostics Person")
|
|
identifier = PersonIdentifier.objects.create(
|
|
user=self.user,
|
|
person=person,
|
|
service="signal",
|
|
identifier="+15554443333",
|
|
)
|
|
self.session = ChatSession.objects.create(user=self.user, identifier=identifier)
|
|
self.workspace_conversation = WorkspaceConversation.objects.create(
|
|
user=self.user,
|
|
platform_type="signal",
|
|
title="Diag Memory Scope",
|
|
platform_thread_id=str(self.session.id),
|
|
)
|
|
self.client.force_login(self.user)
|
|
|
|
def test_event_ledger_smoke_api_returns_counts_and_missing_required(self):
|
|
ConversationEvent.objects.create(
|
|
user=self.user,
|
|
session=self.session,
|
|
ts=1700000000000,
|
|
event_type="message_created",
|
|
direction="in",
|
|
origin_transport="signal",
|
|
payload={"message_id": "m1"},
|
|
raw_payload={},
|
|
)
|
|
response = self.client.get(
|
|
reverse("system_event_ledger_smoke"),
|
|
{
|
|
"minutes": "999999",
|
|
"service": "signal",
|
|
"require_types": "message_created,reaction_added",
|
|
},
|
|
)
|
|
self.assertEqual(200, response.status_code)
|
|
payload = response.json()
|
|
self.assertTrue(payload.get("ok"))
|
|
self.assertEqual("signal", payload.get("service"))
|
|
self.assertIn("event_type_counts", payload)
|
|
self.assertIn("missing_required_types", payload)
|
|
self.assertIn("reaction_added", payload.get("missing_required_types") or [])
|
|
|
|
def test_trace_diagnostics_includes_projection_shadow_links(self):
|
|
trace_id = "trace-system-diag-1"
|
|
event = ConversationEvent.objects.create(
|
|
user=self.user,
|
|
session=self.session,
|
|
ts=1700000001000,
|
|
event_type="message_created",
|
|
direction="in",
|
|
origin_transport="signal",
|
|
trace_id=trace_id,
|
|
payload={"message_id": "m2"},
|
|
raw_payload={},
|
|
)
|
|
response = self.client.get(
|
|
reverse("system_trace_diagnostics"),
|
|
{"trace_id": trace_id},
|
|
)
|
|
self.assertEqual(200, response.status_code)
|
|
payload = response.json()
|
|
self.assertTrue(payload.get("ok"))
|
|
self.assertEqual(1, payload.get("count"))
|
|
self.assertIn(str(self.session.id), payload.get("related_session_ids") or [])
|
|
urls = payload.get("projection_shadow_urls") or []
|
|
self.assertTrue(urls)
|
|
self.assertIn(str(self.session.id), str(urls[0]))
|
|
events = payload.get("events") or []
|
|
self.assertEqual(str(event.id), str(events[0].get("id")))
|
|
self.assertIn(
|
|
str(self.session.id),
|
|
str(events[0].get("projection_shadow_url") or ""),
|
|
)
|
|
|
|
def test_memory_search_status_and_query_api(self):
|
|
request = AIRequest.objects.create(
|
|
user=self.user,
|
|
conversation=self.workspace_conversation,
|
|
window_spec={},
|
|
operation="memory_propose",
|
|
)
|
|
memory = MemoryItem.objects.create(
|
|
user=self.user,
|
|
conversation=self.workspace_conversation,
|
|
memory_kind="fact",
|
|
status="active",
|
|
content={"text": "User prefers concise status updates on WhatsApp."},
|
|
source_request=request,
|
|
)
|
|
status_response = self.client.get(reverse("system_memory_search_status"))
|
|
self.assertEqual(200, status_response.status_code)
|
|
status_payload = status_response.json()
|
|
self.assertTrue(status_payload.get("ok"))
|
|
self.assertIn("status", status_payload)
|
|
|
|
query_response = self.client.get(
|
|
reverse("system_memory_search_query"),
|
|
{"q": "concise status updates"},
|
|
)
|
|
self.assertEqual(200, query_response.status_code)
|
|
query_payload = query_response.json()
|
|
self.assertTrue(query_payload.get("ok"))
|
|
self.assertGreaterEqual(int(query_payload.get("count") or 0), 1)
|
|
first_hit = (query_payload.get("hits") or [{}])[0]
|
|
self.assertEqual(str(memory.id), str(first_hit.get("memory_id") or ""))
|
|
|
|
def test_system_settings_page_renders_searchable_datalists(self):
|
|
ConversationEvent.objects.create(
|
|
user=self.user,
|
|
session=self.session,
|
|
ts=1700000002000,
|
|
event_type="reaction_added",
|
|
direction="system",
|
|
origin_transport="signal",
|
|
trace_id="trace-system-diag-2",
|
|
payload={"message_id": "m3"},
|
|
raw_payload={},
|
|
)
|
|
response = self.client.get(reverse("system_settings"))
|
|
self.assertEqual(200, response.status_code)
|
|
content = response.content.decode("utf-8")
|
|
self.assertIn('datalist id="diagnostics-session-options"', content)
|
|
self.assertIn('datalist id="diagnostics-trace-options"', content)
|
|
self.assertIn('datalist id="diagnostics-service-options"', content)
|
|
self.assertIn('datalist id="diagnostics-event-type-options"', content)
|
|
self.assertIn(str(self.session.id), content)
|
|
self.assertIn("trace-system-diag-2", content)
|