kopia lustrzana https://github.com/simonw/datasette
174 wiersze
4.7 KiB
Python
174 wiersze
4.7 KiB
Python
from datasette.app import Datasette
|
|
import os
|
|
import pytest
|
|
import sqlite3
|
|
import tempfile
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def app_client():
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
filepath = os.path.join(tmpdir, 'test_tables.db')
|
|
conn = sqlite3.connect(filepath)
|
|
conn.executescript(TABLES)
|
|
os.chdir(os.path.dirname(filepath))
|
|
yield Datasette([filepath]).app().test_client
|
|
|
|
|
|
def test_homepage(app_client):
|
|
_, response = app_client.get('/')
|
|
assert response.status == 200
|
|
assert 'test_tables' in response.text
|
|
|
|
# Now try the JSON
|
|
_, 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'] == 5
|
|
|
|
|
|
def test_database_page(app_client):
|
|
_, response = app_client.get('/test_tables', allow_redirects=False)
|
|
assert response.status == 302
|
|
_, response = app_client.get('/test_tables')
|
|
assert 'test_tables' in response.text
|
|
# Test JSON list of tables
|
|
_, response = app_client.get('/test_tables.json')
|
|
data = response.json
|
|
assert 'test_tables' == data['database']
|
|
assert [{
|
|
'columns': ['pk', 'content'],
|
|
'name': 'Table With Space In Name',
|
|
'table_rows': 0,
|
|
}, {
|
|
'columns': ['pk1', 'pk2', 'content'],
|
|
'name': 'compound_primary_key',
|
|
'table_rows': 0,
|
|
}, {
|
|
'columns': ['content'],
|
|
'name': 'no_primary_key',
|
|
'table_rows': 0,
|
|
}, {
|
|
'columns': ['pk', 'content'],
|
|
'name': 'simple_primary_key',
|
|
'table_rows': 2,
|
|
}, {
|
|
'columns': ['pk', 'content'],
|
|
'name': 'table/with/slashes.csv',
|
|
'table_rows': 1,
|
|
}] == data['tables']
|
|
|
|
|
|
def test_custom_sql(app_client):
|
|
_, response = app_client.get(
|
|
'/test_tables.jsono?sql=select+content+from+simple_primary_key'
|
|
)
|
|
data = response.json
|
|
assert {
|
|
'sql': 'select content from simple_primary_key',
|
|
'params': {}
|
|
} == data['query']
|
|
assert [
|
|
{'content': 'hello'},
|
|
{'content': 'world'}
|
|
] == data['rows']
|
|
assert ['content'] == data['columns']
|
|
assert 'test_tables' == data['database']
|
|
|
|
|
|
def test_invalid_custom_sql(app_client):
|
|
_, response = app_client.get(
|
|
'/test_tables?sql=.schema'
|
|
)
|
|
assert response.status == 400
|
|
assert 'Statement must begin with SELECT' in response.text
|
|
_, response = app_client.get(
|
|
'/test_tables.json?sql=.schema'
|
|
)
|
|
assert response.status == 400
|
|
assert response.json['ok'] is False
|
|
assert 'Statement must begin with SELECT' == response.json['error']
|
|
|
|
|
|
def test_table_page(app_client):
|
|
_, response = app_client.get('/test_tables/simple_primary_key')
|
|
assert response.status == 200
|
|
_, response = app_client.get('/test_tables/simple_primary_key.jsono')
|
|
assert response.status == 200
|
|
data = response.json
|
|
assert data['query']['sql'] == 'select * from simple_primary_key order by pk limit 51'
|
|
assert data['query']['params'] == {}
|
|
assert data['rows'] == [{
|
|
'pk': '1',
|
|
'content': 'hello',
|
|
}, {
|
|
'pk': '2',
|
|
'content': 'world',
|
|
}]
|
|
|
|
|
|
def test_table_with_slashes_in_name(app_client):
|
|
_, response = app_client.get('/test_tables/table%2Fwith%2Fslashes.csv')
|
|
assert response.status == 200
|
|
_, response = app_client.get('/test_tables/table%2Fwith%2Fslashes.csv.jsono')
|
|
assert response.status == 200
|
|
data = response.json
|
|
assert data['rows'] == [{
|
|
'pk': '3',
|
|
'content': 'hey',
|
|
}]
|
|
|
|
|
|
def test_view(app_client):
|
|
_, response = app_client.get('/test_tables/simple_view')
|
|
assert response.status == 200
|
|
_, response = app_client.get('/test_tables/simple_view.jsono')
|
|
assert response.status == 200
|
|
data = response.json
|
|
assert data['rows'] == [{
|
|
'upper_content': 'HELLO',
|
|
'content': 'hello',
|
|
}, {
|
|
'upper_content': 'WORLD',
|
|
'content': 'world',
|
|
}]
|
|
|
|
|
|
TABLES = '''
|
|
CREATE TABLE simple_primary_key (
|
|
pk varchar(30) primary key,
|
|
content text
|
|
);
|
|
|
|
CREATE TABLE compound_primary_key (
|
|
pk1 varchar(30),
|
|
pk2 varchar(30),
|
|
content text,
|
|
PRIMARY KEY (pk1, pk2)
|
|
);
|
|
|
|
CREATE TABLE no_primary_key (
|
|
content text
|
|
);
|
|
|
|
CREATE TABLE "Table With Space In Name" (
|
|
pk varchar(30) primary key,
|
|
content text
|
|
);
|
|
|
|
CREATE TABLE "table/with/slashes.csv" (
|
|
pk varchar(30) primary key,
|
|
content text
|
|
);
|
|
|
|
INSERT INTO simple_primary_key VALUES (1, 'hello');
|
|
INSERT INTO simple_primary_key VALUES (2, 'world');
|
|
|
|
INSERT INTO [table/with/slashes.csv] VALUES (3, 'hey');
|
|
|
|
CREATE VIEW simple_view AS
|
|
SELECT content, upper(content) AS upper_content FROM simple_primary_key;
|
|
'''
|