mpeltriaux
c98f41c9a8
* adds handler code list usage to forms and models * updates tests * extends API for handler code handling
208 lines
9.3 KiB
Python
208 lines
9.3 KiB
Python
"""
|
|
Author: Michel Peltriaux
|
|
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
|
|
Contact: michel.peltriaux@sgdnord.rlp.de
|
|
Created on: 28.01.22
|
|
|
|
"""
|
|
|
|
import json
|
|
|
|
from django.contrib.gis import geos
|
|
from django.urls import reverse
|
|
|
|
from api.tests.v1.share.test_api_sharing import BaseAPIV1TestCase
|
|
|
|
|
|
class APIV1UpdateTestCase(BaseAPIV1TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super().setUpTestData()
|
|
|
|
def _run_update_request(self, url, data):
|
|
data = json.dumps(data)
|
|
response = self.client.put(
|
|
url,
|
|
data=data,
|
|
content_type="application/json",
|
|
**self.header_data
|
|
)
|
|
return response
|
|
|
|
def _test_update_object(self, url, put_body):
|
|
""" Tests the API update of a data object.
|
|
|
|
Put body data stored in a local json file
|
|
|
|
Args:
|
|
url (str): The api creation url
|
|
put_body (dict): The put body content as dict
|
|
|
|
Returns:
|
|
|
|
"""
|
|
response = self._run_update_request(url, put_body)
|
|
self.assertEqual(response.status_code, 200, msg=response.content)
|
|
content = json.loads(response.content)
|
|
self.assertIsNotNone(content.get("id", None), msg=response.content)
|
|
|
|
def test_update_intervention(self):
|
|
""" Tests api update
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.intervention.share_with_user(self.superuser)
|
|
modified_on = self.intervention.modified
|
|
url = reverse("api:v1:intervention", args=(str(self.intervention.id),))
|
|
json_file_path = "api/tests/v1/update/intervention_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
self._test_update_object(url, put_body)
|
|
self.intervention.refresh_from_db()
|
|
|
|
put_props = put_body["properties"]
|
|
put_geom = geos.fromstr(json.dumps(put_body))
|
|
self.assertEqual(put_geom, self.intervention.geometry.geom)
|
|
self.assertEqual(put_props["title"], self.intervention.title)
|
|
self.assertNotEqual(modified_on, self.intervention.modified)
|
|
self.assertEqual(put_props["responsible"]["registration_file_number"], self.intervention.responsible.registration_file_number)
|
|
self.assertEqual(put_props["responsible"]["conservation_file_number"], self.intervention.responsible.conservation_file_number)
|
|
self.assertEqual(put_props["legal"]["registration_date"], str(self.intervention.legal.registration_date))
|
|
self.assertEqual(put_props["legal"]["binding_date"], str(self.intervention.legal.binding_date))
|
|
|
|
def test_update_compensation(self):
|
|
""" Tests api update
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.compensation.intervention = self.intervention
|
|
self.compensation.save()
|
|
self.intervention.share_with_user(self.superuser)
|
|
|
|
modified_on = self.compensation.modified
|
|
url = reverse("api:v1:compensation", args=(str(self.compensation.id),))
|
|
json_file_path = "api/tests/v1/update/compensation_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
put_body["properties"]["intervention"] = str(self.intervention.id)
|
|
self._test_update_object(url, put_body)
|
|
self.compensation.refresh_from_db()
|
|
|
|
put_props = put_body["properties"]
|
|
put_geom = geos.fromstr(json.dumps(put_body))
|
|
self.assertEqual(put_geom, self.compensation.geometry.geom)
|
|
self.assertEqual(put_props["title"], self.compensation.title)
|
|
self.assertNotEqual(modified_on, self.compensation.modified)
|
|
self.assertEqual(put_props["is_cef"], self.compensation.is_cef)
|
|
self.assertEqual(put_props["is_coherence_keeping"], self.compensation.is_coherence_keeping)
|
|
self.assertEqual(len(put_props["actions"]), self.compensation.actions.count())
|
|
self.assertEqual(len(put_props["before_states"]), self.compensation.before_states.count())
|
|
self.assertEqual(len(put_props["after_states"]), self.compensation.after_states.count())
|
|
self.assertEqual(len(put_props["deadlines"]), self.compensation.deadlines.count())
|
|
|
|
def test_update_ecoaccount(self):
|
|
""" Tests api update
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.eco_account.share_with_user(self.superuser)
|
|
|
|
modified_on = self.eco_account.modified
|
|
url = reverse("api:v1:ecoaccount", args=(str(self.eco_account.id),))
|
|
json_file_path = "api/tests/v1/update/ecoaccount_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
self._test_update_object(url, put_body)
|
|
self.eco_account.refresh_from_db()
|
|
|
|
put_props = put_body["properties"]
|
|
put_geom = geos.fromstr(json.dumps(put_body))
|
|
self.assertEqual(put_geom, self.eco_account.geometry.geom)
|
|
self.assertEqual(put_props["title"], self.eco_account.title)
|
|
self.assertNotEqual(modified_on, self.eco_account.modified)
|
|
self.assertEqual(put_props["deductable_surface"], str(self.eco_account.deductable_surface))
|
|
self.assertEqual(put_props["responsible"]["conservation_office"], self.eco_account.responsible.conservation_office)
|
|
self.assertEqual(put_props["responsible"]["conservation_file_number"], self.eco_account.responsible.conservation_file_number)
|
|
self.assertEqual(put_props["responsible"]["handler"]["detail"], self.eco_account.responsible.handler.detail)
|
|
self.assertEqual(put_props["legal"]["agreement_date"], str(self.eco_account.legal.registration_date))
|
|
self.assertEqual(len(put_props["actions"]), self.eco_account.actions.count())
|
|
self.assertEqual(len(put_props["before_states"]), self.eco_account.before_states.count())
|
|
self.assertEqual(len(put_props["after_states"]), self.eco_account.after_states.count())
|
|
self.assertEqual(len(put_props["deadlines"]), self.eco_account.deadlines.count())
|
|
|
|
def test_update_ema(self):
|
|
""" Tests api update
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.ema.share_with_user(self.superuser)
|
|
|
|
modified_on = self.ema.modified
|
|
url = reverse("api:v1:ema", args=(str(self.ema.id),))
|
|
json_file_path = "api/tests/v1/update/ema_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
self._test_update_object(url, put_body)
|
|
self.ema.refresh_from_db()
|
|
|
|
put_props = put_body["properties"]
|
|
put_geom = geos.fromstr(json.dumps(put_body))
|
|
self.assertEqual(put_geom, self.ema.geometry.geom)
|
|
self.assertEqual(put_props["title"], self.ema.title)
|
|
self.assertNotEqual(modified_on, self.ema.modified)
|
|
self.assertEqual(put_props["responsible"]["conservation_office"], self.ema.responsible.conservation_office)
|
|
self.assertEqual(put_props["responsible"]["conservation_file_number"], self.ema.responsible.conservation_file_number)
|
|
self.assertEqual(put_props["responsible"]["handler"]["detail"], self.ema.responsible.handler.detail)
|
|
self.assertEqual(len(put_props["actions"]), self.ema.actions.count())
|
|
self.assertEqual(len(put_props["before_states"]), self.ema.before_states.count())
|
|
self.assertEqual(len(put_props["after_states"]), self.ema.after_states.count())
|
|
self.assertEqual(len(put_props["deadlines"]), self.ema.deadlines.count())
|
|
|
|
def test_update_deduction(self):
|
|
""" Tests api update
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.deduction.intervention.share_with_user(self.superuser)
|
|
self.deduction.account.share_with_user(self.superuser)
|
|
|
|
url = reverse("api:v1:deduction", args=(str(self.deduction.id),))
|
|
json_file_path = "api/tests/v1/update/deduction_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
put_body["intervention"] = str(self.deduction.intervention.id)
|
|
put_body["eco_account"] = str(self.deduction.account.id)
|
|
|
|
self._test_update_object(url, put_body)
|
|
self.deduction.refresh_from_db()
|
|
|
|
self.assertEqual(put_body["intervention"], str(self.deduction.intervention.id))
|
|
self.assertEqual(put_body["eco_account"], str(self.deduction.account.id))
|
|
self.assertEqual(put_body["surface"], self.deduction.surface)
|
|
|
|
def test_update_share_intervention(self):
|
|
self.intervention.share_with_user(self.superuser)
|
|
url = reverse("api:v1:intervention-share", args=(str(self.intervention.id),))
|
|
json_file_path = "api/tests/v1/update/intervention_share_update_put_body.json"
|
|
with open(json_file_path) as json_file:
|
|
put_body = json.load(fp=json_file)
|
|
put_body["users"] = [self.user.username]
|
|
put_body["teams"] = [self.team.name]
|
|
|
|
self.assertFalse(self.intervention.is_shared_with(self.user))
|
|
self.assertEqual(0, self.intervention.shared_teams.count())
|
|
|
|
response = self._run_update_request(url, put_body)
|
|
self.assertEqual(response.status_code, 200, msg=response.content)
|
|
self.intervention.refresh_from_db()
|
|
|
|
self.assertEqual(1, self.intervention.shared_teams.count())
|
|
self.assertEqual(2, self.intervention.shared_users.count())
|
|
self.assertEqual(self.team.name, self.intervention.shared_teams.first().name)
|
|
self.assertTrue(self.intervention.is_shared_with(self.user))
|