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)