def run(self):
name = prompt('Role Name')
description = prompt('Role Description')
_security_datastore = LocalProxy(lambda: current_app.extensions['security'].datastore)
_security_datastore.create_role(name=name, description=description)
db.session.commit()
return
def test_save_patron(get_uuid_pid_by_email, reindex, record_indexer, url_for1,
url_for2, send_email, confirm_user, app, db,
minimal_patron_only_record):
"""Test save patron"""
# Convenient references
security = LocalProxy(lambda: app.extensions['security'])
datastore = LocalProxy(lambda: security.datastore)
datastore.create_role(name='patrons')
# hack the return value
get_uuid_pid_by_email.return_value = None, None
email = '*****@*****.**'
u1 = datastore.create_user(email=email,
active=False,
password=hash_password('aafaf4as5fa'))
datastore.commit()
u2 = datastore.find_user(email=email)
assert u1 == u2
assert 1 == User.query.filter_by(email=email).count()
email = minimal_patron_only_record.get('email')
assert datastore.get_user(email) is None
save_patron(minimal_patron_only_record, Patron.provider.pid_type,
Patron.fetcher, Patron.minter, record_indexer, Patron, None)
email = minimal_patron_only_record.get('email')
# Verify that user exists in app's datastore
user_ds = datastore.get_user(email)
assert user_ds
assert user_ds.email == email
def run(self):
name = prompt('Role Name')
description = prompt('Role Description')
_security_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
_security_datastore.create_role(name=name, description=description)
db.session.commit()
return
def create_user_with_role(username, rolename):
_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
user, role = _datastore._prepare_role_modify_args(username, rolename)
if not user:
user = create_test_user(email=username, password='******')
if not role:
role = _datastore.create_role(name=rolename)
_datastore.add_role_to_user(user, role)
return user
def create_user_with_role(username, rolename):
_datastore = LocalProxy(
lambda: current_app.extensions['security'].datastore)
user, role = _datastore._prepare_role_modify_args(username, rolename)
if not user:
user = create_test_user(email=username, password='******')
if not role:
role = _datastore.create_role(name=rolename)
_datastore.add_role_to_user(user, role)
return user
