def test_delete_documents_empty_ti_xcom_info(self, mk_delete_documents): mk_dag_run = MagicMock() kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} kwargs["ti"].xcom_pull.return_value = None delete_documents(**kwargs) mk_delete_documents.assert_not_called() kwargs["ti"].xcom_push.assert_not_called()
def test_delete_documents_gets_ti_xcom_info(self, mk_delete_documents): mk_dag_run = MagicMock() kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} mk_delete_documents.return_value = [], [] delete_documents(**kwargs) kwargs["ti"].xcom_pull.assert_called_once_with( key="xmls_filenames", task_ids="list_docs_task_id")
def test_delete_documents_gets_sps_package_from_dag_run_conf( self, mk_delete_documents): mk_dag_run = MagicMock() kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} mk_delete_documents.return_value = [], [] delete_documents(**kwargs) mk_dag_run.conf.get.assert_called_once_with("sps_package")
def test_delete_documents_calls_delete_documents_operation( self, mk_delete_documents): xmls_filenames = [ "1806-907X-rba-53-01-1-8.xml", "1806-907X-rba-53-01-9-18.xml", "1806-907X-rba-53-01-19-25.xml", ] mk_dag_run = MagicMock() mk_dag_run.conf.get.return_value = "path_to_sps_package/package.zip" kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} kwargs["ti"].xcom_pull.return_value = xmls_filenames delete_documents(**kwargs) mk_delete_documents.assert_called_once_with( "path_to_sps_package/package.zip", xmls_filenames)
def test_delete_documents_pushes_xmls_to_preserve(self, mk_delete_documents): xmls_filenames = [ "1806-907X-rba-53-01-1-8.xml", "1806-907X-rba-53-01-9-18.xml", "1806-907X-rba-53-01-19-25.xml", ] xmls_to_preserve = [ "1806-907X-rba-53-01-9-18.xml", "1806-907X-rba-53-01-19-25.xml", ] mk_dag_run = MagicMock() mk_dag_run.conf.get.return_value = "path_to_sps_package/package.zip" kwargs = {"ti": MagicMock(), "dag_run": mk_dag_run} kwargs["ti"].xcom_pull.return_value = xmls_filenames mk_delete_documents.return_value = xmls_to_preserve delete_documents(**kwargs) kwargs["ti"].xcom_push.assert_called_once_with(key="xmls_to_preserve", value=xmls_to_preserve)