class CreateAccount(PageBase):
new_username = ""
new_password = ""
def __init__(self):
super().__init__()
self.username = self.get_element(By.CSS_SELECTOR, 'input[name="usernameRegisterPage"]')
self.email = self.get_element(By.CSS_SELECTOR, 'input[name="emailRegisterPage"]')
self.password = self.get_element(By.CSS_SELECTOR, 'input[name="passwordRegisterPage"]')
self.confirm_password = self.get_element(By.CSS_SELECTOR, 'input[name="confirm_passwordRegisterPage"]')
self.agree_button = self.get_element(By.CSS_SELECTOR, 'input[name="i_agree"]')
self.register_button = self.get_element(By.CSS_SELECTOR, '#register_btnundefined')
self.generate_email = RandomEmails()
self.generate_username = RandomNicknames()
def enter_valid_details(self): # enters details to register
password = PASSWORD
username = self.generate_username.random_nicks()
email = self.generate_email.randomMail()
self.username.send_keys(username)
self.email.send_keys(email)
self.password.send_keys(password)
self.confirm_password.send_keys(password)
self.agree_button.click()
self.register_button.click()
return [username, password]
def _create_users(self):
rn = RandomNicknames()
for name in rn.random_nicks(count=50):
username = '******' % (slugify(name), random.randrange(1, 99))
user = User.objects.create_user(username=username,
email='*****@*****.**' % username,
password='******')
logger.info('Created user: %s', user.username)
def get_dummy_badges(self, count=20):
rn = RandomNicknames()
badges, slugs = [], []
for name in rn.random_nicks(count=count):
name = '%s #%d' % (name, randrange(99))
slug = slugify(name)
slugs.append(slug)
badges.append(Badge(name=name, slug=slug, description=name))
return badges, slugs
def get_dummy_badges(self, count=20):
rn = RandomNicknames()
badges, slugs = [], []
for name in rn.random_nicks(count=count):
name = '%s #%d' % (name, randrange(99))
slug = slugify(name)
slugs.append(slug)
badges.append(Badge(name=name, slug=slug, description=name))
return badges, slugs
def _create_badges(self):
rn = RandomNicknames()
for name in rn.random_nicks(count=20):
slug = slugify(name)
badge = Badge.objects.create(
name=name,
slug=slug,
description='Lorem ipsum dolor sit amet, consectetur adipisicing elit')
logger.info('Created badge: %s', badge.name)
def _create_users(self):
rn = RandomNicknames()
for name in rn.random_nicks(count=50):
username = '******' % (slugify(name), random.randrange(1, 99))
user = User.objects.create_user(
username=username,
email='*****@*****.**' % username,
password='******')
logger.info('Created user: %s', user.username)
def _create_badges(self):
rn = RandomNicknames()
for name in rn.random_nicks(count=20):
slug = slugify(name)
badge = Badge.objects.create(
name=name,
slug=slug,
description=
'Lorem ipsum dolor sit amet, consectetur adipisicing elit')
logger.info('Created badge: %s', badge.name)
def get_dummy_users(self, count=20):
rn = RandomNicknames()
users, usernames = [], []
for name in rn.random_nicks(count=count):
name = '%s #%d' % (name, randrange(99))
username = slugify(name)
usernames.append(username)
users.append(
User(username=username,
email='*****@*****.**' % username,
password='******'))
return (users, usernames)
def get_dummy_users(self, count=20):
rn = RandomNicknames()
users, usernames = [], []
for name in rn.random_nicks(count=count):
name = '%s #%d' % (name, randrange(99))
username = slugify(name)
usernames.append(username)
users.append(User(
username=username,
email='*****@*****.**' % username,
password='******'))
return (users, usernames)
def get_dummy_recipes(self, count=50):
rn = RandomNicknames()
recipes, slugs = [], []
for name in rn.random_nicks(count=count):
slug = slugify(name)
slugs.append(slug)
name = name.encode('ascii', 'ignore')
attrs = dict(name=name,
slug=slug,
description=name,
image=name,
user_queryset=User.objects.none())
recipes.append(type(name, (BaseRecipe, ), attrs))
return (recipes, slugs)
def get_dummy_recipes(self, count=50):
rn = RandomNicknames()
recipes, slugs = [], []
for name in rn.random_nicks(count=count):
slug = slugify(name)
slugs.append(slug)
name = name.encode('ascii', 'ignore')
attrs = dict(
name=name,
slug=slug,
description=name,
image=name,
user_queryset=User.objects.none())
recipes.append(type(name, (BaseRecipe,), attrs))
return (recipes, slugs)
def generate_songs_file(
limit, artist_id_limit=1000000, popularity_limit=1000000):
"""
generates files like
1|Help|154681|The Beatles|25
"""
rw = RandomWords()
rn = RandomNicknames()
with open(songs_file, 'w+') as f:
for i in range(limit):
line = [str(i)]
# title
line.append(' '.join(rw.random_words(count=randint(1, 5))))
line.append(str(int(popularity_limit * betavariate(2, 5))))
# artist
line.append(' '.join(rn.random_nicks(count=randint(1, 3))))
line.append(str(randint(0, artist_id_limit)))
f.write('|'.join(line) + '\n')
def generate_songs_file(limit,
artist_id_limit=1000000,
popularity_limit=1000000):
"""
generates files like
1|Help|154681|The Beatles|25
"""
rw = RandomWords()
rn = RandomNicknames()
with open(songs_file, 'w+') as f:
for i in range(limit):
line = [str(i)]
# title
line.append(' '.join(rw.random_words(count=randint(1, 5))))
line.append(str(int(popularity_limit * betavariate(2, 5))))
# artist
line.append(' '.join(rn.random_nicks(count=randint(1, 3))))
line.append(str(randint(0, artist_id_limit)))
f.write('|'.join(line) + '\n')
def generate_random_chats(message_count: int,
participants: int = 2,
chats_number: int = 1) -> List[Chat]:
rn = RandomNicknames()
li = LoremIpsum()
chats = []
# Weighted sentence length probability table
message_length_probability = \
[1] * 25 + \
[2] * 35 + \
[3] * 20 + \
[4] * 10 + \
[5] * 7 + \
[6] * 3
for chat_n in range(chats_number):
people = rn.random_nicks(count=participants)
dt = datetime.now()
messages = []
for i in range(message_count):
# Generate random time intervals between 10 seconds and 24 hours
secs = randint(10, 60 * 60 * 24)
dt = dt + timedelta(seconds=secs)
# chooses a random sender
sender = random.choice(people)
# generate random message:
sentences_number = random.choice(message_length_probability)
text = li.get_sentences(sentences_number)
messages.append(Message(sender, dt, text))
chats.append(Chat(f"RandomChat_{chat_n}", people, messages))
return chats
