def test_get_results__no_results(self, download_mock):
context = mock.Mock()
context.bulk.endpoint = "https://test"
context.bulk.create_query_job.return_value = "JOB"
context.bulk.query.return_value = "BATCH"
context.bulk.get_query_batch_result_ids.return_value = ["RESULT"]
download_mock.return_value = io.StringIO(
"Records not found for this query")
query = BulkApiQueryOperation(
sobject="Contact",
api_options={},
context=context,
query="SELECT Id FROM Contact",
)
query._wait_for_job = mock.Mock()
query._wait_for_job.return_value = DataOperationJobResult(
DataOperationStatus.SUCCESS, [], 0, 0)
query.query()
results = list(query.get_results())
context.bulk.get_query_batch_result_ids.assert_called_once_with(
"BATCH", job_id="JOB")
download_mock.assert_called_once_with(
"https://test/job/JOB/batch/BATCH/result/RESULT", context.bulk)
assert list(results) == []
def test_query(self):
context = mock.Mock()
query = BulkApiQueryOperation(
sobject="Contact",
api_options={},
context=context,
query="SELECT Id FROM Contact",
)
query._wait_for_job = mock.Mock()
query._wait_for_job.return_value = DataOperationJobResult(
DataOperationStatus.SUCCESS, [], 0, 0)
query.query()
assert query.job_result.status is DataOperationStatus.SUCCESS
context.bulk.create_query_job.assert_called_once_with(
"Contact", contentType="CSV")
context.bulk.query.assert_called_once_with(
context.bulk.create_query_job.return_value,
"SELECT Id FROM Contact")
query._wait_for_job.assert_called_once_with(
context.bulk.create_query_job.return_value)
context.bulk.close_job.assert_called_once_with(
context.bulk.create_query_job.return_value)
def test_query__contextmanager(self):
context = mock.Mock()
query = BulkApiQueryOperation(
sobject="Contact",
api_options={},
context=context,
query="SELECT Id FROM Contact",
)
query._wait_for_job = mock.Mock()
query._wait_for_job.return_value = DataOperationJobResult(
DataOperationStatus.SUCCESS, [], 0, 0)
with query:
assert query.job_result.status is DataOperationStatus.SUCCESS
def test_query__failure(self):
context = mock.Mock()
query = BulkApiQueryOperation(
sobject="Contact",
api_options={},
context=context,
query="SELECT Id FROM Contact",
)
query._wait_for_job = mock.Mock()
query._wait_for_job.return_value = DataOperationJobResult(
DataOperationStatus.JOB_FAILURE, [], 0, 0)
query.query()
assert query.job_result.status is DataOperationStatus.JOB_FAILURE