def test_zip_directory_bad_id(): zipped_directory = dz.zip_directory('AHRQ_FRDOC_0001-0037', REGULATIONS_PATH + BAD_AHRQ_PATH) archive = zipfile.ZipFile(zipped_directory.filename, 'r') readme_file = bytes.decode(archive.read('README.md')) assert readme_file == 'This file AHRQ_FRDOC_0001-0037 does not exist' os.remove(zipped_directory.filename)
def download_zip(document_id, download_path=PATH): zip_path = ds.search_for_document_in_directory(document_id, download_path) if zip_path != "": zipfile = dz.zip_directory(document_id, zip_path) return zipfile.filename else: return ""
def download_zip(document_id, download_path=HOME_REGULATION_PATH): zip_path = ds.search_for_document_in_directory(document_id, download_path) if zip_path != '': zipfile = dz.zip_directory(document_id, zip_path) return zipfile.filename else: return ''
def test_zip_directory(): zipped_directory = dz.zip_directory('AHRQ_FRDOC_0001-0036', REGULATIONS_PATH + AHRQ_PATH) archive = zipfile.ZipFile(zipped_directory.filename, 'r') readme_file = bytes.decode(archive.read('README.md')) json_file = bytes.decode( archive.read('AHRQ_FRDOC_0001-0036/doc.AHRQ_FRDOC_0001-0036.json')) assert readme_file == 'This is a README file for AHRQ_FRDOC_0001-0036' assert json_file == '{}' os.remove(zipped_directory.filename)