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