Feat: rename populate and use fakers

This commit is contained in:
Bertrand Benjamin 2022-12-27 08:13:34 +01:00
parent de9a4bc4be
commit 4e13d0e32f
2 changed files with 28 additions and 31 deletions

View File

@ -28,3 +28,6 @@ class Tribe:
def to_dict(self) -> dict:
return {"name": self.name, "level": self.level}
def to_tuple(self) -> tuple:
return (self.name, self.level)

View File

@ -5,39 +5,38 @@ import pytest
from backend.adapters.sqlite import create_db
from backend.model.tribe import Tribe
from backend.repository.tribe_sqlite_repository import TribeSQLiteRepository
from tests.model.fakes import build_tribes
def populate_tribe(conn) -> list[Tribe]:
def populate_tribes(conn) -> list[Tribe]:
cursor = conn.cursor()
tribes = [
("Tribe1", "2nd"),
("Tribe2", "2nd"),
("Tribe3", "1ST"),
]
tribes = build_tribes(3)
cursor.executemany(
"""
INSERT INTO tribes(name, level) VALUES (?, ?)
""",
tribes,
[t.to_tuple() for t in tribes],
)
conn.commit()
return [Tribe(*t) for t in tribes]
return tribes
def test_get_tribe(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
name = prebuild_tribes[0].name
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribes = tribe_repo.get("Tribe1")
tribes = tribe_repo.get(name)
assert prebuild_tribes[0] == tribes
def test_get_tribe_not_exists(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
with pytest.raises(ValueError):
@ -46,7 +45,7 @@ def test_get_tribe_not_exists(sqlite_conn):
def test_list_tribes(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribes = tribe_repo.list()
@ -78,42 +77,37 @@ def test_add_tribe(sqlite_conn):
def test_add_tribe_fail_exists(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribe_infos = ("Tribe1", "2nd")
tribe = Tribe(*tribe_infos)
existing_tribe = prebuild_tribes[0]
with pytest.raises(sqlite3.IntegrityError):
tribe_repo.add(tribe)
tribe_repo.add(existing_tribe)
def test_update_tribe(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribe_repo.update("Tribe1", Tribe("Tribe0", "Term"))
name = prebuild_tribes[0].name
new_tribe = Tribe("Tribe0", "Term")
tribe_repo.update(name, new_tribe)
sqlite_conn.commit()
expected = [
Tribe("Tribe0", "Term"),
Tribe("Tribe2", "2nd"),
Tribe("Tribe3", "1ST"),
]
assert tribe_repo.list() == expected
prebuild_tribes[0] = new_tribe
assert tribe_repo.list() == prebuild_tribes
def test_delete_tribe(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
prebuild_tribes = populate_tribes(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribe_repo.delete(prebuild_tribes[0])
deleted_tribe = prebuild_tribes.pop()
tribe_repo.delete(deleted_tribe)
sqlite_conn.commit()
expected = [
Tribe("Tribe2", "2nd"),
Tribe("Tribe3", "1ST"),
]
assert tribe_repo.list() == expected
assert tribe_repo.list() == prebuild_tribes