def test_create_lol_user(discord_id, riot_puuid):
"""
Tests that creating a new LoLUser works correctly
"""
queried_users = session.query(LoLUser).all()
if len(queried_users) > 0:
session.query(LoLUser).delete()
session.commit()
assert len(session.query(LoLUser).all()) == 0
user = LoLUser(discord_id, riot_puuid, winrate=True)
session.add(user)
session.commit()
queried_users = session.query(LoLUser).all()
assert len(queried_users) == 1
queried_user = queried_users[0]
assert queried_user.discord_id == discord_id
assert queried_user.riot_puuid == riot_puuid
assert queried_user.last_updated > 0
assert queried_user.winrate
assert queried_user.wins == 0
assert queried_user.losses == 0
def create_bet(
player: int, better: int, bet: int, prediction: bool, created: int
):
"""
creates a new bet on a player
"""
bet = LoLBets(player, better, bet, prediction, created)
session.add(bet)
session.commit()
def add_lol_game(user: LoLUser, game_id: int, start_time: int, game_mode: str):
"""
Adds an active lol game to the db
"""
active_game = LoLActiveGames(
game_id, user.discord_id, start_time, game_mode
)
session.add(active_game)
session.commit()
def test_create_lol_text(lol_text_group_name, lol_user):
"""
Tests that a lol text can be created successfully
"""
assert len(session.query(LoLText).all()) == 0
text = LoLText(lol_text_group_name, "ending.", lol_user)
session.add(text)
session.commit()
assert len(session.query(LoLText).all()) == 1
def create_new_lol_user(discord_id: int, riot_puuid: str):
"""
Creates a new lol user with the given discord id and riot puuid
Args:
discord_id (int): the id of the discord user
riot_puuid (str): the id of the riot account
"""
user = LoLUser(discord_id, riot_puuid)
session.add(user)
session.commit()
def add_lol_text(group_name: str, content: str, user_id: int):
"""
Adds a new text to the db
Args:
group_name (str): the name of the group being added (must be unique)
content (str): the text to be displayed
user_id (int): the discord id of the user adding the group
"""
text = LoLText(group_name, content, user_id)
session.add(text)
session.commit()
def add_lol_text_group(group_name: str, group_description: str, user_id: int):
"""
Adds a new text group with a description to the db
Args:
group_name (str): the name of the group being added (must be unique)
group_description (str): a description of what this group is used for
user_id (int): the discord id of the user adding the group
"""
group = LoLTextGroup(group_name, group_description, user_id)
session.add(group)
session.commit()
def test_create_lol_text_group(lol_text_group_name, lol_text_group_usage,
lol_user):
"""
Tests that a lol text group can be created successfully
"""
assert len(session.query(LoLTextGroup).all()) == 0
text_group = LoLTextGroup(lol_text_group_name, lol_text_group_usage,
lol_user)
session.add(text_group)
session.commit()
assert len(session.query(LoLTextGroup).all()) == 1
def test_delete_lol_user(discord_id, riot_puuid):
"""
Tests that a LoLUser can be deleted
"""
queried_users = session.query(LoLUser).all()
user = None
if len(queried_users) != 1:
# user is not present, insert new user
user = LoLUser(discord_id, riot_puuid)
session.add(user)
session.commit()
else:
user = queried_users[0]
session.query(LoLUser).filter(LoLUser.discord_id == discord_id).delete()
session.commit()
assert len(session.query(LoLUser).all()) == 0
def test_delete_lol_text_group(lol_text_group_name, lol_text_group_usage,
lol_user):
"""
Tests that a lol text group can be deleted successfully
"""
queried_groups = session.query(LoLTextGroup).all()
group = None
if len(queried_groups) != 1:
group = LoLTextGroup(lol_text_group_name, lol_text_group_usage,
lol_user)
session.add(group)
session.commit()
else:
group = queried_groups[0]
session.query(LoLTextGroup).filter(
LoLTextGroup.name == lol_text_group_name).delete()
session.commit()
assert len(session.query(LoLTextGroup).all()) == 0
def setup_module(module):
""" setup any state specific to the execution of the given module."""
# clean db before insertions
session.query(LoLUser).delete()
session.commit()
session.query(LoLText).delete()
session.commit()
session.query(LoLTextGroup).delete()
session.commit()
# populate db with mock users to test with
users = mock_users()
for user in users:
session.add(user)
session.commit()
text_groups = mock_text_groups()
for text_group in text_groups:
session.add(text_group)
session.commit()
texts = mock_text(text_groups[0], text_groups[1])
for text in texts:
session.add(text)
session.commit()