| |
| 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) |
|
|
