164 lines
7.2 KiB
Python
164 lines
7.2 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(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(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(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"], self.eco_account.responsible.handler)
|
||
|
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(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"], self.ema.responsible.handler)
|
||
|
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())
|