Fix 90 key error (#91)

Author: jtyoung84

src/aind_data_asset_indexer/codeocean_bucket_indexer.py

@@ -93,9 +93,9 @@ def _map_external_list_to_dict(external_recs: List[dict]) -> dict:
         """
         new_records = dict()
         for r in external_recs:
-            location = r["location"]
+            location = r.get("source")
             rec_id = r["id"]
-            if new_records.get(location) is not None:
+            if location is not None and new_records.get(location) is not None:
                 old_id_set = new_records.get(location)
                 old_id_set.add(rec_id)
                 new_records[location] = old_id_set

tests/test_codeocean_bucket_indexer.py

@@ -102,9 +102,9 @@ def setUpClass(cls) -> None:
         ]
 
         cls.example_temp_endpoint_response = [
-            {"id": "abc-123", "location": "s3://bucket/prefix1"},
-            {"id": "def-456", "location": "s3://bucket/prefix1"},
-            {"id": "ghi-789", "location": "s3://bucket/prefix2"},
+            {"id": "abc-123", "source": "s3://bucket/prefix1"},
+            {"id": "def-456", "source": "s3://bucket/prefix1"},
+            {"id": "ghi-789", "source": "s3://bucket/prefix2"},
         ]
 
     @patch("requests.get")