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
