mirror of
https://github.com/umap-project/umap.git
synced 2025-04-28 19:42:36 +02:00
82b8564520
Co-authored-by: David Larlet <david@larlet.fr> Co-authored-by: Aurélie Jallut <aurelie.jallut@beta.gouv.fr>
310 lines
12 KiB
Python
310 lines
12 KiB
Python
import pytest
|
|
from playwright.sync_api import expect
|
|
|
|
from ..base import DataLayerFactory
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
|
|
def getColors(elements):
|
|
return [
|
|
el.evaluate("e => window.getComputedStyle(e).backgroundColor")
|
|
for el in elements.all()
|
|
]
|
|
|
|
|
|
DATALAYER_DATA1 = {
|
|
"type": "FeatureCollection",
|
|
"features": [
|
|
{
|
|
"type": "Feature",
|
|
"properties": {
|
|
"mytype": "even",
|
|
"name": "Point 2",
|
|
"mynumber": 10,
|
|
"myboolean": True,
|
|
"mydate": "2024/04/14 12:19:17",
|
|
"maybeempty": "not empty",
|
|
},
|
|
"geometry": {"type": "Point", "coordinates": [0.065918, 48.385442]},
|
|
},
|
|
{
|
|
"type": "Feature",
|
|
"properties": {
|
|
"mytype": "odd",
|
|
"name": "Point 1",
|
|
"mynumber": 12,
|
|
"myboolean": False,
|
|
"mydate": "2024/03/13 12:20:20",
|
|
"maybeempty": "",
|
|
},
|
|
"geometry": {"type": "Point", "coordinates": [3.55957, 49.767074]},
|
|
},
|
|
],
|
|
"_umap_options": {
|
|
"name": "Calque 1",
|
|
},
|
|
}
|
|
|
|
|
|
DATALAYER_DATA2 = {
|
|
"type": "FeatureCollection",
|
|
"features": [
|
|
{
|
|
"type": "Feature",
|
|
"properties": {
|
|
"mytype": "even",
|
|
"name": "Point 4",
|
|
"mynumber": 10,
|
|
"myboolean": "true",
|
|
"mydate": "2024/08/18 13:14:15",
|
|
"maybeempty": None,
|
|
},
|
|
"geometry": {"type": "Point", "coordinates": [0.856934, 45.290347]},
|
|
},
|
|
{
|
|
"type": "Feature",
|
|
"properties": {
|
|
"mytype": "odd",
|
|
"name": "Point 3",
|
|
"mynumber": 14,
|
|
"mydate": "2024-04-14T10:19:17.000Z",
|
|
},
|
|
"geometry": {"type": "Point", "coordinates": [4.372559, 47.945786]},
|
|
},
|
|
{
|
|
"type": "Feature",
|
|
"properties": {
|
|
"mytype": "odd",
|
|
"name": "Point 5",
|
|
"mynumber": 10,
|
|
"mydate": "2024-04-14T10:19:17.000Z",
|
|
"myboolean": "notaboolean",
|
|
"maybeempty": "foo",
|
|
},
|
|
"geometry": {"type": "Point", "coordinates": [4.1, 47.3]},
|
|
},
|
|
],
|
|
"_umap_options": {
|
|
"name": "Calque 2",
|
|
},
|
|
}
|
|
|
|
|
|
def test_simple_equal_rule_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "mytype=odd", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
|
|
|
|
def test_simple_not_equal_rule_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "mytype!=even", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
|
|
|
|
def test_gt_rule_with_number_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "mynumber>10", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 2
|
|
|
|
|
|
def test_lt_rule_with_number_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "mynumber<14", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 4
|
|
|
|
|
|
def test_lt_rule_with_float_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "mynumber<12.3", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 4
|
|
|
|
|
|
def test_equal_rule_with_boolean_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "myboolean=true", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 2
|
|
|
|
|
|
def test_equal_rule_with_boolean_not_true_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "myboolean!=true", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
|
|
|
|
def test_equal_rule_with_boolean_false_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "myboolean=false", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 1
|
|
|
|
|
|
def test_equal_rule_with_boolean_not_false_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "myboolean!=false", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 4
|
|
|
|
|
|
def test_empty_rule_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "maybeempty=", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
|
|
|
|
def test_not_empty_rule_at_load(live_server, page, map):
|
|
map.settings["properties"]["rules"] = [
|
|
{"condition": "maybeempty!=", "options": {"color": "aliceblue"}}
|
|
]
|
|
map.save()
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=map, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{map.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 2
|
|
|
|
|
|
def test_can_create_new_rule(live_server, page, openmap):
|
|
DataLayerFactory(map=openmap, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=openmap, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{openmap.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
page.get_by_role("button", name="Edit").click()
|
|
page.get_by_role("button", name="Map advanced properties").click()
|
|
page.get_by_text("Conditional style rules").click()
|
|
page.get_by_role("button", name="Add rule").click()
|
|
page.locator("input[name=condition]").click()
|
|
page.locator("input[name=condition]").fill("mytype=odd")
|
|
page.locator(".umap-field-color .define").first.click()
|
|
page.get_by_title("AliceBlue").first.click()
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
page.locator(".edit-undo").click()
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 0
|
|
|
|
|
|
def test_can_deactive_rule_from_list(live_server, page, openmap):
|
|
openmap.settings["properties"]["rules"] = [
|
|
{"condition": "mytype=odd", "options": {"color": "aliceblue"}}
|
|
]
|
|
openmap.save()
|
|
DataLayerFactory(map=openmap, data=DATALAYER_DATA1)
|
|
DataLayerFactory(map=openmap, data=DATALAYER_DATA2)
|
|
page.goto(f"{live_server.url}{openmap.get_absolute_url()}#6/48.948/1.670")
|
|
markers = page.locator(".leaflet-marker-icon .icon_container")
|
|
expect(markers).to_have_count(5)
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
page.get_by_role("button", name="Edit").click()
|
|
page.get_by_role("button", name="Map advanced properties").click()
|
|
page.get_by_text("Conditional style rules").click()
|
|
page.get_by_role("button", name="Show/hide layer").click()
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 0
|
|
page.get_by_role("button", name="Show/hide layer").click()
|
|
colors = getColors(markers)
|
|
assert colors.count("rgb(240, 248, 255)") == 3
|
|
|
|
|
|
def test_autocomplete_datalist(live_server, page, openmap):
|
|
DataLayerFactory(map=openmap, data=DATALAYER_DATA1)
|
|
page.goto(f"{live_server.url}{openmap.get_absolute_url()}?edit#6/48.948/1.670")
|
|
page.get_by_role("button", name="Map advanced properties").click()
|
|
page.locator("summary").filter(has_text="Conditional style rules").click()
|
|
page.get_by_role("button", name="Add rule").click()
|
|
panel = page.locator(".panel.right.on")
|
|
datalist = panel.locator(".umap-field-condition datalist option")
|
|
expect(datalist).to_have_count(6)
|
|
values = {option.inner_text() for option in datalist.all()}
|
|
assert values == {"myboolean", "mytype", "mynumber", "mydate", "name", "maybeempty"}
|
|
page.get_by_placeholder("key=value or key!=value").fill("mytype")
|
|
expect(datalist).to_have_count(4)
|
|
values = {option.inner_text() for option in datalist.all()}
|
|
assert values == {"mytype=", "mytype!=", "mytype>", "mytype<"}
|
|
page.get_by_placeholder("key=value or key!=value").fill("mytype=")
|
|
expect(datalist).to_have_count(2)
|
|
values = {option.inner_text() for option in datalist.all()}
|
|
assert values == {"mytype=even", "mytype=odd"}
|