def setUp(self):
super(TestSecuredRestBase, self).setUp()
postgresql.setup_app()
default_tenant = Tenant.query.filter_by(
name=instance.default_tenant_name).first()
add_users_and_roles_to_userstore(user_datastore,
users=get_test_users(),
roles=get_test_roles(),
default_tenant=default_tenant)
def _add_users_and_roles(self, user_datastore):
"""Add users and roles for the test
:param user_datastore: SQLAlchemyDataUserstore
"""
# Add a fictitious admin user to the user_datastore
utils.add_users_and_roles_to_userstore(
user_datastore,
get_admin_user(),
get_admin_role()
)
def _add_users_and_roles(config, default_tenant):
print 'Adding users and roles to the DB'
# Need to load security_configuration with a yaml loader, as it's a string
security_config = yaml.load(config['security_configuration'])
userstore = security_config.get('userstore', {})
add_users_and_roles_to_userstore(
user_datastore,
users=userstore.get('users', []),
roles=userstore.get('roles', []),
default_tenant=default_tenant
)
print 'Users and roles added successfully'
def reset_data():
logger.info('Resetting PostgreSQL DB')
setup_app()
# Rebuild the DB
db.drop_all()
db.create_all()
add_users_and_roles_to_userstore(user_datastore, get_admin_user(),
get_admin_role())
# Clear the connection
db.session.remove()
db.get_engine(app).dispose()
def reset_data():
logger.info('Resetting PostgreSQL DB')
setup_app()
# Rebuild the DB
db.drop_all()
db.create_all()
default_tenant = Tenant(name=instance.default_tenant_name)
db.session.add(default_tenant)
add_users_and_roles_to_userstore(user_datastore, utils.get_admin_user(),
get_admin_role(), default_tenant)
# Clear the connection
db.session.remove()
db.get_engine(app).dispose()
def _add_users_and_roles(self, user_datastore, default_tenant):
"""Add users and roles for the test
:param user_datastore: SQLAlchemyDataUserstore
"""
# Add a fictitious admin user to the user_datastore
utils.add_users_and_roles_to_userstore(
user_datastore,
self._get_users(),
self._get_roles(),
default_tenant
)
for user in self._get_users():
user = User.query.filter_by(username=user['username']).first()
tenant = self.sm.get_tenant_by_name(DEFAULT_TENANT_NAME)
if tenant not in user.tenants:
user.tenants.append(tenant)
self.sm.update_entity(user)
def _add_users_and_roles(self, user_datastore):
# Override the parent method to add more users/roles to the userstore
add_users_and_roles_to_userstore(user_datastore, get_test_users(),
get_test_roles())