def validate(self) -> None:
"""
Validate that message object contains all the right things.
:raises exceptions.ValidationError: when message fails validation
"""
from taskhawk.task_manager import Task
if (not self.id or not self.version or self.version not in self.VERSIONS or not self.timestamp or
self.headers is None or not self.task_name or self.args is None or self.kwargs is None):
raise ValidationError
try:
self._task = Task.find_by_name(self.task_name)
except TaskNotFound:
raise ValidationError
def test_find_by_name_fail(self):
with pytest.raises(TaskNotFound):
Task.find_by_name('invalid')
def test_find_by_name(self):
assert Task.find_by_name('tests.tasks.send_email') == send_email.task
