From faed59477105b9dcfe75bf018d6b8229ff048a2e Mon Sep 17 00:00:00 2001
From: Viktor Bozhinov <viktor_bozinov1@hotmail.com>
Date: Mon, 17 Jan 2022 17:39:40 +0000
Subject: [PATCH] test: unit test `from_icat` `Parameter` entity creation #265

---
 test/search_api/test_models.py | 46 ++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/test/search_api/test_models.py b/test/search_api/test_models.py
index 16e2e97f..5fa0f29d 100644
--- a/test/search_api/test_models.py
+++ b/test/search_api/test_models.py
@@ -484,3 +484,49 @@ def test_from_icat_member_entity_with_data_for_all_related_entities(self):
         )
 
         assert member_entity.dict(by_alias=True) == expected_entity_data
+
+    def test_from_icat_parameter_entity_without_data_for_related_entities(self):
+        icat_data = INVESTIGATION_PARAMETER_ICAT_DATA.copy()
+        icat_data["type"] = PARAMETER_TYPE_ICAT_DATA
+
+        parameter_entity = models.Parameter.from_icat(icat_data, [])
+
+        assert parameter_entity.dict(by_alias=True) == PARAMETER_PANOSC_DATA
+
+    def test_from_icat_parameter_entity_with_investigation_parameter_data(self):
+        expected_entity_data = PARAMETER_PANOSC_DATA.copy()
+        expected_entity_data["dataset"] = DATASET_PANOSC_DATA
+        expected_entity_data["document"] = DOCUMENT_PANOSC_DATA
+
+        icat_data = INVESTIGATION_PARAMETER_ICAT_DATA.copy()
+        icat_data["type"] = PARAMETER_TYPE_ICAT_DATA
+        icat_data["investigation"] = INVESTIGATION_ICAT_DATA.copy()
+        icat_data["investigation"]["type"] = INVESTIGATION_TYPE_ICAT_DATA
+        icat_data["investigation"]["keywords"] = [KEYWORD_ICAT_DATA]
+        icat_data["investigation"]["investigationInstruments"] = [
+            {"instrument": INSTRUMENT_ICAT_DATA.copy()},
+            {"instrument": INSTRUMENT_ICAT_DATA.copy()},
+        ]
+        icat_data["investigation"]["investigationInstruments"][0]["instrument"][
+            "datasetInstruments"
+        ] = [{"dataset": DATASET_ICAT_DATA}, {"dataset": DATASET_ICAT_DATA}]
+        icat_data["investigation"]["investigationInstruments"][1]["instrument"][
+            "datasetInstruments"
+        ] = []
+
+        parameter_entity = models.Parameter.from_icat(icat_data, ["dataset"])
+
+        assert parameter_entity.dict(by_alias=True) == expected_entity_data
+
+    def test_from_icat_parameter_entity_with_dataset_parameter_data(self):
+        expected_entity_data = PARAMETER_PANOSC_DATA.copy()
+        expected_entity_data["value"] = DATASET_PARAMETER_ICAT_DATA["stringValue"]
+        expected_entity_data["dataset"] = DATASET_PANOSC_DATA
+
+        icat_data = DATASET_PARAMETER_ICAT_DATA.copy()
+        icat_data["dataset"] = DATASET_ICAT_DATA
+        icat_data["type"] = PARAMETER_TYPE_ICAT_DATA
+
+        parameter_entity = models.Parameter.from_icat(icat_data, ["dataset"])
+
+        assert parameter_entity.dict(by_alias=True) == expected_entity_data