Feat: add get_tribe for sqlite

This commit is contained in:
2022-12-26 19:05:50 +01:00
parent 6fbe238e59
commit fe92433311
2 changed files with 36 additions and 2 deletions

View File

@@ -1,9 +1,11 @@
import pytest
from backend.adapters.sqlite import create_db
from backend.model.tribe import Tribe
from backend.repository.tribe_sqlite_repository import TribeSQLiteRepository
def populate_tribe(conn) -> None:
def populate_tribe(conn) -> list[Tribe]:
cursor = conn.cursor()
tribes = [
("Tribe1", "2nd"),
@@ -21,6 +23,25 @@ def populate_tribe(conn) -> None:
return [Tribe(*t) for t in tribes]
def test_get_tribe(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
tribes = tribe_repo.get("Tribe1")
assert prebuild_tribes[0] == tribes
def test_get_tribe_not_exists(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)
tribe_repo = TribeSQLiteRepository(sqlite_conn)
with pytest.raises(ValueError):
tribe_repo.get("Tribe0")
def test_list_tribes(sqlite_conn):
create_db(sqlite_conn)
prebuild_tribes = populate_tribe(sqlite_conn)