109 lines
4.2 KiB
Python
109 lines
4.2 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from core.models import ContactAvailabilitySettings, ChatSession, ConversationEvent, Person, PersonIdentifier, User
|
|
|
|
|
|
class AvailabilitySettingsPageTests(TestCase):
|
|
def setUp(self):
|
|
self.user = User.objects.create_user("avail-user", "avail@example.com", "x")
|
|
self.client.force_login(self.user)
|
|
|
|
def test_get_page_renders(self):
|
|
response = self.client.get(reverse("behavioral_signals_settings"))
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertContains(response, "Behavioral Signals")
|
|
|
|
def test_post_updates_settings(self):
|
|
response = self.client.post(
|
|
reverse("behavioral_signals_settings"),
|
|
{
|
|
"enabled": "1",
|
|
"show_in_chat": "1",
|
|
"show_in_groups": "0",
|
|
"inference_enabled": "1",
|
|
"retention_days": "120",
|
|
"fade_threshold_seconds": "300",
|
|
},
|
|
)
|
|
self.assertEqual(200, response.status_code)
|
|
row = ContactAvailabilitySettings.objects.get(user=self.user)
|
|
self.assertTrue(row.enabled)
|
|
self.assertTrue(row.show_in_chat)
|
|
self.assertFalse(row.show_in_groups)
|
|
self.assertTrue(row.inference_enabled)
|
|
self.assertEqual(120, row.retention_days)
|
|
self.assertEqual(300, row.fade_threshold_seconds)
|
|
|
|
@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")
|
|
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["behavioral_stats"])
|
|
self.assertEqual(1, len(stats))
|
|
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 ""),
|
|
)
|