Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/storage_testing.py
| Show First 20 Lines • Show All 49 Lines • ▼ Show 20 Lines | def setUp(self): | ||||
| }, | }, | ||||
| }, | }, | ||||
| }, | }, | ||||
| } | } | ||||
| self.storage = get_storage(**self.storage_config) | self.storage = get_storage(**self.storage_config) | ||||
| def tearDown(self): | def tearDown(self): | ||||
| self.objtmp.cleanup() | self.objtmp.cleanup() | ||||
| self.storage = None | |||||
| super().tearDown() | super().tearDown() | ||||
| def reset_storage_tables(self): | def reset_storage_tables(self): | ||||
| excluded = {'dbversion', 'entity', 'entity_history', 'listable_entity', | excluded = {'dbversion', 'entity', 'entity_history', 'listable_entity', | ||||
| 'tool'} | 'tool'} | ||||
| self.reset_db_tables(self.TEST_STORAGE_DB_NAME, excluded=excluded) | self.reset_db_tables(self.TEST_STORAGE_DB_NAME, excluded=excluded) | ||||
| db = self.test_db[self.TEST_STORAGE_DB_NAME] | db = self.test_db[self.TEST_STORAGE_DB_NAME] | ||||
| conn = db.conn | conn = db.conn | ||||
| cursor = db.cursor | cursor = db.cursor | ||||
| cursor.execute('delete from entity where generated=true') | cursor.execute('delete from entity where generated=true') | ||||
| cursor.execute('delete from entity_history where generated=true') | cursor.execute('delete from entity_history where generated=true') | ||||
| conn.commit() | conn.commit() | ||||