def test_reminder_numbers(self): # add one number to the whitelist WhitelistFactory(phone_number="9871") # reminder_number should be in the message reminder_number = 38 message_reminder_task(2, reminder_number, 'whitelist', self.election) batch = Batch.objects.get() self.assertIn(str(reminder_number), batch.messages.get().message)
def test_empty(self): # With valid arguments, should at least create a batch self.assertEqual(0, Batch.objects.count()) message_reminder_task(1, 1, 'whitelist', self.election) batch = Batch.objects.get() # no messages self.assertEqual(0, batch.messages.count()) self.assertEqual(Batch.APPROVED, batch.status) self.assertEqual(Batch.PRIORITY_TIME_CRITICAL, batch.priority)
def test_with_numbers(self, mock_get_numbers): shortcode = None number_list = [("1", "message1", shortcode), ("22", "message2", shortcode)] mock_get_numbers.return_value = number_list self.assertEqual(0, Batch.objects.count()) message_reminder_task(1, 1, 'whitelist', self.election) batch = Batch.objects.get() # 2 messages self.assertEqual(2, batch.messages.count()) self.assertEqual(Batch.APPROVED, batch.status) self.assertEqual(Batch.PRIORITY_TIME_CRITICAL, batch.priority)
def test_reminder_text(self): # add one number to the whitelist and the registered list WhitelistFactory(phone_number="9871") StaffPhoneFactory(phone_number='9871') # message_numbers 1, 2, 3 should return message REMINDER_CHECKIN (code=54) message_reminder_task(1, 1, 'whitelist', self.election) message_reminder_task(2, 1, 'whitelist', self.election) message_reminder_task(3, 1, 'whitelist', self.election) batches = Batch.objects.all() for batch in batches: # assert that proper message_code is in the outgoing text self.assertIn(str(constants.REMINDER_CHECKIN), batch.messages.get().message) Batch.objects.all().delete() # message_numbers 4 and 5 should return message REMINDER_REPORT (code=55) message_reminder_task(4, 1, 'registered', self.election) message_reminder_task(5, 1, 'registered', self.election) batches = Batch.objects.all() for batch in batches: self.assertIn(str(constants.REMINDER_REPORT), batch.messages.get().message) Batch.objects.all().delete() # message_number 6 should return message REMINDER_LAST_REPORT (code=56) message_reminder_task(6, 1, 'registered', self.election) batch = Batch.objects.get() self.assertIn(str(constants.REMINDER_LAST_REPORT), batch.messages.get().message) Batch.objects.all().delete() # message_number 7 should return message REMINDER_CLOSE (code=57) message_reminder_task(7, 1, 'registered', self.election) batch = Batch.objects.get() self.assertIn(str(constants.REMINDER_CLOSE), batch.messages.get().message)