class Users(Iterator):
def __init__(self, hostname, auth):
self.admin = AdminClient(hostname)
self.auth = auth
def next(self, seqid, channel):
name = 'user-{}'.format(seqid)
password = '******'
logger.info('Creating new user {}, auth by {}'.format(name, self.auth))
self.admin.add_user(name=name, password=password, channels=[channel])
if self.auth == 'cookie':
cookies = self.admin.create_session(name=name)
return {'cookies': cookies}
else:
return {'name': name, 'password': password}
def __init__(self, hostname, auth):
self.admin = AdminClient(hostname)
self.auth = auth
