Changeset View
Changeset View
Standalone View
Standalone View
swh/core/tests/db_testing.py
Show First 20 Lines • Show All 221 Lines • ▼ Show 20 Lines | def setUpClass(cls): | ||||
super().setUpClass() | super().setUpClass() | ||||
@classmethod | @classmethod | ||||
def tearDownClass(cls): | def tearDownClass(cls): | ||||
super().tearDownClass() | super().tearDownClass() | ||||
for name, context in cls._DB_LIST.items(): | for name, context in cls._DB_LIST.items(): | ||||
context.__exit__() | context.__exit__() | ||||
def setUp(self): | def setUp(self, *args, **kwargs): | ||||
self.test_db = {} | self.test_db = {} | ||||
for name in self._DB_LIST.keys(): | for name in self._DB_LIST.keys(): | ||||
self.test_db[name] = DbTestConn(name) | self.test_db[name] = DbTestConn(name) | ||||
self.test_db[name].__enter__() | self.test_db[name].__enter__() | ||||
super().setUp() | super().setUp(*args, **kwargs) | ||||
def tearDown(self): | def tearDown(self): | ||||
super().tearDown() | super().tearDown() | ||||
for name in self._DB_LIST.keys(): | for name in self._DB_LIST.keys(): | ||||
self.test_db[name].__exit__() | self.test_db[name].__exit__() | ||||
def reset_db_tables(self, name, excluded=None): | def reset_db_tables(self, name, excluded=None): | ||||
db = self.test_db[name] | db = self.test_db[name] | ||||
▲ Show 20 Lines • Show All 57 Lines • ▼ Show 20 Lines | def setUpClass(cls): | ||||
all_dump_files = [(x, DB_DUMP_TYPES[os.path.splitext(x)[1]]) | all_dump_files = [(x, DB_DUMP_TYPES[os.path.splitext(x)[1]]) | ||||
for x in all_dump_files] | for x in all_dump_files] | ||||
cls.add_db(name=cls.TEST_DB_NAME, | cls.add_db(name=cls.TEST_DB_NAME, | ||||
dumps=all_dump_files) | dumps=all_dump_files) | ||||
super().setUpClass() | super().setUpClass() | ||||
def setUp(self): | def setUp(self, *args, **kwargs): | ||||
super().setUp() | super().setUp(*args, **kwargs) | ||||
db = self.test_db[self.TEST_DB_NAME] | db = self.test_db[self.TEST_DB_NAME] | ||||
self.conn = db.conn | self.conn = db.conn | ||||
self.cursor = db.cursor | self.cursor = db.cursor |