def test_get_tables():
# default parameters
table_set = PropertyMock()
table_set.keys.return_value = ['table1', 'table2']
table_set.values.return_value = ['tab1val', 'tab2val', 'tab3val']
with patch('astroquery.cadc.core.pyvo.dal.TAPService', autospec=True) as tapservice_mock:
tapservice_mock.return_value.tables = table_set
cadc = Cadc()
assert len(cadc.get_tables(only_names=True)) == 2
assert len(cadc.get_tables()) == 3
def test_list_tables(self):
cadc = Cadc()
table_names = cadc.get_tables(only_names=True)
assert len(table_names) > 20
assert 'caom2.Observation' in table_names
assert 'ivoa.ObsCore' in table_names
assert 'tap_schema.tables' in table_names
tables = cadc.get_tables()
assert len(table_names) == len(tables)
for table in tables:
assert table.name in table_names
table = cadc.get_table('caom2.Observation')
assert 'caom2.Observation' == table.name
def test_get_tables(monkeypatch):
dummyTapHandler = DummyTapHandler()
monkeypatch.setattr(cadc_core, 'get_access_url', get_access_url_mock)
tap = Cadc(tap_plus_handler=dummyTapHandler)
# default parameters
parameters = {}
parameters['only_names'] = False
parameters['verbose'] = False
tap.get_tables()
dummyTapHandler.check_call('get_tables', parameters)
# test with parameters
dummyTapHandler.reset()
parameters = {}
parameters['only_names'] = True
parameters['verbose'] = True
tap.get_tables(True, True)
dummyTapHandler.check_call('get_tables', parameters)
