def test_async(self):
# test async calls
cadc = Cadc()
# run the query in sync mode first
expected = cadc.exec_sync(
"select top 3 observationID from caom2.Observation where "
"collection='IRIS' order by observationID")
# now run the query in async mode
job = cadc.create_async(
"select top 3 observationID from caom2.Observation where "
"collection='IRIS' order by observationID")
job = job.run().wait()
job.raise_if_error()
result = job.fetch_result().to_table()
assert len(expected) == len(result)
for ii in range(0, 2):
assert expected['observationID'][ii] == result['observationID'][ii]
# load job again
loaded_job = cadc.load_async_job(job.job_id)
result = loaded_job.fetch_result().to_table()
assert len(expected) == len(result)
for ii in range(0, 2):
assert expected['observationID'][ii] == result['observationID'][ii]
def test_load_async_job(monkeypatch):
dummyTapHandler = DummyTapHandler()
monkeypatch.setattr(cadc_core, 'get_access_url', get_access_url_mock)
tap = Cadc(tap_plus_handler=dummyTapHandler)
jobid = '123'
# default parameters
parameters = {}
parameters['jobid'] = jobid
parameters['verbose'] = False
tap.load_async_job(jobid)
dummyTapHandler.check_call('load_async_job', parameters)
# test with parameters
dummyTapHandler.reset()
parameters['jobid'] = jobid
parameters['verbose'] = True
tap.load_async_job(jobid, verbose=True)
dummyTapHandler.check_call('load_async_job', parameters)
def test_load_async_job():
with patch('astroquery.cadc.core.pyvo.dal.TAPService', autospec=True) as tapservice_mock:
with patch('astroquery.cadc.core.pyvo.dal.AsyncTAPJob',
autospec=True) as tapjob_mock:
tapservice_mock.return_value.baseurl.return_value = 'https://www.example.com/tap'
mock_job = Mock()
mock_job.job_id = '123'
tapjob_mock.return_value = mock_job
cadc = Cadc()
jobid = '123'
job = cadc.load_async_job(jobid)
assert job.job_id == '123'
