def test_link_documents_to_documentsbundle_calls_link_documents_to_documentsbundle_operation( self, mk_link_documents): documents = [{ 'scielo_id': 'JV5Lb3v3HBYmPPdG6QD9jGQ', 'issn': '1806-907X', 'year': '2003', 'order': '00001', 'xml_package_name': '1806-907X-rba-53-01-1-8', 'assets': [], 'pdfs': [{ 'lang': 'pt', 'filename': '1806-907X-rba-53-01-1-8.pdf', 'mimetype': 'application/pdf' }], 'volume': '53', 'number': '01', 'xml_url': 'http://192.168.169.185:9000/documentstore/1806-907X/JV5Lb3v3HBYmPPdG6QD9jGQ/e8a6df175375a6f922cf8a3bf2ef4a0ce2b09c93.xml' }] kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} kwargs["ti"].xcom_pull.return_value = documents link_documents_to_documentsbundle(**kwargs) mk_link_documents.assert_called_once_with(documents)
def test_link_documents_to_documentsbundle_gets_sps_package_from_dag_run_conf( self, mk_link_documents): mk_dag_run = MagicMock() kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} mk_link_documents.return_value = [], [] link_documents_to_documentsbundle(**kwargs) mk_dag_run.conf.get.assert_called_once_with("sps_package")
def test_link_documents_to_documentsbundle_gets_ti_xcom_documents( self, mk_link_documents): kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} link_documents_to_documentsbundle(**kwargs) kwargs["ti"].xcom_pull.assert_called_once_with( key="documents", task_ids="register_update_docs_id")
def test_link_documents_to_documentsbundle_gets_ti_xcom_documents( self, mk_link_documents): kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} mk_link_documents.return_value = [], [] link_documents_to_documentsbundle(**kwargs) kwargs["ti"].xcom_pull.assert_any_call( key="documents", task_ids="register_update_docs_id")
def test_link_documents_to_documentsbundle_empty_ti_xcom_documents( self, mk_link_documents): kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} kwargs["ti"].xcom_pull.return_value = None link_documents_to_documentsbundle(**kwargs) mk_link_documents.assert_not_called() kwargs["ti"].xcom_push.assert_not_called()
def test_link_documents_to_documentsbundle_gets_ti_xcom_title_json_path( self, mk_link_documents): kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} link_documents_to_documentsbundle(**kwargs) kwargs["ti"].xcom_pull.assert_any_call( task_ids="process_journals_task", dag_id="kernel-gate", key="issn_index_json_path", include_prior_dates=True)
def test_link_documents_to_documentsbundle_does_not_push_if_no_documents( self, mk_link_documents): documents = [] kwargs = {"ti": MagicMock(), "dag_run": MagicMock()} kwargs["ti"].xcom_pull.return_value = documents mk_link_documents.return_value = [] link_documents_to_documentsbundle(**kwargs) kwargs["ti"].xcom_push.assert_not_called()