| | |
| | from __future__ import with_statement |
| | import unittest |
| | from oxt_tool.storage import _zipfile |
| | from mixin_storage import StorageTestMixin |
| | import contextlib |
| |
|
| |
|
| | class TestZipFile(unittest.TestCase, StorageTestMixin): |
| |
|
| | @property |
| | def zipfile_path(self): |
| | return self.id() + '.zip' |
| |
|
| | def zipfile_create(self): |
| | import zipfile |
| | return zipfile.ZipFile(self.zipfile_path, 'w') |
| |
|
| | def zipfile_get(self): |
| | import zipfile |
| | return zipfile.ZipFile(self.zipfile_path, 'r') |
| |
|
| | def test_zipfile_folder(self): |
| | import os |
| | zf = self.zipfile_create() |
| | try: |
| | zf.writestr(os.sep.join(['foo', 'bar.txt']), 'Hello') |
| | _zipfile.ZipFileFolder(zf, 'foo') |
| | finally: |
| | zf.close() |
| |
|
| | def test_zipfile_file(self): |
| | import os |
| | zf = self.zipfile_create() |
| | try: |
| | path = os.sep.join(['foo', 'bar.txt']) |
| | zf.writestr(path, 'Hello') |
| | _zipfile.ZipFileFile(zf, path) |
| | finally: |
| | zf.close() |
| |
|
| | def test_zipfile_file_put(self): |
| | path = self.id() + '.txt' |
| | with file(path, 'w') as f: |
| | f.write('new-file-content') |
| | with self.create_fixture_folder() as folder: |
| | folder.file('new-file').put(path) |
| | with self.get_fixture_folder() as folder: |
| | with folder['new-file'].open() as f: |
| | self.assertEqual('new-file-content', f.read()) |
| |
|
| | def create_fixture_storage(self): |
| | return _zipfile.ZipFileStorage(self.zipfile_path, 'w') |
| |
|
| | @contextlib.contextmanager |
| | def create_fixture_zipfile(self): |
| | import os |
| | zf = self.zipfile_create() |
| | try: |
| | zf.writestr(os.sep.join(['foo', 'bar.txt']), 'Hello') |
| | zf.writestr(os.sep.join(['foo', 'baz.txt']), 'World') |
| | zf.writestr(os.sep.join(['foo', 'bar', 'baz']), 'Hello World') |
| | yield zf |
| | finally: |
| | zf.close() |
| |
|
| | @contextlib.contextmanager |
| | def create_fixture_folder(self): |
| | with self.create_fixture_zipfile() as zf: |
| | yield _zipfile.ZipFileFolder(zf, 'foo') |
| |
|
| | @contextlib.contextmanager |
| | def get_fixture_folder(self): |
| | zf = self.zipfile_get() |
| | try: |
| | yield _zipfile.ZipFileFolder(zf, 'foo') |
| | finally: |
| | zf.close() |
| |
|
| | def test_zipfile_nodes(self): |
| | import os.path |
| | from oxt_tool.storage._zipfile import zipfile_nodes |
| | with self.create_fixture_zipfile() as zipfile: |
| | nodes = dict(zipfile_nodes(zipfile)) |
| | self.assertEqual(set(['foo', |
| | os.path.join('foo', 'bar'), |
| | os.path.join('foo', 'bar.txt'), |
| | os.path.join('foo', 'baz.txt'), |
| | os.path.join('foo', 'bar', 'baz')]), |
| | set(nodes.keys())) |
| | self.assertTrue(hasattr(nodes['foo'], '__getitem__')) |
| | self.assertTrue(hasattr(nodes[os.path.join('foo', 'bar')], |
| | '__getitem__')) |
| | self.assertTrue(hasattr(nodes[os.path.join('foo', 'bar.txt')], |
| | 'open')) |
| | self.assertTrue(hasattr(nodes[os.path.join('foo', 'baz.txt')], |
| | 'open')) |
| | self.assertTrue(hasattr(nodes[os.path.join('foo', 'bar', 'baz')], |
| | 'open')) |
| |
|
