def test_heartbeat_no_token(self, mCreateSession): iSession = MockSession() mCreateSession.return_value = (iSession, '123456') task = TaskMixin() with self.assertRaises(Exception): task.heartbeat()
def test_heartbeat(self, mCreateSession): iSession = MockSession() client = iSession.client('stepfunctions') mCreateSession.return_value = (iSession, '123456') task = TaskMixin() task.token = 'token' task.heartbeat() call = mock.call.send_task_heartbeat(taskToken='token') self.assertEqual(client.mock_calls, [call])