Increase security and reformat
This commit is contained in:
@@ -6,9 +6,9 @@ from django.test import TestCase
|
||||
from core.models import (
|
||||
ChatSession,
|
||||
ContactAvailabilityEvent,
|
||||
Message,
|
||||
Person,
|
||||
PersonIdentifier,
|
||||
Message,
|
||||
User,
|
||||
)
|
||||
from core.presence.inference import now_ms
|
||||
@@ -16,7 +16,9 @@ from core.presence.inference import now_ms
|
||||
|
||||
class BackfillContactAvailabilityCommandTests(TestCase):
|
||||
def setUp(self):
|
||||
self.user = User.objects.create_user("backfill-user", "backfill@example.com", "x")
|
||||
self.user = User.objects.create_user(
|
||||
"backfill-user", "backfill@example.com", "x"
|
||||
)
|
||||
self.person = Person.objects.create(user=self.user, name="Backfill Person")
|
||||
self.identifier = PersonIdentifier.objects.create(
|
||||
user=self.user,
|
||||
@@ -24,7 +26,9 @@ class BackfillContactAvailabilityCommandTests(TestCase):
|
||||
service="signal",
|
||||
identifier="+15551234567",
|
||||
)
|
||||
self.session = ChatSession.objects.create(user=self.user, identifier=self.identifier)
|
||||
self.session = ChatSession.objects.create(
|
||||
user=self.user, identifier=self.identifier
|
||||
)
|
||||
|
||||
def test_backfill_creates_message_and_read_receipt_availability_events(self):
|
||||
base_ts = now_ms()
|
||||
@@ -58,7 +62,9 @@ class BackfillContactAvailabilityCommandTests(TestCase):
|
||||
)
|
||||
|
||||
events = list(
|
||||
ContactAvailabilityEvent.objects.filter(user=self.user).order_by("ts", "source_kind")
|
||||
ContactAvailabilityEvent.objects.filter(user=self.user).order_by(
|
||||
"ts", "source_kind"
|
||||
)
|
||||
)
|
||||
self.assertEqual(3, len(events))
|
||||
self.assertTrue(any(row.source_kind == "message_in" for row in events))
|
||||
|
||||
Reference in New Issue
Block a user