kopia lustrzana https://github.com/simonw/datasette
328 wiersze
10 KiB
Python
328 wiersze
10 KiB
Python
from datasette.facets import ColumnFacet, ArrayFacet, DateFacet
|
|
from datasette.utils import detect_json1
|
|
from .fixtures import app_client # noqa
|
|
from .utils import MockRequest
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_column_facet_suggest(app_client):
|
|
facet = ColumnFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
suggestions = await facet.suggest()
|
|
assert [
|
|
{"name": "created", "toggle_url": "http://localhost/?_facet=created"},
|
|
{"name": "planet_int", "toggle_url": "http://localhost/?_facet=planet_int"},
|
|
{"name": "on_earth", "toggle_url": "http://localhost/?_facet=on_earth"},
|
|
{"name": "state", "toggle_url": "http://localhost/?_facet=state"},
|
|
{"name": "city_id", "toggle_url": "http://localhost/?_facet=city_id"},
|
|
{"name": "neighborhood", "toggle_url": "http://localhost/?_facet=neighborhood"},
|
|
{"name": "tags", "toggle_url": "http://localhost/?_facet=tags"},
|
|
{
|
|
"name": "complex_array",
|
|
"toggle_url": "http://localhost/?_facet=complex_array",
|
|
},
|
|
] == suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_column_facet_suggest_skip_if_already_selected(app_client):
|
|
facet = ColumnFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/?_facet=planet_int&_facet=on_earth"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
suggestions = await facet.suggest()
|
|
assert [
|
|
{
|
|
"name": "created",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=created",
|
|
},
|
|
{
|
|
"name": "state",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=state",
|
|
},
|
|
{
|
|
"name": "city_id",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=city_id",
|
|
},
|
|
{
|
|
"name": "neighborhood",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=neighborhood",
|
|
},
|
|
{
|
|
"name": "tags",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=tags",
|
|
},
|
|
{
|
|
"name": "complex_array",
|
|
"toggle_url": "http://localhost/?_facet=planet_int&_facet=on_earth&_facet=complex_array",
|
|
},
|
|
] == suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_column_facet_suggest_skip_if_enabled_by_metadata(app_client):
|
|
facet = ColumnFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
metadata={"facets": ["city_id"]},
|
|
)
|
|
suggestions = [s["name"] for s in await facet.suggest()]
|
|
assert [
|
|
"created",
|
|
"planet_int",
|
|
"on_earth",
|
|
"state",
|
|
"neighborhood",
|
|
"tags",
|
|
"complex_array",
|
|
] == suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_column_facet_results(app_client):
|
|
facet = ColumnFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/?_facet=city_id"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
buckets, timed_out = await facet.facet_results()
|
|
assert [] == timed_out
|
|
assert {
|
|
"city_id": {
|
|
"name": "city_id",
|
|
"type": "column",
|
|
"hideable": True,
|
|
"toggle_url": "/",
|
|
"results": [
|
|
{
|
|
"value": 1,
|
|
"label": "San Francisco",
|
|
"count": 6,
|
|
"toggle_url": "http://localhost/?_facet=city_id&city_id=1",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 2,
|
|
"label": "Los Angeles",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?_facet=city_id&city_id=2",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 3,
|
|
"label": "Detroit",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?_facet=city_id&city_id=3",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 4,
|
|
"label": "Memnonia",
|
|
"count": 1,
|
|
"toggle_url": "http://localhost/?_facet=city_id&city_id=4",
|
|
"selected": False,
|
|
},
|
|
],
|
|
"truncated": False,
|
|
}
|
|
} == buckets
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_column_facet_from_metadata_cannot_be_hidden(app_client):
|
|
facet = ColumnFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
metadata={"facets": ["city_id"]},
|
|
)
|
|
buckets, timed_out = await facet.facet_results()
|
|
assert [] == timed_out
|
|
assert {
|
|
"city_id": {
|
|
"name": "city_id",
|
|
"type": "column",
|
|
"hideable": False,
|
|
"toggle_url": "/",
|
|
"results": [
|
|
{
|
|
"value": 1,
|
|
"label": "San Francisco",
|
|
"count": 6,
|
|
"toggle_url": "http://localhost/?city_id=1",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 2,
|
|
"label": "Los Angeles",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?city_id=2",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 3,
|
|
"label": "Detroit",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?city_id=3",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": 4,
|
|
"label": "Memnonia",
|
|
"count": 1,
|
|
"toggle_url": "http://localhost/?city_id=4",
|
|
"selected": False,
|
|
},
|
|
],
|
|
"truncated": False,
|
|
}
|
|
} == buckets
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(not detect_json1(), reason="Requires the SQLite json1 module")
|
|
async def test_array_facet_suggest(app_client):
|
|
facet = ArrayFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
suggestions = await facet.suggest()
|
|
assert [
|
|
{
|
|
"name": "tags",
|
|
"type": "array",
|
|
"toggle_url": "http://localhost/?_facet_array=tags",
|
|
}
|
|
] == suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(not detect_json1(), reason="Requires the SQLite json1 module")
|
|
async def test_array_facet_suggest_not_if_all_empty_arrays(app_client):
|
|
facet = ArrayFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/"),
|
|
database="fixtures",
|
|
sql="select * from facetable where tags = '[]'",
|
|
table="facetable",
|
|
)
|
|
suggestions = await facet.suggest()
|
|
assert [] == suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.skipif(not detect_json1(), reason="Requires the SQLite json1 module")
|
|
async def test_array_facet_results(app_client):
|
|
facet = ArrayFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/?_facet_array=tags"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
buckets, timed_out = await facet.facet_results()
|
|
assert [] == timed_out
|
|
assert {
|
|
"tags": {
|
|
"name": "tags",
|
|
"type": "array",
|
|
"results": [
|
|
{
|
|
"value": "tag1",
|
|
"label": "tag1",
|
|
"count": 2,
|
|
"toggle_url": "http://localhost/?_facet_array=tags&tags__arraycontains=tag1",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": "tag2",
|
|
"label": "tag2",
|
|
"count": 1,
|
|
"toggle_url": "http://localhost/?_facet_array=tags&tags__arraycontains=tag2",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": "tag3",
|
|
"label": "tag3",
|
|
"count": 1,
|
|
"toggle_url": "http://localhost/?_facet_array=tags&tags__arraycontains=tag3",
|
|
"selected": False,
|
|
},
|
|
],
|
|
"hideable": True,
|
|
"toggle_url": "/",
|
|
"truncated": False,
|
|
}
|
|
} == buckets
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_date_facet_results(app_client):
|
|
facet = DateFacet(
|
|
app_client.ds,
|
|
MockRequest("http://localhost/?_facet_date=created"),
|
|
database="fixtures",
|
|
sql="select * from facetable",
|
|
table="facetable",
|
|
)
|
|
buckets, timed_out = await facet.facet_results()
|
|
assert [] == timed_out
|
|
assert {
|
|
"created": {
|
|
"name": "created",
|
|
"type": "date",
|
|
"results": [
|
|
{
|
|
"value": "2019-01-14",
|
|
"label": "2019-01-14",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?_facet_date=created&created__date=2019-01-14",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": "2019-01-15",
|
|
"label": "2019-01-15",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?_facet_date=created&created__date=2019-01-15",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": "2019-01-17",
|
|
"label": "2019-01-17",
|
|
"count": 4,
|
|
"toggle_url": "http://localhost/?_facet_date=created&created__date=2019-01-17",
|
|
"selected": False,
|
|
},
|
|
{
|
|
"value": "2019-01-16",
|
|
"label": "2019-01-16",
|
|
"count": 3,
|
|
"toggle_url": "http://localhost/?_facet_date=created&created__date=2019-01-16",
|
|
"selected": False,
|
|
},
|
|
],
|
|
"hideable": True,
|
|
"toggle_url": "/",
|
|
"truncated": False,
|
|
}
|
|
} == buckets
|