def test_failure_no_token(self, mCreateSession): iSession = MockSession() mCreateSession.return_value = (iSession, '123456') task = TaskMixin() with self.assertRaises(Exception): task.failure(None, None) self.assertEqual(task.token, None)
def test_failure_timeout(self, mCreateSession): iSession = MockSession() mCreateSession.return_value = (iSession, '123456') client = iSession.client('stepfunctions') client.send_task_failure.side_effect = TimeoutError() task = TaskMixin() task.token = 'token' task.failure(None, None) self.assertEqual(task.token, None)
def test_failure(self, mCreateSession): iSession = MockSession() client = iSession.client('stepfunctions') mCreateSession.return_value = (iSession, '123456') task = TaskMixin() task.token = 'token' self.assertEqual(task.token, 'token') task.failure(None, None) self.assertEqual(task.token, None) call = mock.call.send_task_failure(taskToken='token', error=None, cause=None) self.assertEqual(client.mock_calls, [call])