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)
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 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 handle(self, *args, **options):
n = options["nr_messages"]
MessageFactory.create_batch(n)
