def test_task_name(self, task_mock):
"""Ensure the resultant task name contains enough to be unique."""
from sosbeacon.event.message import get_student_broadcast_task
student_key = Mock()
student_key.urlsafe.return_value = "STUDENTKEY"
event_key = Mock()
event_key.urlsafe.return_value = "EVENTKEY"
message_key = Mock()
message_key.urlsafe.return_value = "MESSAGEKEY"
batch_id = "BATCHID"
get_student_broadcast_task(student_key, event_key, message_key, batch_id)
task_name = task_mock.call_args[1]["name"]
self.assertIn("STUDENTKEY", task_name)
self.assertNotIn("EVENTKEY", task_name)
self.assertIn("MESSAGEKEY", task_name)
self.assertIn("BATCHID", task_name)
def test_task_params(self, task_mock):
"""Ensure the resultant task parms contain all info."""
from sosbeacon.event.message import get_student_broadcast_task
student_key = Mock()
student_key.urlsafe.return_value = "ASTUDENTKEY"
event_key = Mock()
event_key.urlsafe.return_value = "ANEVENTKEY"
message_key = Mock()
message_key.urlsafe.return_value = "SOMEMESSAGEKEY"
batch_id = "THEBATCHID"
get_student_broadcast_task(student_key, event_key, message_key, batch_id)
check_params = {
"student": "ASTUDENTKEY",
"event": "ANEVENTKEY",
"message": "SOMEMESSAGEKEY",
"batch": "THEBATCHID",
}
self.assertEqual(check_params, task_mock.call_args[1]["params"])