def list_documents(dag_run, **kwargs): _sps_package = dag_run.conf.get("sps_package") _xmls_filenames = sync_documents_to_kernel_operations.list_documents(_sps_package) if _xmls_filenames: kwargs["ti"].xcom_push(key="xmls_filenames", value=_xmls_filenames) return True else: return False
def test_list_document_empty_list_if_no_xml_in_zip(self, MockZipFile): sps_package_file_lists = [ "v53n1a01.pdf", "0123-4567-abc-50-1-8-gpn1a01t1.htm", "0123-4567-abc-50-1-8-gpn1a01g1.htm", "v53n1a02.pdf", ] MockZipFile.return_value.__enter__.return_value.namelist.return_value = ( sps_package_file_lists) result = list_documents(self.sps_package) self.assertEqual(result, [])
def test_list_document_reads_all_xmls_from_zip(self, MockZipFile): sps_package_file_lists = [ "0123-4567-abc-50-1-8.xml", "v53n1a01.pdf", "0123-4567-abc-50-1-8-gpn1a01t1.htm", "0123-4567-abc-50-1-8-gpn1a01g1.htm", "0123-4567-abc-50-9-18.xml", "v53n1a02.pdf", ] MockZipFile.return_value.__enter__.return_value.namelist.return_value = ( sps_package_file_lists) result = list_documents(self.sps_package) self.assertEqual( result, ["0123-4567-abc-50-1-8.xml", "0123-4567-abc-50-9-18.xml"])
def test_list_document_opens_all_zips(self, MockZipFile): list_documents(self.sps_package) MockZipFile.assert_called_once_with(self.sps_package)