def __init__(self, ini, config_file_name, extra_environ, setup_request=False, setup_rdb_context=False, remote_user_id=None): wsgiapp = loadapp('config:' + ini.ini_file_path, name=ini.app_name) if not remote_user_id is None: extra_environ['REMOTE_USER'] = remote_user_id self.__app = EverestTestApp(wsgiapp, ini.package_name, extra_environ=extra_environ) self.__config_file_name = config_file_name if setup_request: kw = dict(application_url=ini.app_url, host_url=ini.app_url, path_url=ini.app_url, url=ini.app_url, registry=self.__app.config.registry) if not remote_user_id is None: self.__app.config.testing_securitypolicy(remote_user_id) kw['environ'] = dict(REMOTE_USER=remote_user_id) self.__request = DummyRequest(**kw) elif not remote_user_id is None: raise ValueError('The remote_user_id argument requires the ' 'setup_request argument to be True.') else: self.__request = None if setup_rdb_context: self.__rdb_context = RdbContextManager() else: self.__rdb_context = None
def _test_load(self, num_entities=2): with RdbContextManager() as session: query = session.query(self.entity_class) entities = query.limit(num_entities).all() self.assert_equal(len(entities), num_entities) for entity in entities: self.assert_equal(entity.__class__, self.entity_class)
def __init__(self, ini, config_file_name, extra_environ, setup_request=False, setup_rdb_context=False, remote_user_id=None): wsgiapp = loadapp('config:' + ini.ini_file_path, name=ini.app_name) if not remote_user_id is None: extra_environ['REMOTE_USER'] = remote_user_id self.__app = EverestTestApp(wsgiapp, ini.package_name, extra_environ=extra_environ) self.__config_file_name = config_file_name if setup_request: kw = dict(application_url=ini.app_url, host_url=ini.app_url, path_url=ini.app_url, url=ini.app_url, registry=self.__app.config.registry) if not remote_user_id is None: self.__app.config.testing_securitypolicy(remote_user_id) kw['environ'] = dict(REMOTE_USER=remote_user_id) self.__request = DummyRequest(**kw) elif not remote_user_id is None: raise ValueError('The remote_user_id argument requires the ' 'setup_request argument to be True.') else: self.__request = None if setup_rdb_context: self.__rdb_context = RdbContextManager() else: self.__rdb_context = None
def _test_persist(self): with RdbContextManager() as session: attrs = self._get_data() persist(session, self.entity_class, attrs, True)
def nested_session(): with RdbContextManager() as sess: yield sess
class AppCreatorContextManager(object): """ Context manager for setting up the everest framework for application level (functional) operations. """ def __init__(self, ini, config_file_name, extra_environ, setup_request=False, setup_rdb_context=False, remote_user_id=None): wsgiapp = loadapp('config:' + ini.ini_file_path, name=ini.app_name) if not remote_user_id is None: extra_environ['REMOTE_USER'] = remote_user_id self.__app = EverestTestApp(wsgiapp, ini.package_name, extra_environ=extra_environ) self.__config_file_name = config_file_name if setup_request: kw = dict(application_url=ini.app_url, host_url=ini.app_url, path_url=ini.app_url, url=ini.app_url, registry=self.__app.config.registry) if not remote_user_id is None: self.__app.config.testing_securitypolicy(remote_user_id) kw['environ'] = dict(REMOTE_USER=remote_user_id) self.__request = DummyRequest(**kw) elif not remote_user_id is None: raise ValueError('The remote_user_id argument requires the ' 'setup_request argument to be True.') else: self.__request = None if setup_rdb_context: self.__rdb_context = RdbContextManager() else: self.__rdb_context = None def __enter__(self): self.__app.config.begin(request=self.__request) if not self.__request is None: srvc = self.__app.config.get_registered_utility(IService) self.__request.root = srvc if not self.__config_file_name is None: self.__app.config.load_zcml(self.__config_file_name) repo_mgr = \ self.__app.config.get_registered_utility(IRepositoryManager) repo_mgr.initialize_all() if not self.__request is None: srvc.start() if not self.__rdb_context is None: self.__rdb_context.__enter__() return self.__app def __exit__(self, ext_type, value, tb): if self.__rdb_context is None: transaction.abort() else: self.__rdb_context.__exit__(ext_type, value, tb) if not self.__request is None: srvc = self.__app.config.get_registered_utility(IService) srvc.stop() repo_mgr = \ self.__app.config.get_registered_utility(IRepositoryManager) repo_mgr.reset_all() self.__app.config.end() tear_down_registry(self.__app.config.registry)
class AppCreatorContextManager(object): """ Context manager for setting up the everest framework for application level (functional) operations. """ def __init__(self, ini, config_file_name, extra_environ, setup_request=False, setup_rdb_context=False, remote_user_id=None): wsgiapp = loadapp('config:' + ini.ini_file_path, name=ini.app_name) if not remote_user_id is None: extra_environ['REMOTE_USER'] = remote_user_id self.__app = EverestTestApp(wsgiapp, ini.package_name, extra_environ=extra_environ) self.__config_file_name = config_file_name if setup_request: kw = dict(application_url=ini.app_url, host_url=ini.app_url, path_url=ini.app_url, url=ini.app_url, registry=self.__app.config.registry) if not remote_user_id is None: self.__app.config.testing_securitypolicy(remote_user_id) kw['environ'] = dict(REMOTE_USER=remote_user_id) self.__request = DummyRequest(**kw) elif not remote_user_id is None: raise ValueError('The remote_user_id argument requires the ' 'setup_request argument to be True.') else: self.__request = None if setup_rdb_context: self.__rdb_context = RdbContextManager() else: self.__rdb_context = None def __enter__(self): self.__app.config.begin(request=self.__request) if not self.__request is None: srvc = self.__app.config.get_registered_utility(IService) self.__request.root = srvc if not self.__config_file_name is None: self.__app.config.load_zcml(self.__config_file_name) repo_mgr = \ self.__app.config.get_registered_utility(IRepositoryManager) repo_mgr.initialize_all() if not self.__request is None: srvc.start() if not self.__rdb_context is None: self.__rdb_context.__enter__() return self.__app def __exit__(self, ext_type, value, tb): if self.__rdb_context is None: transaction.abort() else: self.__rdb_context.__exit__(ext_type, value, tb) if not self.__request is None: srvc = self.__app.config.get_registered_utility(IService) srvc.stop() repo_mgr = \ self.__app.config.get_registered_utility(IRepositoryManager) repo_mgr.reset_all() self.__app.config.end() tear_down_registry(self.__app.config.registry)