Gdt 116 update locations

tests/fixtures/aardvark_records.jsonl

@@ -1,2 +1,3 @@
-{"dct_accessRights_s": "Access rights", "dct_references_s": "", "dct_title_s": "Test title 1", "gbl_mdModified_dt": "", "gbl_mdVersion_s": "", "gbl_resourceClass_sm": "", "id": "mit:123", "locn_geometry": "ENVELOPE(-111.1, -104.0, 45.0, 40.9)"}
-{"dct_accessRights_s": "Access rights", "dct_references_s": "", "dct_title_s": "Test title 2", "gbl_mdModified_dt": "", "gbl_mdVersion_s": "", "gbl_resourceClass_sm": "", "id": "ogm:456", "locn_geometry": "ENVELOPE(-111.1, -104.0, 45.0, 40.9)"}
+{"dct_accessRights_s": "Access rights", "dct_references_s": "", "dct_title_s": "Test title 1", "gbl_mdModified_dt": "", "gbl_mdVersion_s": "", "gbl_resourceClass_sm": "", "id": "mit:123", "locn_geometry": ""}
+{"dct_accessRights_s": "Access rights", "dct_references_s": "", "dct_title_s": "Test title 2", "gbl_mdModified_dt": "", "gbl_mdVersion_s": "", "gbl_resourceClass_sm": "", "id": "ogm:456", "locn_geometry": ""}
+{"dct_accessRights_s": "Access rights", "dct_references_s": "", "dct_title_s": "Test title 3", "gbl_mdModified_dt": "", "gbl_mdVersion_s": "", "gbl_resourceClass_sm": "", "gbl_suppressed_b": true, "id": "ogm:789", "locn_geometry": ""}

tests/sources/json/test_aardvark.py

@@ -1,9 +1,33 @@
+from pathlib import Path
+
 import pytest
 
 import transmogrifier.models as timdex
 from transmogrifier.sources.json.aardvark import MITAardvark
 
 
+def test_mitaardvark_transform_and_write_output_files_writes_output_files(
+    tmp_path, aardvark_records
+):
+    output_file = str(tmp_path / "output_file.json")
+    transformer = MITAardvark("cool-repo", aardvark_records)
+    assert not Path(tmp_path / "output_file.json").exists()
+    assert not Path(tmp_path / "output_file.txt").exists()
+    transformer.transform_and_write_output_files(output_file)
+    assert Path(tmp_path / "output_file.json").exists()
+    assert Path(tmp_path / "output_file.txt").exists()
+
+
+def test_mitaardvark_transform_and_write_output_files_no_txt_file_if_not_needed(
+    tmp_path, aardvark_record_all_fields
+):
+    output_file = str(tmp_path / "output_file.json")
+    transformer = MITAardvark("cool-repo", aardvark_record_all_fields)
+    transformer.transform_and_write_output_files(output_file)
+    assert len(list(tmp_path.iterdir())) == 1
+    assert next(tmp_path.iterdir()).name == "output_file.json"
+
+
 def test_aardvark_get_required_fields_returns_expected_values(aardvark_records):
     transformer = MITAardvark("cool-repo", aardvark_records)
     assert transformer.get_required_fields(next(aardvark_records)) == {
@@ -43,6 +67,34 @@ def test_aardvark_get_main_titles_success(aardvark_record_all_fields):
     ]
 
 
+def test_aardvark_record_is_deleted_returns_false_if_field_missing(
+    aardvark_record_all_fields,
+):
+    assert MITAardvark.record_is_deleted(next(aardvark_record_all_fields)) is False
+
+
+def test_aardvark_record_is_deleted_returns_false_if_value_is_string(
+    aardvark_record_all_fields,
+):
+    aardvark_record = next(aardvark_record_all_fields)
+    aardvark_record["gbl_supressed_b"] = "True"
+    assert MITAardvark.record_is_deleted(aardvark_record) is False
+
+
+def test_aardvark_record_is_deleted_returns_false_if_value_is_false(
+    aardvark_record_all_fields,
+):
+    aardvark_record = next(aardvark_record_all_fields)
+    aardvark_record["gbl_suppressed_b"] = False
+    assert MITAardvark.record_is_deleted(aardvark_record) is False
+
+
+def test_aardvark_record_is_deleted_success(aardvark_record_all_fields):
+    aardvark_record = next(aardvark_record_all_fields)
+    aardvark_record["gbl_suppressed_b"] = True
+    assert MITAardvark.record_is_deleted(aardvark_record) is True
+
+
 def test_aardvark_get_source_record_id_success(aardvark_record_all_fields):
     assert MITAardvark.get_source_record_id(next(aardvark_record_all_fields)) == "123"
 

transmogrifier/sources/json/aardvark.py

@@ -90,6 +90,8 @@ def record_is_deleted(cls, source_record: dict) -> bool:
         Args:
             source_record: A JSON object representing a source record.
         """
+        if source_record.get("gbl_suppressed_b") is True:
+            return True
         return False
 
     def get_optional_fields(self, source_record: dict) -> dict | None: