def test_messages(self):
messages = get_messages()
self.assertIn('TaskFailureNotification', messages)
self.assertIsInstance(messages['TaskFailureNotification'],
TemplateNotificationMessage)
self.assertIn('TaskRescheduleFailed', messages)
self.assertIsInstance(messages['TaskRescheduleFailed'],
TemplateNotificationMessage)
def test_messages(self):
messages = get_messages()
self.assertIn('TaskFailureNotification', messages)
self.assertIsInstance(messages['TaskFailureNotification'],
TemplateNotificationMessage)
self.assertIn('TaskRescheduleFailed', messages)
self.assertIsInstance(messages['TaskRescheduleFailed'],
TemplateNotificationMessage)
def test_notify_task_reschedule_failed(self, m):
task = Backup()
h = NotificationHandler()
h.notify_task_reschedule_failed(Backup())
result = ('Task Reschedule Failed',
get_messages()['TaskRescheduleFailed'].get_message(
{'task': task}))
self.assertTupleEqual(result, m.call_args[0])
def test_notify_task_reschedule_failed(self, m):
task = Backup()
h = NotificationHandler()
h.notify_task_reschedule_failed(Backup())
result = (
'Task Reschedule Failed',
get_messages()['TaskRescheduleFailed'].get_message({
'task': task})
)
self.assertTupleEqual(result, m.call_args[0])
def test_notifiy_on_task_failure(self, m):
task = Backup()
exception = Exception('blah')
h = NotificationHandler()
h.notify_on_task_failure(task, exception, None)
result = (
'Task failed',
get_messages()['TaskFailureNotification'].get_message({
'id': task.id,
'task': task,
'exception': exception,
'trace': None})
)
self.assertTupleEqual(result, m.call_args[0])
def test_notifiy_on_task_failure(self, m):
task = Backup()
exception = Exception('blah')
h = NotificationHandler()
h.notify_on_task_failure(task, exception, None)
result = ('Task failed',
get_messages()['TaskFailureNotification'].get_message({
'id':
task.id,
'task':
task,
'exception':
exception,
'trace':
None
}))
self.assertTupleEqual(result, m.call_args[0])
