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