def test_execute_and_wait(self):
query = Notebook()
query.execute = Mock(return_value={'history_uuid': 1, 'status': 0})
query.check_status = Mock(side_effect=check_status_side_effect)
request = Mock()
resp = query.execute_and_wait(request=request)
assert_equal(1, resp['history_uuid'])
assert_equal(2, query.check_status.call_count)
def test_execute_and_wait(self):
query = Notebook()
query.execute = Mock(return_value={'history_uuid': 1, 'status': 0})
query.check_status = Mock(side_effect=check_status_side_effect)
query.fetch_result_data = Mock(return_value={'results': [[1], [2]]})
request = Mock()
resp = query.execute_and_wait(request=request, include_results=True)
assert_equal(0, resp.get('status'), resp)
assert_equal('available', resp['query_status']['status'], resp)
assert_equal([[1], [2]], resp.get('results'), resp)
assert_equal(2, query.check_status.call_count)
def test_check_status(self):
query = Notebook()
request = Mock()
operation_id = Mock()
with patch(
'notebook.api.Document2.objects.get_by_uuid') as get_by_uuid:
with patch('notebook.api.get_api') as get_api:
with patch('notebook.api.Notebook') as NotebookMock:
get_api.return_value = Mock(check_status=Mock(
return_value={'status': 0}))
resp = query.check_status(request=request,
operation_id=operation_id)
assert_equal(0, resp['status'])
assert_equal(0, resp['query_status']['status'])