updated METS package publishing + tests

Author: kaleanych

dags/pipelines/notice_processor_pipelines.py

@@ -84,7 +84,7 @@ def notice_validation_pipeline(notice: Notice, mongodb_client: MongoClient) -> N
     return NoticePipelineOutput(notice=notice)
 
 
-def notice_package_pipeline(notice: Notice) -> NoticePipelineOutput:
+def notice_package_pipeline(notice: Notice, mongodb_client: MongoClient = None) -> NoticePipelineOutput:
     """
 
     """
@@ -101,7 +101,7 @@ def notice_package_pipeline(notice: Notice) -> NoticePipelineOutput:
     return NoticePipelineOutput(notice=packaged_notice)
 
 
-def notice_publish_pipeline(notice: Notice) -> NoticePipelineOutput:
+def notice_publish_pipeline(notice: Notice, mongodb_client: MongoClient = None) -> NoticePipelineOutput:
     """
 
     """

tests/features/notice_publisher/conftest.py

@@ -30,7 +30,9 @@ def publish_eligible_notice(publicly_available_notice) -> Notice:
     notice = publicly_available_notice
     notice.update_status_to(NoticeStatus.ELIGIBLE_FOR_PUBLISHING)
     notice._mets_manifestation = METSManifestation(
-        object_data=base64.b64encode("METS manifestation content".encode("utf-8")))
+        object_data=base64.b64encode("METS manifestation content".encode("utf-8")),
+        package_name="test_package.zip"
+    )
     return notice
 
 

tests/features/notice_publisher/test_notice_publisher.py

@@ -10,7 +10,6 @@
 from ted_sws.core.model.notice import Notice, NoticeStatus
 from ted_sws.data_manager.adapters.repository_abc import NoticeRepositoryABC
 from ted_sws.notice_publisher.adapters.sftp_notice_publisher import SFTPPublisher
-from ted_sws.notice_publisher.adapters.sftp_publisher_abc import SFTPPublisherABC
 from ted_sws.notice_publisher.services.notice_publisher import publish_notice, publish_notice_by_id