def test_task_check(ts): api = MyAPI() api._check('asdf') print(ts.mock_calls) assert ts.call_count == 1 ts.return_value.check_task.assert_called_once_with('asdf')
def test_index_page(): api = MyAPI() api._start = mock.Mock() api._check = mock.Mock() api.index() api._start.assert_called_once_with() api.index('asdf') api._check.assert_called_once_with('asdf')
def test_task_creation(ts): api = MyAPI() api._start() assert ts.call_count == 1 assert ts.return_value.post_task.call_count == 1 first_args = ts.return_value.post_task.call_args ts.reset_mock() api._start() assert ts.call_count == 1 assert ts.return_value.post_task.call_count == 1 second_args = ts.return_value.post_task.call_args assert first_args != second_args
def test_cancel_page(ts): api = MyAPI() api.cancel('asdf') assert ts.call_count == 1 ts.return_value.cancel_task.assert_called_once_with('asdf')