def test_ingest_4():
"""Functional Test: Test table replacement"""
tablename = str(uuid4())
with app.test_client() as client:
res = client.post('/ingest',
data=dict(resource=(open(shapefile,
'rb'), 'geo.zip'),
response='prompt',
tablename=tablename))
assert res.status_code == 200
sleep(1.0)
with app.test_client() as client:
res = client.post('/ingest',
data=dict(resource=(open(shapefile,
'rb'), 'geo.zip'),
response='prompt',
tablename=tablename))
assert res.status_code == 500
sleep(1.0)
with app.test_client() as client:
res = client.post('/ingest',
data=dict(resource=(open(shapefile,
'rb'), 'geo.zip'),
response='prompt',
tablename=tablename,
replace='true'))
assert res.status_code == 200
def test_ingest_3():
"""Functional Test: Test complete ingest functionality"""
idempotency_key = str(uuid4())
with app.test_client() as client:
res = client.post('/ingest',
data=dict(resource=(open(shapefile,
'rb'), 'geo.zip'),
response='deferred'),
headers={'X-Idempotency-Key': idempotency_key},
content_type='multipart/form-data')
assert res.status_code == 202
r = res.get_json()
ticket = r.get('ticket')
endpoint = r.get('status')
assert ticket is not None
assert endpoint is not None
assert r.get('type') == 'deferred'
sleep(1.0)
with app.test_client() as client:
res = client.get(endpoint)
assert res.status_code == 200
r = res.get_json()
assert r.get('comment') is None
assert r.get('completed') is not None
assert r.get('executionTime') is not None
assert r.get('requested') is not None
assert r.get('success') is not None
res = client.get('/ticket_by_key/%s' % (idempotency_key))
assert res.status_code == 200
r = res.get_json()
assert r.get('request') == 'ingest'
assert r.get('ticket') == ticket
def test_ingest_5():
"""Functional Test: Test drop table"""
tablename = str(uuid4())
with app.test_client() as client:
res = client.post('/ingest',
data=dict(resource=(open(shapefile,
'rb'), 'geo.zip'),
response='prompt',
tablename=tablename))
assert res.status_code == 200
sleep(1.0)
with app.test_client() as client:
res = client.delete(f'/ingest/{tablename}')
assert res.status_code == 204
def test_ingest_1():
"""Functional Test: Test KML ingest"""
with app.test_client() as client:
res = client.post('/ingest', data=dict(resource='geo.kml'))
assert res.status_code == 200
r = res.get_json()
assert r.get('schema') is not None
assert r.get('table') is not None
assert r.get('length') == 3
def test_ingest_2():
"""Functional Test: Test Shapefile ingest; streaming resource"""
with app.test_client() as client:
res = client.post('/ingest', data=dict(resource='geo.zip'))
assert res.status_code == 200
r = res.get_json()
assert r.get('schema') is not None
assert r.get('table') is not None
assert r.get('length') == 3
def test_get_health_check():
with app.test_client() as client:
res = client.get('/_health', query_string=dict(), headers=dict())
assert res.status_code == 200
r = res.get_json()
if 'reason' in r:
logging.error('The service is unhealthy: %(reason)s\n%(detail)s',
r)
logging.debug("From /_health: %s" % (r))
assert r['status'] == 'OK'
def test_get_documentation_1():
with app.test_client() as client:
res = client.get('/', query_string=dict(), headers=dict())
assert res.status_code == 200
r = res.get_json()
assert not (r.get('openapi') is None)