def user(auth_index, customer, another_customer):
_user = User(
email='*****@*****.**',
customers=[customer.name, another_customer.name]
)
_user.save(index=auth_index._name, refresh=True)
yield _user
_user.delete(index=auth_index._name, ignore=409)
def create_user(email, customer_names):
"""Create an User."""
click.secho('Creating user: %s' % email, bold=True)
for customer_name in customer_names:
try:
Customer.get_by_name(customer_name)
except NotFoundError:
click.secho('* customer does not exist: %s' % customer_name,
fg='red')
raise click.Abort()
try:
user = User.get_by_email(email)
click.secho('* user exists already: %s' % user, fg='red')
except NotFoundError:
user = User(email=email, customers=customer_names)
user.save()
click.secho('* user created: %s' % user, fg='green')
def user(user_index):
"""Create a random `User`."""
user = User(username=uuid4().hex, email='*****@*****.**' % uuid4().hex)
user.save(index=user_index._name, refresh=True)
return user