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_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_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_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)
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)
