Implement Manticore fully and re-theme

core/tests/test_availability_settings_page.py

@@ -1,14 +1,11 @@
 from __future__ import annotations
 
+from unittest.mock import patch
+
 from django.test import TestCase
 from django.urls import reverse
 
-from core.models import (
-    ContactAvailabilityEvent,
-    ContactAvailabilitySettings,
-    Person,
-    User,
-)
+from core.models import ContactAvailabilitySettings, ChatSession, ConversationEvent, Person, PersonIdentifier, User
 
 
 class AvailabilitySettingsPageTests(TestCase):
@@ -17,13 +14,13 @@ class AvailabilitySettingsPageTests(TestCase):
         self.client.force_login(self.user)
 
     def test_get_page_renders(self):
-        response = self.client.get(reverse("availability_settings"))
+        response = self.client.get(reverse("behavioral_signals_settings"))
         self.assertEqual(200, response.status_code)
-        self.assertContains(response, "Availability Settings")
+        self.assertContains(response, "Behavioral Signals")
 
     def test_post_updates_settings(self):
         response = self.client.post(
-            reverse("availability_settings"),
+            reverse("behavioral_signals_settings"),
             {
                 "enabled": "1",
                 "show_in_chat": "1",
@@ -42,35 +39,70 @@ class AvailabilitySettingsPageTests(TestCase):
         self.assertEqual(120, row.retention_days)
         self.assertEqual(300, row.fade_threshold_seconds)
 
-    def test_contact_event_stats_are_aggregated(self):
+    @patch("core.views.availability.get_behavioral_availability_stats")
+    def test_behavioral_manticore_stats_are_in_context(self, mocked_stats):
         person = Person.objects.create(user=self.user, name="Alice")
-        ContactAvailabilityEvent.objects.create(
-            user=self.user,
-            person=person,
-            service="whatsapp",
-            source_kind="message_in",
-            availability_state="available",
-            confidence=0.9,
-            ts=1000,
-            payload={},
-        )
-        ContactAvailabilityEvent.objects.create(
-            user=self.user,
-            person=person,
-            service="whatsapp",
-            source_kind="inferred_timeout",
-            availability_state="fading",
-            confidence=0.5,
-            ts=2000,
-            payload={},
-        )
-        response = self.client.get(reverse("availability_settings"))
+        mocked_stats.return_value = [
+            {
+                "person_id": str(person.id),
+                "transport": "whatsapp",
+                "total_events": 9,
+                "presence_events": 2,
+                "read_events": 3,
+                "typing_events": 2,
+                "message_events": 1,
+                "abandoned_events": 1,
+                "last_event_ts": 5555,
+            }
+        ]
+
+        response = self.client.get(reverse("behavioral_signals_settings"))
+
         self.assertEqual(200, response.status_code)
-        stats = list(response.context["contact_stats"])
+        stats = list(response.context["behavioral_stats"])
         self.assertEqual(1, len(stats))
-        self.assertEqual("Alice", stats[0]["person__name"])
-        self.assertEqual(2, stats[0]["total_events"])
-        self.assertEqual(1, stats[0]["available_events"])
-        self.assertEqual(1, stats[0]["fading_events"])
-        self.assertEqual(1, stats[0]["message_activity_events"])
-        self.assertEqual(1, stats[0]["inferred_timeout_events"])
+        self.assertEqual("Alice", stats[0]["person_name"])
+        self.assertEqual(1, stats[0]["abandoned_events"])
+        self.assertEqual("manticore", response.context["behavioral_stats_source"])
+        self.assertContains(response, "Behavioral Event Statistics")
+        self.assertNotIn("contact_stats", response.context)
+        self.assertNotIn("parity_rows", response.context)
+
+    @patch("core.views.availability.get_behavioral_availability_stats")
+    def test_behavioral_stats_fallback_to_conversation_event_shadow(self, mocked_stats):
+        person = Person.objects.create(user=self.user, name="Alice")
+        identifier = PersonIdentifier.objects.create(
+            user=self.user,
+            person=person,
+            service="signal",
+            identifier="+15551230000",
+        )
+        session = ChatSession.objects.create(user=self.user, identifier=identifier)
+        ConversationEvent.objects.create(
+            user=self.user,
+            session=session,
+            ts=1234,
+            event_type="presence_available",
+            direction="system",
+            origin_transport="signal",
+        )
+        mocked_stats.return_value = []
+
+        response = self.client.get(reverse("behavioral_signals_settings"))
+
+        self.assertEqual(200, response.status_code)
+        stats = list(response.context["behavioral_stats"])
+        self.assertEqual(1, len(stats))
+        self.assertEqual("Alice", stats[0]["person_name"])
+        self.assertEqual(1, int(stats[0]["presence_events"]))
+        self.assertEqual(
+            "conversation_event_shadow", response.context["behavioral_stats_source"]
+        )
+
+    def test_legacy_availability_url_redirects(self):
+        response = self.client.get(reverse("availability_settings"))
+        self.assertEqual(302, response.status_code)
+        self.assertIn(
+            reverse("behavioral_signals_settings"),
+            str(response.headers.get("Location") or ""),
+        )