From: Javier Sancho Date: Wed, 23 May 2018 09:50:01 +0000 (+0200) Subject: Tests for api homepage and api database page X-Git-Url: https://git.jsancho.org/?a=commitdiff_plain;h=06bdd1ac2d8836a2ab8e14074f2435a4c3a595b5;p=datasette-pytables.git Tests for api homepage and api database page --- diff --git a/setup.py b/setup.py index 466ce4c..04f4253 100644 --- a/setup.py +++ b/setup.py @@ -26,6 +26,7 @@ setup( 'pytables = datasette_pytables' ], }, - install_requires=['datasette', 'tables', 'moz-sql-parser', 'mo-future'] + install_requires=['datasette', 'tables', 'moz-sql-parser', 'mo-future'], + tests_require=['pytest'] ) diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/fixtures.py b/tests/fixtures.py new file mode 100644 index 0000000..d5fdb05 --- /dev/null +++ b/tests/fixtures.py @@ -0,0 +1,51 @@ +from datasette.app import Datasette +import os +from tables import * +import tempfile + +def app_client(max_returned_rows=None): + with tempfile.TemporaryDirectory() as tmpdir: + filepath = os.path.join(tmpdir, 'test_tables.h5') + populate_file(filepath) + ds = Datasette( + [filepath], + config={ + 'default_page_size': 50, + 'max_returned_rows': max_returned_rows or 100, + } + ) + client = ds.app().test_client + client.ds = ds + yield client + +def populate_file(filepath): + class Particle(IsDescription): + identity = StringCol(itemsize=22, dflt=' ', pos=0) + idnumber = Int16Col(dflt=1, pos=1) + speed = Float32Col(dflt=1, pos=2) + + h5file = open_file(filepath, mode='w') + root = h5file.root + + group1 = h5file.create_group(root, 'group1') + group2 = h5file.create_group(root, 'group2') + + array1 = h5file.create_array(root, 'array1', ['string', 'array']) + + table1 = h5file.create_table(group1, 'table1', Particle) + table2 = h5file.create_table(group2, 'table2', Particle) + + array2 = h5file.create_array(group1, 'array2', [x for x in range(10000)]) + + for table in (table1, table2): + row = table.row + + for i in range(10000): + row['identity'] = 'This is particle: %2d' % (i) + row['idnumber'] = i + row['speed'] = i * 2. + row.append() + + table.flush() + + h5file.close() diff --git a/tests/test_api.py b/tests/test_api.py new file mode 100644 index 0000000..80c6409 --- /dev/null +++ b/tests/test_api.py @@ -0,0 +1,54 @@ +from .fixtures import app_client +import pytest + +pytest.fixture(scope='module')(app_client) + +def test_homepage(app_client): + _, response = app_client.get('/.json') + assert response.status == 200 + assert response.json.keys() == {'test_tables': 0}.keys() + d = response.json['test_tables'] + assert d['name'] == 'test_tables' + assert d['tables_count'] == 4 + +def test_database_page(app_client): + response = app_client.get('/test_tables.json', gather_request=False) + data = response.json + assert 'test_tables' == data['database'] + assert [{ + 'name': '/array1', + 'columns': [], + 'primary_keys': [], + 'count': 2, + 'label_column': None, + 'hidden': False, + 'fts_table': None, + 'foreign_keys': {'incoming': [], 'outgoing': []} + }, { + 'name': '/group1/array2', + 'columns': [], + 'primary_keys': [], + 'count': 10000, + 'label_column': None, + 'hidden': False, + 'fts_table': None, + 'foreign_keys': {'incoming': [], 'outgoing': []} + }, { + 'name': '/group1/table1', + 'columns': ['identity', 'idnumber', 'speed'], + 'primary_keys': [], + 'count': 10000, + 'label_column': None, + 'hidden': False, + 'fts_table': None, + 'foreign_keys': {'incoming': [], 'outgoing': []} + }, { + 'name': '/group2/table2', + 'columns': ['identity', 'idnumber', 'speed'], + 'primary_keys': [], + 'count': 10000, + 'label_column': None, + 'hidden': False, + 'fts_table': None, + 'foreign_keys': {'incoming': [], 'outgoing': []} + }] == data['tables']