Update test_daily_supra_notice_manager.py

Author: CaptainOfHacks

tests/unit/supra_notice_manager/test_daily_supra_notice_manager.py

@@ -1,12 +1,22 @@
-from datetime import datetime, time
+from datetime import datetime, time, date
 
-from ted_sws.data_manager.adapters.supra_notice_repository import DailySupraNoticeRepository
 from ted_sws.supra_notice_manager.services.daily_supra_notice_manager import \
     create_and_store_in_mongo_db_daily_supra_notice
 
 
 def test_daily_supra_notice_manager(mongodb_client, daily_supra_notice_repository):
     notice_ids = ["1", "2", "3"]
-    create_and_store_in_mongo_db_daily_supra_notice(notice_ids=notice_ids, mongodb_client=mongodb_client)
-    result = daily_supra_notice_repository.get(reference=datetime.combine(datetime.today(), time()))
+    notice_publication_date = date(2020, 1, 1)
+    create_and_store_in_mongo_db_daily_supra_notice(notice_ids=notice_ids, mongodb_client=mongodb_client,
+                                                    notice_publication_date=notice_publication_date
+                                                    )
+    for result in daily_supra_notice_repository.list():
+        assert result
+    result = daily_supra_notice_repository.get(reference=notice_publication_date)
     assert result
+    assert result.notice_publication_date == notice_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_publication_date)
+    assert result.notice_ids == notice_ids