rename notice_fetched_date into ted_publication_date for SupraNotice

Author: CaptainOfHacks

ted_sws/core/model/supra_notice.py

@@ -47,6 +47,6 @@ class DailySupraNotice(SupraNotice):
     """
         This is an aggregate over the notices published in TED in a specific day.
     """
-    notice_fetched_date: date
+    ted_publication_date: date
     validation_report: Optional[SupraNoticeValidationReport]
     validation_summary: Optional[ValidationSummaryReport] = None

ted_sws/core/service/__init__.py

@@ -8,7 +8,7 @@
 from ted_sws.data_manager.adapters import inject_date_string_fields, remove_date_string_fields
 from ted_sws.data_manager.adapters.repository_abc import DailySupraNoticeRepositoryABC
 
-DAILY_SUPRA_NOTICE_FETCHED_DATE = "notice_fetched_date"
+DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE = "ted_publication_date"
 DAILY_SUPRA_NOTICE_CREATED_AT = "created_at"
 DAILY_SUPRA_NOTICE_ID = "_id"
 
@@ -26,7 +26,8 @@ def __init__(self, mongodb_client: MongoClient, database_name: str = None):
         daily_supra_notice_db = mongodb_client[self._database_name]
         self.collection = daily_supra_notice_db[self._collection_name]
         self.collection.create_index(
-            [(DAILY_SUPRA_NOTICE_FETCHED_DATE, ASCENDING)])  # TODO: index creation may bring race condition error.
+            [(DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE,
+              ASCENDING)])  # TODO: index creation may bring race condition error.
 
     def _create_dict_from_daily_supra_notice(self, daily_supra_notice: DailySupraNotice) -> dict:
         """
@@ -35,10 +36,11 @@ def _create_dict_from_daily_supra_notice(self, daily_supra_notice: DailySupraNot
         :return:
         """
         daily_supra_notice_dict = daily_supra_notice.dict()
-        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE] = datetime.combine(
-            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE], time())
-        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID] = daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE].isoformat()
-        inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_FETCHED_DATE)
+        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE] = datetime.combine(
+            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE], time())
+        daily_supra_notice_dict[DAILY_SUPRA_NOTICE_ID] = daily_supra_notice_dict[
+            DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE].isoformat()
+        inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE)
         inject_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_CREATED_AT)
         return daily_supra_notice_dict
 
@@ -50,9 +52,10 @@ def _create_daily_supra_notice_from_dict(self, daily_supra_notice_dict: dict) ->
         """
         if daily_supra_notice_dict is not None:
             daily_supra_notice_dict.pop(DAILY_SUPRA_NOTICE_ID, None)
-            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_FETCHED_DATE] = daily_supra_notice_dict[
-                DAILY_SUPRA_NOTICE_FETCHED_DATE].date()
-            remove_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_FETCHED_DATE)
+            daily_supra_notice_dict[DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE] = daily_supra_notice_dict[
+                DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE].date()
+            remove_date_string_fields(data=daily_supra_notice_dict,
+                                      date_field_name=DAILY_SUPRA_NOTICE_TED_PUBLICATION_DATE)
             remove_date_string_fields(data=daily_supra_notice_dict, date_field_name=DAILY_SUPRA_NOTICE_CREATED_AT)
             daily_supra_notice = DailySupraNotice.parse_obj(daily_supra_notice_dict)
             return daily_supra_notice

ted_sws/data_manager/adapters/supra_notice_repository.py

@@ -25,6 +25,6 @@ def test_supra_notice_unpacking_in_file_system():
     supra_notice_repository = DailySupraNoticeRepository(mongodb_client=mongodb_client)
     unpacking_folder = pathlib.Path("./unpacking_supra_notice_result")
     for index, supra_notice in enumerate(itertools.islice(supra_notice_repository.list(), 5)):
-        supra_notice_unpacking_folder = unpacking_folder / f"supra_notice_{supra_notice.notice_fetched_date.strftime('%Y_%m_%d')}"
+        supra_notice_unpacking_folder = unpacking_folder / f"supra_notice_{supra_notice.ted_publication_date.strftime('%Y_%m_%d')}"
         supra_notice_unpacking_folder.mkdir(parents=True, exist_ok=True)
         unpack_supra_notice(supra_notice=supra_notice, unpack_path=supra_notice_unpacking_folder)

tests/e2e/data_manager/_test_supra_notice_unpacking_in_file_system.py

@@ -6,17 +6,17 @@
 def test_daily_supra_notice_repository(mongodb_client, daily_supra_notice):
     daily_supra_notice_repository = DailySupraNoticeRepository(mongodb_client=mongodb_client)
     daily_supra_notice_repository.add(daily_supra_notice=daily_supra_notice)
-    result_supra_notice = daily_supra_notice_repository.get(reference=daily_supra_notice.notice_fetched_date)
+    result_supra_notice = daily_supra_notice_repository.get(reference=daily_supra_notice.ted_publication_date)
     assert result_supra_notice
     assert len(result_supra_notice.notice_ids) == 3
-    assert result_supra_notice.notice_fetched_date == daily_supra_notice.notice_fetched_date
+    assert result_supra_notice.ted_publication_date == daily_supra_notice.ted_publication_date
     assert result_supra_notice.created_at == daily_supra_notice.created_at
     assert result_supra_notice.notice_ids == daily_supra_notice.notice_ids
     daily_supra_notice_repository.update(daily_supra_notice=daily_supra_notice)
-    result_supra_notice = daily_supra_notice_repository.get(reference=daily_supra_notice.notice_fetched_date)
+    result_supra_notice = daily_supra_notice_repository.get(reference=daily_supra_notice.ted_publication_date)
     assert result_supra_notice
     assert len(result_supra_notice.notice_ids) == 3
-    assert result_supra_notice.notice_fetched_date == daily_supra_notice.notice_fetched_date
+    assert result_supra_notice.ted_publication_date == daily_supra_notice.ted_publication_date
     assert result_supra_notice.created_at == daily_supra_notice.created_at
     assert result_supra_notice.notice_ids == daily_supra_notice.notice_ids
     result = list(daily_supra_notice_repository.list())

tests/unit/data_manager/test_supra_notice_repository.py

@@ -6,16 +6,16 @@
 
 def test_daily_supra_notice_manager(mongodb_client, daily_supra_notice_repository):
     notice_ids = ["1", "2", "3"]
-    notice_fetched_date = date(2020, 1, 1)
+    ted_publication_date = date(2020, 1, 1)
     create_and_store_in_mongo_db_daily_supra_notice(notice_ids=notice_ids, mongodb_client=mongodb_client,
-                                                    notice_fetched_date=notice_fetched_date)
+                                                    notice_fetched_date=ted_publication_date)
     for result in daily_supra_notice_repository.list():
         assert result
-    result = daily_supra_notice_repository.get(reference=notice_fetched_date)
+    result = daily_supra_notice_repository.get(reference=ted_publication_date)
     assert result
-    assert result.notice_fetched_date == notice_fetched_date
+    assert result.ted_publication_date == ted_publication_date
     notice_ids.append("4")
     result.notice_ids = notice_ids
     daily_supra_notice_repository.update(daily_supra_notice=result)
-    result = daily_supra_notice_repository.get(reference=notice_fetched_date)
+    result = daily_supra_notice_repository.get(reference=ted_publication_date)
     assert result.notice_ids == notice_ids