def test_archive_filename():
expected = '/data/guide/lol/pics.zip'
arg = '/data/guide/lol/result.json'
result = archive_filename(arg)
assert result == expected
return
def test_remove_zip():
"""
must remove the given filenames from the archive
"""
guide_name = '/tmp/test_archive/result.json'
removed_pic_name = ['hello.txt']
# call the function.
remove_from_zip(guide_name, removed_pic_name)
# now get the content of the archive and make sure it only contains
# world.txt
archive_name = archive_filename(guide_name)
content = None
with ZipFile(archive_name, 'r') as z:
content = z.namelist()
assert content == ['world.txt']
return
