From 06bdd1ac2d8836a2ab8e14074f2435a4c3a595b5 Mon Sep 17 00:00:00 2001
From: Javier Sancho <jsf@jsancho.org>
Date: Wed, 23 May 2018 11:50:01 +0200
Subject: [PATCH] Tests for api homepage and api database page

---
 setup.py          |  3 ++-
 tests/__init__.py |  0
 tests/fixtures.py | 51 ++++++++++++++++++++++++++++++++++++++++++++
 tests/test_api.py | 54 +++++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 107 insertions(+), 1 deletion(-)
 create mode 100644 tests/__init__.py
 create mode 100644 tests/fixtures.py
 create mode 100644 tests/test_api.py

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']
-- 
2.39.2