def setUp(self): TestCase.setUp(self) User.__password_strategy__ = _CLEAR_PWD if self.CLEAR_PASSWORDS else _DEFAULT_PWD if not self.TESTING_BUILD_ASSETS: extensions = self.app.jinja_env.extensions assets_ext = extensions['webassets.ext.jinja2.AssetsExtension'] assets_ext.BundleClass = NullBundle # session_repository must be started before session is created, as it must # receive all transactions events. It also requires 'repository'. self.app.services['repository'].start() self.app.services['session_repository'].start() self.session = self.db.session() if self.db.engine.name == 'postgresql': # ensure we are on a clean DB: let's use our own schema self.__pg_schema = 'test_{}'.format(str(time()).replace('.', '_')) username = self.db.engine.url.username or getpass.getuser() with self.db.engine.connect() as conn: with conn.begin(): conn.execute('DROP SCHEMA IF EXISTS {} CASCADE'.format( self.__pg_schema)) conn.execute('CREATE SCHEMA {}'.format(self.__pg_schema)) conn.execute('SET search_path TO {}'.format( self.__pg_schema)) conn.execute( 'ALTER ROLE {username} SET search_path TO {schema}' ''.format( username=username, schema=self.__pg_schema)) conn.execute('COMMIT') self.app.create_db() for svc in self.SERVICES: svc = self.app.services[svc] if not svc.running: svc.start()
def setUp(self): TestCase.setUp(self) User.__password_strategy__ = _CLEAR_PWD if self.CLEAR_PASSWORDS else _DEFAULT_PWD if not self.TESTING_BUILD_ASSETS: extensions = self.app.jinja_env.extensions assets_ext = extensions['webassets.ext.jinja2.AssetsExtension'] assets_ext.BundleClass = NullBundle # session_repository must be started before session is created, as it must # receive all transactions events. It also requires 'repository'. self.app.services['repository'].start() self.app.services['session_repository'].start() self.session = self.db.session() if self.db.engine.name == 'postgresql': # ensure we are on a clean DB: let's use our own schema self.__pg_schema = 'test_{}'.format(str(time()).replace('.', '_')) username = self.db.engine.url.username with self.db.engine.connect() as conn: with conn.begin(): conn.execute('DROP SCHEMA IF EXISTS {} CASCADE'.format(self.__pg_schema)) conn.execute('CREATE SCHEMA {}'.format(self.__pg_schema)) conn.execute('SET search_path TO {}'.format(self.__pg_schema)) conn.execute( 'ALTER ROLE {username} SET search_path TO {schema}' ''.format(username=username, schema=self.__pg_schema)) conn.execute('COMMIT') self.app.create_db() for svc in self.SERVICES: svc = self.app.services[svc] if not svc.running: svc.start()
def setUp(self): TestCase.setUp(self) db.create_all()
def setUp(self): BaseTestCase.setUp(self) os.environ.update(self._set_environ())