| | |
| | from __future__ import with_statement |
| | import unittest |
| | from contextlib import contextmanager |
| | from mixin_storage import StorageTestMixin |
| |
|
| |
|
| | class TestFileSystem(unittest.TestCase, StorageTestMixin): |
| |
|
| | @property |
| | def fixture_path(self): |
| | return self.id() |
| |
|
| | def create_fixture_storage(self): |
| | from oxt_tool.storage.fs import FileSystemStorage |
| | return FileSystemStorage(self.fixture_path, 'a') |
| | |
| | @contextmanager |
| | def create_fixture_folder(self): |
| | import os.path |
| | import shutil |
| | path = self.fixture_path |
| | if os.path.exists(path): |
| | shutil.rmtree(path) |
| | os.mkdir(path) |
| | os.mkdir(os.path.join(path, 'bar')) |
| | with file(os.path.join(path, 'bar.txt'), 'w') as f: |
| | f.write('Hello') |
| | with file(os.path.join(path, 'baz.txt'), 'w') as f: |
| | f.write('World') |
| | from oxt_tool.storage.fs import FileSystemFolder |
| | yield FileSystemFolder(path) |
| |
|
| | @contextmanager |
| | def get_fixture_folder(self): |
| | from oxt_tool.storage.fs import FileSystemFolder |
| | yield FileSystemFolder(self.fixture_path) |
| |
|
