Example #1
0
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 ''
Example #4
0
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)