def check_database(corpus_name, token=load_token(), port=8080):
host = 'http://localhost:{}'.format(port)
client = PGDBClient(host, token)
try:
client.start_database(corpus_name)
except Exception as e:
print("Database problem: {}".format(e))
def test_client_delete_corpus(localhost, auth_token):
client = PGDBClient(localhost, token=auth_token)
client.start_database('test_database')
assert 'test' in [x['name'] for x in client.list_corpora('test_database')]
client.delete_corpus('test')
assert 'test' not in [x['name'] for x in client.list_corpora('test_database')]
client.stop_database('test_database')
def test_client_delete_corpus(localhost):
client = PGDBClient(localhost)
client.start_database('test_database')
assert 'test' in client.list_corpora('test_database')
client.delete_corpus('test')
assert 'test' not in client.list_corpora('test_database')
client.stop_database('test_database')
def test_client_import(localhost, auth_token):
client = PGDBClient(localhost, token=auth_token)
with pytest.raises(ClientError):
client.import_corpus('test', 'acoustic', 'M', 'test_database')
client.start_database('test_database')
client.import_corpus('test', 'acoustic', 'M', 'test_database', blocking=True)
client.stop_database('test_database')
assert client.corpus_status('test')['status'] == 'Imported'
assert 'test' in [x['name'] for x in client.list_corpora('test_database')]
def test_query_basic(localhost, auth_token):
client = PGDBClient(localhost, token=auth_token)
client.start_database('test_database')
hierarchy = client.hierarchy('test')
q = client.generate_query(hierarchy.phone)
q.filter(hierarchy.phone.label == 'aa')
q.columns(hierarchy.phone.label.column_name('phone_name'))
results = client.run_query(q, blocking=True)
assert len(results) > 0
assert all(x['phone_name'] == 'aa' for x in results)
client.stop_database('test_database')
def graph_db(localhost):
from polyglotdb.client.client import PGDBClient, ClientError
client = PGDBClient(localhost)
dbs = client.list_databases()
for d in dbs:
if 'test' in d:
client.delete_database(d)
client.create_database('main_test_database')
ports = client.get_ports('main_test_database')
ports['data_dir'] = client.get_directory('main_test_database')
ports['host'] = 'localhost'
client.start_database('main_test_database')
return ports
def test_client_import(localhost):
client = PGDBClient(localhost)
with pytest.raises(ClientError):
client.import_corpus('test', 'acoustic', 'M', 'test_database')
client.start_database('test_database')
client.import_corpus('test',
'acoustic',
'M',
'test_database',
blocking=True)
client.stop_database('test_database')
assert client.corpus_status('test') == 'Imported'
assert 'test' in client.list_corpora('test_database')
def graph_db(localhost, auth_token):
from polyglotdb.client.client import PGDBClient, ClientError
client = PGDBClient(localhost, token=auth_token)
dbs = client.list_databases()
print(dbs)
for d in dbs:
if d['name'] == 'main_test_database':
client.delete_database(d['name'])
client.create_database('main_test_database')
ports = client.get_ports('main_test_database')
ports['data_dir'] = client.get_directory('main_test_database')
ports['host'] = 'localhost'
client.start_database('main_test_database')
return ports
