def test_task_params(self, task_mock): from sosbeacon.event.message import get_broadcast_method_to_user_task event_key = Mock() event_key.urlsafe.return_value = "EVENT" message_key = Mock() message_key.urlsafe.return_value = "MESSAGE" user_key = Mock() user_key.urlsafe.return_value = "USER" method = "*****@*****.**" get_broadcast_method_to_user_task(event_key, message_key, user_key, method) check_params = {"user": "******", "event": "EVENT", "message": "MESSAGE", "method": "*****@*****.**"} self.assertEqual(check_params, task_mock.call_args[1]["params"])
def test_task_name(self, task_mock): from sosbeacon.event.message import get_broadcast_method_to_user_task event_key = Mock() event_key.urlsafe.return_value = "EVENT" message_key = Mock() message_key.urlsafe.return_value = "MESSAGE" user_key = Mock() user_key.urlsafe.return_value = "USER" method = "*****@*****.**" get_broadcast_method_to_user_task(event_key, message_key, user_key, method) task_name = task_mock.call_args[1]["name"] self.assertIn("USER", task_name) self.assertIn("MESSAGE", task_name) self.assertIn("EVENT", task_name)