77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from core.models import (
|
|
ContactAvailabilityEvent,
|
|
ContactAvailabilitySettings,
|
|
Person,
|
|
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("availability_settings"))
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertContains(response, "Availability Settings")
|
|
|
|
def test_post_updates_settings(self):
|
|
response = self.client.post(
|
|
reverse("availability_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)
|
|
|
|
def test_contact_event_stats_are_aggregated(self):
|
|
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"))
|
|
self.assertEqual(200, response.status_code)
|
|
stats = list(response.context["contact_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"])
|