import py, os, stat
import zipfile, zipfilepart

udir = py.path.local.make_numbered_dir(prefix = 'usession-')


class TestZipFilePart:

    def setup_class(cls):
        cls.filename = str(udir.join('test_zipfilepart.zip'))
        cls.content = {}
        f = zipfile.ZipFile(cls.filename, 'w',
                            compression=zipfile.ZIP_DEFLATED)
        for fn in os.listdir('.'):
            if stat.S_ISREG(os.lstat(fn).st_mode):
                f.write(fn)
                cls.content[fn] = None
        f.close()

    def test_read(self):
        files = list(zipfilepart.enum_zip_file(self.filename))
        assert dict.fromkeys(dict(files)) == self.content
        for key, value in files:
            assert value == open(key, 'rb').read()