143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
"""
|
|
Author: Michel Peltriaux
|
|
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
|
|
Contact: michel.peltriaux@sgdnord.rlp.de
|
|
Created on: 27.01.22
|
|
|
|
"""
|
|
import json
|
|
|
|
from django.urls import reverse
|
|
|
|
from api.models import ExternalIdentifier
|
|
from api.tests.v1.share.test_api_sharing import BaseAPIV1TestCase
|
|
|
|
|
|
class APIV1CreateTestCase(BaseAPIV1TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
super().setUpTestData()
|
|
|
|
def _run_create_request(self, url, data):
|
|
data = json.dumps(data)
|
|
response = self.client.post(
|
|
url,
|
|
data=data,
|
|
content_type="application/json",
|
|
**self.header_data
|
|
)
|
|
return response
|
|
|
|
def _test_create_object(self, url, post_body):
|
|
""" Tests the API creation of a new data object.
|
|
|
|
Post body data stored in a local json file
|
|
|
|
Args:
|
|
url (str): The api creation url
|
|
post_body (dict): The post body content as dict
|
|
|
|
Returns:
|
|
|
|
"""
|
|
response = self._run_create_request(url, post_body)
|
|
self.assertEqual(response.status_code, 200, msg=response.content)
|
|
content = json.loads(response.content)
|
|
_id = content.get("id", None)
|
|
self.assertIsNotNone(_id, msg=response.content)
|
|
return _id
|
|
|
|
def _test_external_identifier_created(self, internal_id, external_id):
|
|
""" Tests whether an external identifier has been created
|
|
|
|
Args:
|
|
internal_id ():
|
|
external_id ():
|
|
|
|
Returns:
|
|
|
|
"""
|
|
external_identifier = ExternalIdentifier.objects.get(internal_id=internal_id)
|
|
self.assertEqual(external_identifier.external_id, external_id)
|
|
|
|
def test_create_intervention(self):
|
|
""" Tests api creation
|
|
|
|
Returns:
|
|
|
|
"""
|
|
url = reverse("api:v1:intervention")
|
|
json_file_path = "api/tests/v1/create/intervention_create_post_body.json"
|
|
with open(json_file_path) as json_file:
|
|
post_body = json.load(fp=json_file)
|
|
internal_id = self._test_create_object(url, post_body)
|
|
self._test_external_identifier_created(internal_id, post_body["properties"]["external_identifier"])
|
|
|
|
def test_create_compensation(self):
|
|
""" Tests api creation
|
|
|
|
Returns:
|
|
|
|
"""
|
|
url = reverse("api:v1:compensation")
|
|
json_file_path = "api/tests/v1/create/compensation_create_post_body.json"
|
|
with open(json_file_path) as json_file:
|
|
post_body = json.load(fp=json_file)
|
|
post_body["properties"]["intervention"] = str(self.intervention.id)
|
|
|
|
# Expect this first request to fail, since user has no shared access on the intervention, we want to create
|
|
# a compensation for
|
|
response = self._run_create_request(url, post_body)
|
|
self.assertEqual(response.status_code, 400, msg=response.content)
|
|
content = json.loads(response.content)
|
|
self.assertGreater(len(content.get("errors", [])), 0, msg=response.content)
|
|
|
|
# Add the user to the shared users of the intervention and try again! Now everything should work as expected.
|
|
self.intervention.users.add(self.superuser)
|
|
internal_id = self._test_create_object(url, post_body)
|
|
self._test_external_identifier_created(internal_id, post_body["properties"]["external_identifier"])
|
|
|
|
def test_create_eco_account(self):
|
|
""" Tests api creation
|
|
|
|
Returns:
|
|
|
|
"""
|
|
url = reverse("api:v1:ecoaccount")
|
|
json_file_path = "api/tests/v1/create/ecoaccount_create_post_body.json"
|
|
with open(json_file_path) as json_file:
|
|
post_body = json.load(fp=json_file)
|
|
internal_id = self._test_create_object(url, post_body)
|
|
self._test_external_identifier_created(internal_id, post_body["properties"]["external_identifier"])
|
|
|
|
def test_create_ema(self):
|
|
""" Tests api creation
|
|
|
|
Returns:
|
|
|
|
"""
|
|
url = reverse("api:v1:ema")
|
|
json_file_path = "api/tests/v1/create/ema_create_post_body.json"
|
|
with open(json_file_path) as json_file:
|
|
post_body = json.load(fp=json_file)
|
|
internal_id = self._test_create_object(url, post_body)
|
|
self._test_external_identifier_created(internal_id, post_body["properties"]["external_identifier"])
|
|
|
|
def test_create_deduction(self):
|
|
""" Tests api creation
|
|
|
|
Returns:
|
|
|
|
"""
|
|
self.intervention.share_with_user(self.superuser)
|
|
self.eco_account.share_with_user(self.superuser)
|
|
|
|
url = reverse("api:v1:deduction")
|
|
json_file_path = "api/tests/v1/create/deduction_create_post_body.json"
|
|
with open(json_file_path) as json_file:
|
|
post_body = json.load(fp=json_file)
|
|
post_body["intervention"] = str(self.intervention.id)
|
|
post_body["eco_account"] = str(self.eco_account.id)
|
|
self._test_create_object(url, post_body)
|
|
|