from .fixtures import app_client
import pytest
+from urllib.parse import urlencode
pytest.fixture(scope='module')(app_client)
def test_custom_sql(app_client):
response = app_client.get(
- '/test_tables.json?sql=select+identity+from+[/group1/table1]&_shape=objects',
+ '/test_tables.json?' + urlencode({
+ 'sql': 'select identity from [/group1/table1] where speed > 100 and idnumber < 55',
+ '_shape': 'objects'
+ }),
gather_request=False
)
data = response.json
+ print("*************************", data)
assert {
- 'sql': 'select identity from [/group1/table1]',
+ 'sql': 'select identity from [/group1/table1] where speed > 100 and idnumber < 55',
'params': {}
} == data['query']
- assert 50 == len(data['rows'])
+ assert 4 == len(data['rows'])
assert [
- {'identity': 'This is particle: 0'},
- {'identity': 'This is particle: 1'},
- {'identity': 'This is particle: 2'}
+ {'identity': 'This is particle: 51'},
+ {'identity': 'This is particle: 52'},
+ {'identity': 'This is particle: 53'},
+ {'identity': 'This is particle: 54'}
+ ] == data['rows']
+ assert ['identity'] == data['columns']
+ assert 'test_tables' == data['database']
+ assert False == data['truncated']
+
+def test_custom_pytables_sql(app_client):
+ response = app_client.get(
+ '/test_tables.json?' + urlencode({
+ 'sql': 'select identity from [/group1/table1] where (speed > 100) & (speed < 500)',
+ '_shape': 'objects'
+ }),
+ gather_request=False
+ )
+ data = response.json
+ assert {
+ 'sql': 'select identity from [/group1/table1] where (speed > 100) & (speed < 500)',
+ 'params': {}
+ } == data['query']
+ assert 199 == len(data['rows'])
+ assert [
+ {'identity': 'This is particle: 51'},
+ {'identity': 'This is particle: 52'},
+ {'identity': 'This is particle: 53'}
] == data['rows'][:3]
assert ['identity'] == data['columns']
assert 'test_tables' == data['database']
'status': 400,
'title': None,
} == response.json
+
+@pytest.mark.parametrize('path, expected_rows, expected_pages', [
+ ('/test_tables/%2Farray1.json', 2, 1),
+ ('/test_tables/%2Farray1.json?_size=1', 2, 2),
+ ('/test_tables/%2Fgroup1%2Farray2.json?_size=1000', 10000, 10),
+])
+def test_paginate_tables_and_arrays(app_client, path, expected_rows, expected_pages):
+ fetched = []
+ count = 0
+ while path:
+ response = app_client.get(path, gather_request=False)
+ print("*****", response.json)
+ assert 200 == response.status
+ count += 1
+ fetched.extend(response.json['rows'])
+ path = response.json['next_url']
+ if path:
+ assert response.json['next']
+ assert '_next={}'.format(response.json['next']) in path
+
+ assert expected_rows == len(fetched)
+ assert expected_pages == count