Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_api_client.py
| Show All 17 Lines | |||||
| import swh.storage.storage | import swh.storage.storage | ||||
| from swh.storage.tests.test_storage import \ | from swh.storage.tests.test_storage import \ | ||||
| CommonTestStorage, CommonPropTestStorage, StorageTestDbFixture | CommonTestStorage, CommonPropTestStorage, StorageTestDbFixture | ||||
| class RemotePgStorageFixture(StorageTestDbFixture, ServerTestFixture, | class RemotePgStorageFixture(StorageTestDbFixture, ServerTestFixture, | ||||
| unittest.TestCase): | unittest.TestCase): | ||||
| def setUp(self): | def setUp(self): | ||||
| journal_writer = get_journal_writer(cls='inmemory') | journal_writer = get_journal_writer(cls='memory') | ||||
| def mock_get_journal_writer(cls, args=None): | def mock_get_journal_writer(cls, args=None): | ||||
| assert cls == 'inmemory' | assert cls == 'memory' | ||||
| return journal_writer | return journal_writer | ||||
| self.journal_writer = journal_writer | self.journal_writer = journal_writer | ||||
| server.storage = None | server.storage = None | ||||
| self.get_journal_writer = get_journal_writer | self.get_journal_writer = get_journal_writer | ||||
| swh.storage.storage.get_journal_writer = mock_get_journal_writer | swh.storage.storage.get_journal_writer = mock_get_journal_writer | ||||
| # ServerTestFixture needs to have self.objroot for | # ServerTestFixture needs to have self.objroot for | ||||
| Show All 11 Lines | def setUp(self): | ||||
| 'objstorage': { | 'objstorage': { | ||||
| 'cls': 'pathslicing', | 'cls': 'pathslicing', | ||||
| 'args': { | 'args': { | ||||
| 'root': self.storage_base, | 'root': self.storage_base, | ||||
| 'slicing': '0:2', | 'slicing': '0:2', | ||||
| }, | }, | ||||
| }, | }, | ||||
| 'journal_writer': { | 'journal_writer': { | ||||
| 'cls': 'inmemory', | 'cls': 'memory', | ||||
| } | } | ||||
| } | } | ||||
| } | } | ||||
| } | } | ||||
| self.app = app | self.app = app | ||||
| super().setUp() | super().setUp() | ||||
| self.storage = RemoteStorage(self.url()) | self.storage = RemoteStorage(self.url()) | ||||
| Show All 10 Lines | |||||
| class RemoteMemStorageFixture(ServerTestFixture, unittest.TestCase): | class RemoteMemStorageFixture(ServerTestFixture, unittest.TestCase): | ||||
| def setUp(self): | def setUp(self): | ||||
| self.config = { | self.config = { | ||||
| 'storage': { | 'storage': { | ||||
| 'cls': 'memory', | 'cls': 'memory', | ||||
| 'args': { | 'args': { | ||||
| 'journal_writer': { | 'journal_writer': { | ||||
| 'cls': 'inmemory', | 'cls': 'memory', | ||||
| } | } | ||||
| } | } | ||||
| } | } | ||||
| } | } | ||||
| self.__storage = InMemoryStorage( | self.__storage = InMemoryStorage( | ||||
| journal_writer={'cls': 'inmemory'}) | journal_writer={'cls': 'memory'}) | ||||
| self._get_storage_patcher = unittest.mock.patch( | self._get_storage_patcher = unittest.mock.patch( | ||||
| 'swh.storage.api.server.get_storage', return_value=self.__storage) | 'swh.storage.api.server.get_storage', return_value=self.__storage) | ||||
| self._get_storage_patcher.start() | self._get_storage_patcher.start() | ||||
| self.app = app | self.app = app | ||||
| super().setUp() | super().setUp() | ||||
| self.storage = RemoteStorage(self.url()) | self.storage = RemoteStorage(self.url()) | ||||
| self.journal_writer = self.__storage.journal_writer | self.journal_writer = self.__storage.journal_writer | ||||
| ▲ Show 20 Lines • Show All 48 Lines • Show Last 20 Lines | |||||