From 0d0a1e4875952175ad63dc9882058051f4dcf2d2 Mon Sep 17 00:00:00 2001 From: Viktor Bozhinov Date: Mon, 17 Jan 2022 17:29:01 +0000 Subject: [PATCH] test: unit test `from_icat` `Instrument` entity creation #265 --- test/search_api/test_models.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/test/search_api/test_models.py b/test/search_api/test_models.py index d456c9f1..01fbe3e8 100644 --- a/test/search_api/test_models.py +++ b/test/search_api/test_models.py @@ -425,3 +425,23 @@ def test_from_icat_file_entity_with_data_for_all_related_entities(self): file_entity = models.File.from_icat(icat_data, []) assert file_entity.dict(by_alias=True) == expected_entity_data + + def test_from_icat_instrument_entity_without_data_for_related_entities(self): + icat_data = INSTRUMENT_ICAT_DATA.copy() + icat_data["facility"] = FACILITY_ICAT_DATA + + instrument_entity = models.Instrument.from_icat(icat_data, []) + + assert instrument_entity.dict(by_alias=True) == INSTRUMENT_PANOSC_DATA + + def test_from_icat_instrument_entity_with_data_for_all_related_entities(self): + expected_entity_data = INSTRUMENT_PANOSC_DATA.copy() + expected_entity_data["datasets"] = [DATASET_PANOSC_DATA] + + icat_data = INSTRUMENT_ICAT_DATA.copy() + icat_data["facility"] = FACILITY_ICAT_DATA + icat_data["datasetInstruments"] = [{"dataset": DATASET_ICAT_DATA.copy()}] + + instrument_entity = models.Instrument.from_icat(icat_data, ["datasets"]) + + assert instrument_entity.dict(by_alias=True) == expected_entity_data