def setUpClass(cls, *mocks):
get_api_spec.cache_clear()
generate_classes.cache_clear()
setup_vcr = vcr.VCR(filter_headers=['Authorization'])
setup_cassette = os.path.join(cassette_dir(), 'io_setup.yml')
with setup_vcr.use_cassette(setup_cassette):
# create a file
buf = StringIO()
buf.write('a,b,c\n1,2,3')
buf.seek(0)
file_id = civis.io.file_to_civis(buf, 'somename')
cls.file_id = file_id
# create the table. assumes this function works.
sql = """
DROP TABLE IF EXISTS scratch.api_client_test_fixture;
CREATE TABLE scratch.api_client_test_fixture (
a int,
b int,
c int
);
INSERT INTO scratch.api_client_test_fixture
VALUES (1,2,3);
"""
res = civis.io.query_civis(sql,
'redshift-general',
polling_interval=POLL_INTERVAL)
res.result() # block
# create an export to check get_url. also tests export_csv
with TemporaryDirectory() as temp_dir:
fname = os.path.join(temp_dir, 'tempfile')
sql = "SELECT * FROM scratch.api_client_test_fixture"
database = 'redshift-general'
result = civis.io.civis_to_csv(fname,
sql,
database,
polling_interval=POLL_INTERVAL)
result = result.result()
cls.export_url = result['output'][0]['path']
assert result.state == 'succeeded'
cls.export_job_id = result.sql_id
def tearDownClass(cls):
get_api_spec.cache_clear()
generate_classes.cache_clear()
def setUpClass(cls):
get_api_spec.cache_clear()
generate_classes.cache_clear()
def clear_lru_cache():
# LRU cache persists between tests so these caches need to be cleared
# when different api specs are used in different test cases
get_api_spec.cache_clear()
generate_classes.cache_clear()
def tearDownClass(cls):
get_swagger_spec.cache_clear()
generate_classes.cache_clear()
def setUpClass(cls):
get_swagger_spec.cache_clear()
generate_classes.cache_clear()
