""" Author: Michel Peltriaux Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany Contact: ksp-servicestelle@sgdnord.rlp.de Created on: 24.08.23 """ import json from datetime import timedelta from django.core.exceptions import ObjectDoesNotExist from django.urls import reverse from django.utils.timezone import now from django.utils.translation import gettext_lazy as _ from django.test import RequestFactory from intervention.forms.intervention import NewInterventionForm, EditInterventionForm from intervention.forms.modals.revocation import NewRevocationModalForm, EditRevocationModalForm, \ RemoveRevocationModalForm from intervention.forms.modals.share import ShareModalForm from intervention.models import Revocation from konova.forms import SimpleGeomForm from konova.settings import DEFAULT_GROUP, ZB_GROUP from konova.tests.test_views import BaseTestCase from konova.utils.generators import generate_random_string from konova.utils.message_templates import REVOCATION_ADDED, REVOCATION_EDITED, REVOCATION_REMOVED from user.models import UserAction class NewInterventionFormTestCase(BaseTestCase): def setUp(self) -> None: super().setUp() def test_init(self): form = NewInterventionForm() self.assertEqual(form.form_title, str(_("New intervention"))) self.assertEqual(form.action_url, reverse("intervention:new")) self.assertEqual(form.cancel_redirect, reverse("intervention:index")) initial_identifier = form.fields["identifier"].initial self.assertIsNotNone(initial_identifier) self.assertIn("EIV-", initial_identifier) def test_is_valid(self): data = { "identifier": generate_random_string(length=15, use_letters_uc=True), "title": generate_random_string(length=15, use_letters_uc=True), } form = NewInterventionForm({}) self.assertFalse(form.is_valid()) form = NewInterventionForm(data) self.assertTrue(form.is_valid(), msg=form.errors) def test_save(self): data = { "identifier": generate_random_string(length=15, use_letters_uc=True), "title": generate_random_string(length=15, use_letters_uc=True), } test_geom = self.create_dummy_geometry() geom_form_data = self.create_geojson( test_geom ) geom_form_data = json.loads(geom_form_data) geom_form_data = { "geom": json.dumps(geom_form_data) } geom_form = SimpleGeomForm(geom_form_data) form = NewInterventionForm(data) self.assertTrue(form.is_valid()) self.assertTrue(geom_form.is_valid()) obj = form.save(self.superuser, geom_form) self.assertEqual(obj.identifier, data["identifier"]) self.assertEqual(obj.title, data["title"]) self.assertIsNotNone(obj.legal) self.assertIsNotNone(obj.responsible) self.assertIsNotNone(obj.responsible.handler) self.assertEqual(obj.created.action, UserAction.CREATED) self.assertEqual(obj.created.user, self.superuser) self.assertEqual(obj.created, obj.log.first()) self.assertEqual(obj.created, obj.modified) self.assertIn(self.superuser, obj.shared_users) self.assertTrue(test_geom.equals_exact(obj.geometry.geom, 0.000001)) class EditInterventionFormTestCase(NewInterventionFormTestCase): def test_init(self): today = now().date() data = { "identifier": self.intervention.identifier, "title": generate_random_string(length=5, use_letters_lc=True), "comment": generate_random_string(length=5, use_letters_lc=True), "registration_date": today, "binding_date": today, "registration_file_number": generate_random_string(length=5, use_numbers=True), "conservation_file_number": generate_random_string(length=5, use_numbers=True), } test_geom = self.create_dummy_geometry() geom_form_data = self.create_geojson( test_geom ) geom_form_data = json.loads(geom_form_data) geom_form_data = { "geom": json.dumps(geom_form_data) } geom_form = SimpleGeomForm(geom_form_data) form = EditInterventionForm(data, instance=self.intervention) self.assertTrue(geom_form.is_valid()) self.assertTrue(form.is_valid()) obj = form.save(self.superuser, geom_form) last_log = obj.log.first() self.assertEqual(last_log.user, self.superuser) self.assertEqual(last_log.action, UserAction.EDITED) self.assertEqual(last_log, obj.modified) self.assertEqual(obj.identifier, self.intervention.identifier) self.assertIsNotNone(obj.legal) self.assertIsNotNone(obj.responsible) self.assertIsNotNone(obj.responsible.handler) self.assertEqual(obj.title, data["title"]) self.assertEqual(obj.comment, data["comment"]) self.assertTrue(test_geom.equals_exact(obj.geometry.geom, 0.000001)) self.assertEqual(obj.legal.binding_date, today) self.assertEqual(obj.legal.registration_date, today) self.assertEqual(obj.responsible.registration_file_number, data["registration_file_number"]) self.assertEqual(obj.responsible.conservation_file_number, data["conservation_file_number"]) class ShareModalFormTestCase(BaseTestCase): def setUp(self) -> None: super().setUp() self.request = RequestFactory().request() self.request.user = self.superuser def test_init(self): self.intervention.access_token = None self.intervention.save() form = ShareModalForm( request=self.request, instance=self.intervention ) self.assertIsNotNone(self.intervention.access_token) self.assertEqual(form.form_title, str(_("Share"))) self.assertEqual(form.form_caption, str(_("Share settings for {}").format( self.intervention.identifier ))) self.assertEqual(form.template, "modal/modal_form.html") self.assertEqual(form.instance, self.intervention) self.assertEqual(form.user, self.superuser) def test_is_valid_and_save(self): # make user default-group-only (special treatment) self.superuser.groups.set( self.groups.filter( name=DEFAULT_GROUP ) ) self.assertNotIn(self.superuser, self.intervention.shared_users) self.assertNotIn(self.team, self.intervention.shared_teams) # Add new sharing data ## Default-only is able to add new sharing but can not remove existing ones data = { "users": [self.superuser.id,], "teams": [self.team.id,], } form = ShareModalForm( data, request=self.request, instance=self.intervention, ) self.assertTrue(form.is_valid(), msg=form.errors) form.save() self.assertIn(self.superuser, self.intervention.shared_users) self.assertIn(self.team, self.intervention.shared_teams) # Try to remove sharing data das default-only user data = { "users": [], "teams": [], } form = ShareModalForm( data, request=self.request, instance=self.intervention, ) self.assertFalse(form.is_valid(), msg=form.errors) self.assertTrue(form.has_error("teams")) self.assertTrue(form.has_error("users")) # Add another permission group for user self.superuser.groups.add( self.groups.get( name=ZB_GROUP ) ) form = ShareModalForm( data, request=self.request, instance=self.intervention, ) self.assertTrue(form.is_valid(), msg=form.errors) self.assertIn(self.superuser, self.intervention.shared_users) self.assertIn(self.team, self.intervention.shared_teams) form.save() self.assertNotIn(self.superuser, self.intervention.shared_users) self.assertNotIn(self.team, self.intervention.shared_teams) class NewRevocationModalFormTestCase(BaseTestCase): def setUp(self) -> None: super().setUp() self.request = RequestFactory().request() self.request.user = self.superuser def test_init(self): form = NewRevocationModalForm( request=self.request, instance=self.intervention ) self.assertEqual(form.instance, self.intervention) self.assertEqual(form.user, self.request.user) self.assertEqual(form.request, self.request) self.assertEqual(form.form_title, str(_("Add revocation"))) self.assertEqual(form.form_caption, "") self.assertEqual(form.form_attrs, { "enctype": "multipart/form-data" }) def test_save(self): data = { "date": now().date(), "file": None, "comment": generate_random_string(20, use_letters_uc=True) } form = NewRevocationModalForm( data, request=self.request, instance=self.intervention ) self.assertTrue(form.is_valid(), msg=form.errors) obj = form.save() self.assertEqual(obj.intervention, self.intervention) self.assertEqual(obj.date, data["date"]) self.assertEqual(obj.legal, self.intervention.legal) self.assertEqual(obj.comment, data["comment"]) last_log = self.intervention.log.first() self.assertEqual(last_log.user, self.superuser) self.assertEqual(last_log.action, UserAction.EDITED) self.assertEqual(last_log.comment, REVOCATION_ADDED) class EditRevocationModalFormTestCase(NewRevocationModalFormTestCase): def setUp(self) -> None: super().setUp() self.revoc = Revocation.objects.get_or_create( date=now().date(), comment="TEST", legal=self.intervention.legal, )[0] def test_init(self): new_date = now().date() - timedelta(days=10) data = { "date": new_date, "comment": generate_random_string(20, use_letters_lc=True) } form = EditRevocationModalForm( data, request=self.request, instance=self.intervention, revocation=self.revoc ) self.assertTrue(form.is_valid(), msg=form.errors) obj = form.save() self.assertEqual(obj.date, new_date) self.assertEqual(obj.comment, data["comment"]) self.assertEqual(obj.legal, self.intervention.legal) self.assertEqual(obj.intervention, self.intervention) last_log = self.intervention.log.first() self.assertEqual(last_log.action, UserAction.EDITED) self.assertEqual(last_log.user, self.superuser) self.assertEqual(last_log.comment, REVOCATION_EDITED) class RemoveRevocationModalFormTestCase(EditRevocationModalFormTestCase): def setUp(self) -> None: super().setUp() def test_init(self): form = RemoveRevocationModalForm( request=self.request, instance=self.intervention, revocation=self.revoc, ) self.assertEqual(form.instance, self.intervention) self.assertEqual(form.revocation, self.revoc) self.assertEqual(form.request, self.request) self.assertEqual(form.user, self.request.user) def test_save(self): data = { "confirm": True, } form = RemoveRevocationModalForm( data, request=self.request, instance=self.intervention, revocation=self.revoc ) self.assertTrue(form.is_valid(), msg=form.errors) form.save() try: self.revoc.refresh_from_db() self.fail("Revocation should not exist anymore") except ObjectDoesNotExist: pass last_log = self.intervention.log.first() self.assertEqual(last_log.action, UserAction.EDITED) self.assertEqual(last_log.user, self.superuser) self.assertEqual(last_log.comment, REVOCATION_REMOVED)