test: unit test from_icat Technique entity creation #265

ral-facilities · Jan 17, 2022 · fb836c7 · fb836c7
1 parent 077c279
commit fb836c7
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/test/search_api/test_models.py b/test/search_api/test_models.py
@@ -572,3 +572,19 @@ def test_from_icat_sample_entity_with_data_for_all_related_entities(self):
         sample_entity = models.Sample.from_icat(icat_data, ["datasets"])
 
         assert sample_entity.dict(by_alias=True) == expected_entity_data
+
+    def test_from_icat_technique_entity_without_data_for_related_entities(self):
+        technique_entity = models.Technique.from_icat(TECHNIQUE_ICAT_DATA, [])
+
+        assert technique_entity.dict(by_alias=True) == TECHNIQUE_PANOSC_DATA
+
+    def test_from_icat_technique_entity_with_data_for_all_related_entities(self):
+        expected_entity_data = TECHNIQUE_PANOSC_DATA.copy()
+        expected_entity_data["datasets"] = [DATASET_PANOSC_DATA]
+
+        icat_data = TECHNIQUE_ICAT_DATA.copy()
+        icat_data["datasetTechniques"] = [{"dataset": DATASET_ICAT_DATA}]
+
+        technique_entity = models.Technique.from_icat(icat_data, ["datasets"])
+
+        assert technique_entity.dict(by_alias=True) == expected_entity_data