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)
