mirror of
https://github.com/umap-project/umap.git
synced 2025-04-28 19:42:36 +02:00
293 lines
9.2 KiB
Python
293 lines
9.2 KiB
Python
import tempfile
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.core.files.base import ContentFile
|
|
|
|
from umap.models import DataLayer, Map
|
|
|
|
from .base import DataLayerFactory, MapFactory
|
|
|
|
pytestmark = pytest.mark.django_db
|
|
|
|
|
|
def test_datalayers_should_be_ordered_by_rank(map, datalayer):
|
|
datalayer.rank = 5
|
|
datalayer.save()
|
|
c4 = DataLayerFactory(map=map, rank=4)
|
|
c1 = DataLayerFactory(map=map, rank=1)
|
|
c3 = DataLayerFactory(map=map, rank=3)
|
|
c2 = DataLayerFactory(map=map, rank=2)
|
|
assert list(map.datalayer_set.all()) == [c1, c2, c3, c4, datalayer]
|
|
|
|
|
|
def test_upload_to(map, datalayer):
|
|
map.pk = 302
|
|
datalayer.pk = 17
|
|
assert datalayer.upload_to().startswith("datalayer/2/0/302/17_")
|
|
|
|
|
|
def test_save_should_use_pk_as_name(map, datalayer):
|
|
assert "/{}_".format(datalayer.pk) in datalayer.geojson.name
|
|
|
|
|
|
def test_same_geojson_file_name_will_be_suffixed(map, datalayer):
|
|
before = datalayer.geojson.name
|
|
datalayer.geojson.save(before, ContentFile("{}"))
|
|
assert datalayer.geojson.name != before
|
|
assert "/{}_".format(datalayer.pk) in datalayer.geojson.name
|
|
|
|
|
|
def test_clone_should_return_new_instance(map, datalayer):
|
|
clone = datalayer.clone()
|
|
assert datalayer.pk != clone.pk
|
|
assert datalayer.name == clone.name
|
|
assert datalayer.map == clone.map
|
|
|
|
|
|
def test_clone_should_update_map_if_passed(datalayer, user, licence):
|
|
map = MapFactory(owner=user, licence=licence)
|
|
clone = datalayer.clone(map_inst=map)
|
|
assert datalayer.pk != clone.pk
|
|
assert datalayer.name == clone.name
|
|
assert datalayer.map != clone.map
|
|
assert map == clone.map
|
|
|
|
|
|
def test_clone_should_clone_geojson_too(datalayer):
|
|
clone = datalayer.clone()
|
|
assert datalayer.pk != clone.pk
|
|
assert clone.geojson is not None
|
|
assert clone.geojson.path != datalayer.geojson.path
|
|
|
|
|
|
def test_should_remove_old_versions_on_save(map, settings):
|
|
datalayer = DataLayerFactory(uuid="0f1161c0-c07f-4ba4-86c5-8d8981d8a813", old_id=17)
|
|
settings.UMAP_KEEP_VERSIONS = 3
|
|
root = Path(datalayer.storage_root())
|
|
before = len(datalayer.geojson.storage.listdir(root)[1])
|
|
newer = f"{datalayer.pk}_1440924889.geojson"
|
|
medium = f"{datalayer.pk}_1440923687.geojson"
|
|
older = f"{datalayer.pk}_1440918637.geojson"
|
|
with_old_id = f"{datalayer.old_id}_1440918537.geojson"
|
|
other = "123456_1440918637.geojson"
|
|
for path in [medium, newer, older, with_old_id, other]:
|
|
datalayer.geojson.storage.save(root / path, ContentFile("{}"))
|
|
datalayer.geojson.storage.save(root / f"{path}.gz", ContentFile("{}"))
|
|
assert len(datalayer.geojson.storage.listdir(root)[1]) == 10 + before
|
|
files = datalayer.geojson.storage.listdir(root)[1]
|
|
# Those files should be present before save, which will purge them
|
|
assert older in files
|
|
assert older + ".gz" in files
|
|
assert with_old_id in files
|
|
assert with_old_id + ".gz" in files
|
|
datalayer.save()
|
|
files = datalayer.geojson.storage.listdir(root)[1]
|
|
# Flat + gz files, but not latest gz, which is created at first datalayer read.
|
|
# older and with_old_id should have been removed
|
|
assert len(files) == 5
|
|
assert newer in files
|
|
assert medium in files
|
|
assert Path(datalayer.geojson.path).name in files
|
|
# File from another datalayer, purge should have impacted it.
|
|
assert other in files
|
|
assert other + ".gz" in files
|
|
assert older not in files
|
|
assert older + ".gz" not in files
|
|
assert with_old_id not in files
|
|
assert with_old_id + ".gz" not in files
|
|
names = [v["name"] for v in datalayer.versions]
|
|
assert names == [Path(datalayer.geojson.name).name, newer, medium]
|
|
|
|
|
|
def test_anonymous_cannot_edit_in_editors_mode(datalayer, fake_request):
|
|
datalayer.edit_status = DataLayer.COLLABORATORS
|
|
datalayer.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_owner_can_edit_in_editors_mode(datalayer, user, fake_request):
|
|
datalayer.edit_status = DataLayer.COLLABORATORS
|
|
datalayer.save()
|
|
fake_request.user = datalayer.map.owner
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_editor_can_edit_in_collaborators_mode(datalayer, user, fake_request):
|
|
map = datalayer.map
|
|
map.editors.add(user)
|
|
map.save()
|
|
datalayer.edit_status = DataLayer.COLLABORATORS
|
|
datalayer.save()
|
|
fake_request.user = user
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_team_members_can_edit_in_collaborators_mode(
|
|
datalayer, user, team, fake_request
|
|
):
|
|
user.teams.add(team)
|
|
user.save()
|
|
map = datalayer.map
|
|
map.team = team
|
|
map.save()
|
|
datalayer.edit_status = DataLayer.COLLABORATORS
|
|
datalayer.save()
|
|
fake_request.user = user
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_anonymous_can_edit_in_public_mode(datalayer, fake_request):
|
|
datalayer.edit_status = DataLayer.ANONYMOUS
|
|
datalayer.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_owner_can_edit_in_public_mode(datalayer, user, fake_request):
|
|
datalayer.edit_status = DataLayer.ANONYMOUS
|
|
datalayer.save()
|
|
fake_request.user = datalayer.map.owner
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_editor_can_edit_in_public_mode(datalayer, user, fake_request):
|
|
map = datalayer.map
|
|
map.editors.add(user)
|
|
map.save()
|
|
datalayer.edit_status = DataLayer.ANONYMOUS
|
|
datalayer.save()
|
|
fake_request.user = user
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_anonymous_cannot_edit_in_anonymous_owner_mode(datalayer, fake_request):
|
|
datalayer.edit_status = DataLayer.OWNER
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.owner = None
|
|
map.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_owner_can_edit_in_inherit_mode_and_map_in_owner_mode(datalayer, fake_request):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.edit_status = Map.OWNER
|
|
map.save()
|
|
fake_request.user = map.owner
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_editors_cannot_edit_in_inherit_mode_and_map_in_owner_mode(
|
|
datalayer, user, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.editors.add(user)
|
|
map.edit_status = Map.OWNER
|
|
map.save()
|
|
fake_request.user = user
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_team_members_cannot_edit_in_inherit_mode_and_map_in_owner_mode(
|
|
datalayer, user, team, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
user.teams.add(team)
|
|
team.save()
|
|
map = datalayer.map
|
|
map.team = team
|
|
map.edit_status = Map.OWNER
|
|
map.save()
|
|
fake_request.user = user
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_anonymous_cannot_edit_in_inherit_mode_and_map_in_owner_mode(
|
|
datalayer, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.edit_status = Map.OWNER
|
|
map.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_owner_can_edit_in_inherit_mode_and_map_in_editors_mode(
|
|
datalayer, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.edit_status = Map.COLLABORATORS
|
|
map.save()
|
|
fake_request.user = map.owner
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_editors_can_edit_in_inherit_mode_and_map_in_editors_mode(
|
|
datalayer, user, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.editors.add(user)
|
|
map.edit_status = Map.COLLABORATORS
|
|
map.save()
|
|
fake_request.user = user
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_anonymous_cannot_edit_in_inherit_mode_and_map_in_editors_mode(
|
|
datalayer, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.edit_status = Map.COLLABORATORS
|
|
map.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert not datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_anonymous_can_edit_in_inherit_mode_and_map_in_public_mode(
|
|
datalayer, fake_request
|
|
):
|
|
datalayer.edit_status = DataLayer.INHERIT
|
|
datalayer.save()
|
|
map = datalayer.map
|
|
map.edit_status = Map.ANONYMOUS
|
|
map.save()
|
|
fake_request.user = AnonymousUser()
|
|
assert datalayer.can_edit(fake_request)
|
|
|
|
|
|
def test_should_remove_all_versions_on_delete(map, settings):
|
|
datalayer = DataLayerFactory(uuid="0f1161c0-c07f-4ba4-86c5-8d8981d8a813", old_id=17)
|
|
root = Path(datalayer.storage_root())
|
|
before = len(datalayer.geojson.storage.listdir(root)[1])
|
|
other = "123456_1440918637.geojson"
|
|
files = [
|
|
f"{datalayer.pk}_1440924889.geojson",
|
|
f"{datalayer.pk}_1440923687.geojson",
|
|
f"{datalayer.pk}_1440918637.geojson",
|
|
f"{datalayer.old_id}_1440918537.geojson",
|
|
other,
|
|
]
|
|
for path in files:
|
|
datalayer.geojson.storage.save(root / path, ContentFile("{}"))
|
|
datalayer.geojson.storage.save(root / f"{path}.gz", ContentFile("{}"))
|
|
assert len(datalayer.geojson.storage.listdir(root)[1]) == 10 + before
|
|
datalayer.delete()
|
|
found = datalayer.geojson.storage.listdir(root)[1]
|
|
assert found == [other, f"{other}.gz"]
|