class OratorTestCase(TestCase): def tearDown(self): if hasattr(self, "local_database"): os.remove(self.local_database) flexmock_teardown() def init_database(self): self.local_database = "/tmp/orator_test_database.db" if os.path.exists(self.local_database): os.remove(self.local_database) self.manager = DatabaseManager({ "default": "sqlite", "sqlite": { "driver": "sqlite", "database": self.local_database }, }) with self.manager.transaction(): try: self.manager.statement( "CREATE TABLE `users` (" "id INTEGER PRIMARY KEY NOT NULL, " "name CHAR(50) NOT NULL, " "created_at DATETIME DEFAULT CURRENT_TIMESTAMP, " "updated_at DATETIME DEFAULT CURRENT_TIMESTAMP" ")") except Exception: pass Model.set_connection_resolver(self.manager) self.manager.disconnect() def assertRegex(self, *args, **kwargs): if PY2: return self.assertRegexpMatches(*args, **kwargs) else: return super(OratorTestCase, self).assertRegex(*args, **kwargs) def assertNotRegex(self, *args, **kwargs): if PY2: return self.assertNotRegexpMatches(*args, **kwargs) else: return super(OratorTestCase, self).assertNotRegex(*args, **kwargs)
class OratorTestCase(TestCase): def tearDown(self): if hasattr(self, 'local_database'): os.remove(self.local_database) def init_database(self): self.local_database = '/tmp/orator_test_database.db' if os.path.exists(self.local_database): os.remove(self.local_database) self.manager = DatabaseManager({ 'default': 'sqlite', 'sqlite': { 'driver': 'sqlite', 'database': self.local_database } }) with self.manager.transaction(): try: self.manager.statement( 'CREATE TABLE `users` (' 'id INTEGER PRIMARY KEY NOT NULL, ' 'name CHAR(50) NOT NULL, ' 'created_at DATETIME DEFAULT CURRENT_TIMESTAMP, ' 'updated_at DATETIME DEFAULT CURRENT_TIMESTAMP' ')' ) except Exception: pass Model.set_connection_resolver(self.manager) self.manager.disconnect() def assertRegex(self, *args, **kwargs): if PY2: return self.assertRegexpMatches(*args, **kwargs) else: return super(OratorTestCase, self).assertRegex(*args, **kwargs) def assertNotRegex(self, *args, **kwargs): if PY2: return self.assertNotRegexpMatches(*args, **kwargs) else: return super(OratorTestCase, self).assertNotRegex(*args, **kwargs)