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)
