diff options
Diffstat (limited to 'src/match_up/tests')
| -rw-r--r-- | src/match_up/tests/__init__.py | 35 |
1 files changed, 26 insertions, 9 deletions
diff --git a/src/match_up/tests/__init__.py b/src/match_up/tests/__init__.py index 38e377f..4dbd883 100644 --- a/src/match_up/tests/__init__.py +++ b/src/match_up/tests/__init__.py @@ -1,4 +1,5 @@ from os import remove +from os.path import isfile from random import randint from pathlib import PurePath from names import get_first_name, get_last_name @@ -9,16 +10,32 @@ MIN_SKILL = 1 MAX_SKILL = 10 -def generate_test_database(total_players: int = 20, active_players: int = 17) -> None: - p = PlayerDatabase(TEST_DATABASE_LOCATION) - for _ in range(total_players): - p.add_player( - Player( - first_name=get_first_name(), - last_name=get_last_name(), - skill=randint(MIN_SKILL, MAX_SKILL), +def generate_test_database(total_players: int = 20) -> None: + try: + if isfile(TEST_DATABASE_LOCATION): + remove_test_database() + db = PlayerDatabase(TEST_DATABASE_LOCATION) + for _ in range(total_players): + db.add_player( + Player( + first_name=get_first_name(), + last_name=get_last_name(), + skill=randint(MIN_SKILL, MAX_SKILL), + ) ) - ) + finally: + db.close() + + +def activate_random_players(activate_players: int = 17) -> None: + try: + db = PlayerDatabase(TEST_DATABASE_LOCATION) + player_list = db.get_players() + for index, player in enumerate(player_list): + if index < activate_players: + db.activate_player(player) + finally: + db.close() def remove_test_database() -> None: |