konova/api/tests/v1/update/test_api_update.py

"""
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(put_props["is_pik"], self.compensation.is_pik)
        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))