def test_get_meesages(self):
self.assertFalse(
Message.objects.filter(sender=self.user1,
reciever=self.user2).exists())
messages_1to2 = MessageFactory.create_batch(3,
sender=self.user1,
reciever=self.user2)
messages_2to1 = MessageFactory.create_batch(5,
sender=self.user1,
reciever=self.user2)
messages_1to2_2 = MessageFactory.create_batch(2,
sender=self.user1,
reciever=self.user2)
response_user2 = self.user2_client.get(self.user1_path,
data=self.example_data)
response_user1 = self.user1_client.get(self.user2_path,
data=self.example_data)
self.assertEqual(response_user1.data["results"],
response_user2.data["results"])
self.assertEqual(response_user1.data["count"], 10)
result_ids = [x["id"] for x in response_user1.data["results"]]
expected_ids = [
x.id for x in messages_1to2 + messages_2to1 + messages_1to2_2
]
expected_ids.reverse()
self.assertEqual(expected_ids, result_ids)
class TestMessageCategories(unittest.TestCase):
def setUp(self):
self.message = MessageFactory(user=None)
self.category = CategoryFactory()
def tearDown(self):
self.message.delete()
self.category.delete()
def test_add_category(self):
before = len(self.message.category.all())
self.message.category.add(self.category)
self.assertEqual(len(self.message.category.all()), before + 1)
def test_added_category_stored(self):
self.message.category.add(self.category)
m = Message.objects.get(id=self.message.pk)
self.assertIn(self.category, m.category.all())
def test_added_category_twice(self):
before = len(self.message.category.all())
self.message.category.add(self.category)
self.message.category.add(self.category)
self.assertEqual(len(self.message.category.all()), before + 1)
def test_remove_category(self):
self.message.category.add(self.category)
before = self.message.category.count()
self.message.category.remove(self.category)
self.assertNotIn(self.category, self.message.category.all())
self.assertEqual(self.message.category.count(), before - 1)
def handle(self, *args, **options):
if environ['DJANGO_SETTINGS_MODULE'] != "olaf.settings.dev":
raise CommandError('This command can only be run in the development environment')
user = User.objects.get(id=options['user_id'])
phone_numbers = PhoneNumber.objects.filter(contact__user=user)
# messages sent by the contacts of our user
for phone_number in phone_numbers:
MessageFactory.create_batch(random.randint(5, 25), sender=phone_number, user=user)
print("Created messages for all contacts of user %d" % options['user_id'])
def test_blocked_user_meesages(self):
self.assertFalse(
Message.objects.filter(sender=self.user1,
reciever=self.user2).exists())
messages_not_blocked = MessageFactory(sender=self.user1,
reciever=self.user2,
is_blocked=False)
MessageFactory(sender=self.user1, reciever=self.user2, is_blocked=True)
response = self.user1_client.get(self.user2_path,
data=self.example_data)
self.assertEqual(response.data["count"], 1)
self.assertEqual(response.data["results"][0]["id"],
messages_not_blocked.id)
class TestMessageLocation(unittest.TestCase):
def setUp(self):
self.message = MessageFactory()
def tearDown(self):
user = self.message.user
self.message.delete()
user.delete()
def test_location_latitude(self):
self.assertIsNotNone(self.message.location.x)
def test_location_longitude(self):
self.assertIsNotNone(self.message.location.y)
def test_location(self):
self.assertIsNotNone(self.message.location)
def handle(self, *args, **options):
if environ['DJANGO_SETTINGS_MODULE'] != "olaf.settings.dev":
raise CommandError(
'This command can only be run in the development environment')
user = User.objects.get(id=options['user_id'])
phone_numbers = PhoneNumber.objects.filter(contact__user=user)
# messages sent by the contacts of our user
for phone_number in phone_numbers:
MessageFactory.create_batch(random.randint(5, 25),
sender=phone_number,
user=user)
print("Created messages for all contacts of user %d" %
options['user_id'])
def test_get_meesages_read_datetime_set(self):
message = MessageFactory(sender=self.user1, reciever=self.user2)
self.assertEqual(message.read_datetime, None)
self.user1_client.get(self.user2_path, data=self.example_data)
message = Message.objects.get(id=message.id)
self.assertEqual(message.read_datetime, None)
self.user2_client.get(self.user1_path, data=self.example_data)
message = Message.objects.get(id=message.id)
self.assertNotEqual(message.read_datetime, None)
def handle(self, *args, **options):
n = options["nr_messages"]
MessageFactory.create_batch(n)
def setUp(self):
self.user = UserFactory()
self.region = RegionFactory()
self.message = MessageFactory.build(georegion=self.region)
def setUp(self):
self.message = MessageFactory(user=None)
self.category = CategoryFactory()
def setUp(self):
self.message = MessageFactory()
