166 lines
4.8 KiB
Python
166 lines
4.8 KiB
Python
import pytest
|
|
|
|
from backend.model.student import Student
|
|
from backend.model.tribe import Tribe
|
|
from backend.repository.abstract_repository import AbstractRepository
|
|
from backend.repository.student_sqlite_repository import StudentRepositoryError
|
|
from backend.repository.tribe_sqlite_repository import TribeRepositoryError
|
|
from backend.service import services
|
|
from backend.service.services import TribeDosNotExists, TribeExists
|
|
from tests.model.fakes import build_tribes
|
|
|
|
|
|
class FakeTribeRepository(AbstractRepository):
|
|
def __init__(self, tribes: list[Tribe] = []) -> None:
|
|
self._tribes = {t.name: t for t in tribes}
|
|
|
|
def add(self, tribe: Tribe) -> None:
|
|
if tribe.name not in self._tribes.keys():
|
|
self._tribes[tribe.name] = tribe
|
|
else:
|
|
raise TribeRepositoryError(f"{tribe} already exists")
|
|
|
|
def update(self, name: str, tribe: Tribe) -> None:
|
|
try:
|
|
self._tribes.pop(name)
|
|
self._tribes[tribe.name] = tribe
|
|
except KeyError:
|
|
raise TribeRepositoryError(f"The tribe {tribe} does not exists")
|
|
|
|
def list(self) -> list[Tribe]:
|
|
return list(self._tribes.values())
|
|
|
|
def get(self, name: str) -> Tribe:
|
|
try:
|
|
return self._tribes[name]
|
|
except KeyError:
|
|
raise TribeRepositoryError(f"The tribe {tribe} does not exists")
|
|
|
|
def delete(self, name: str) -> None:
|
|
try:
|
|
self._tribes.pop(name)
|
|
except KeyError:
|
|
raise TribeRepositoryError(f"The tribe {name} does not exists")
|
|
|
|
|
|
class FakeStudentRepository(AbstractRepository):
|
|
def __init__(self, students: list[Student] = []) -> None:
|
|
self._students = {s.id: s for s in students}
|
|
|
|
def add(self, student: Student) -> None:
|
|
if student.id not in self._students.keys():
|
|
self._students[student.id] = student
|
|
raise StudentRepositoryError(f"{student} already exists")
|
|
|
|
def update(self, id: str, student: Student) -> None:
|
|
try:
|
|
self._students.pop(id)
|
|
self._students[student.id] = student
|
|
except KeyError:
|
|
raise StudentRepositoryError(f"The student {student} does not exists")
|
|
|
|
def list(self) -> list[Student]:
|
|
return list(self._students.values())
|
|
|
|
def get(self, id: str) -> Student:
|
|
try:
|
|
return self._students[id]
|
|
except KeyError:
|
|
raise KeyError(f"The student {student} does not exists")
|
|
|
|
def delete(self, id: str) -> None:
|
|
try:
|
|
self._students.pop(id)
|
|
except KeyError:
|
|
raise StudentRepositoryError(f"The student {student} does not exists")
|
|
|
|
|
|
class FakeConn:
|
|
committed = False
|
|
|
|
def commit(self):
|
|
self.committed = True
|
|
|
|
def reset_commit(self):
|
|
self.committed = False
|
|
|
|
|
|
def test_add_tribe():
|
|
tribe_repo = FakeTribeRepository()
|
|
tribe = build_tribes(1)[0]
|
|
conn = FakeConn()
|
|
services.add_tribe(
|
|
name=tribe.name, level=tribe.level, tribe_repo=tribe_repo, conn=conn
|
|
)
|
|
|
|
assert conn.committed == True
|
|
assert tribe_repo.list() == [tribe]
|
|
|
|
|
|
def test_add_tribe_fail_exists():
|
|
tribe_repo = FakeTribeRepository()
|
|
tribe = build_tribes(1)[0]
|
|
conn = FakeConn()
|
|
services.add_tribe(
|
|
name=tribe.name, level=tribe.level, tribe_repo=tribe_repo, conn=conn
|
|
)
|
|
conn.reset_commit()
|
|
|
|
with pytest.raises(TribeExists):
|
|
services.add_tribe(
|
|
name=tribe.name, level=tribe.level, tribe_repo=tribe_repo, conn=conn
|
|
)
|
|
|
|
assert conn.committed == False
|
|
|
|
|
|
def test_update_tribe():
|
|
tribes = build_tribes(3)
|
|
tribe_repo = FakeTribeRepository(tribes)
|
|
conn = FakeConn()
|
|
|
|
other_level = "iouiouiou"
|
|
tribes[0].level = other_level
|
|
services.update_tribe(
|
|
name=tribes[0].name, level=other_level, tribe_repo=tribe_repo, conn=conn
|
|
)
|
|
|
|
assert conn.committed == True
|
|
assert set(tribe_repo.list()) == set(tribes)
|
|
|
|
|
|
def test_update_tribe_fail_not_exists():
|
|
tribes = build_tribes(3)
|
|
tribe_repo = FakeTribeRepository(tribes)
|
|
conn = FakeConn()
|
|
|
|
with pytest.raises(TribeDosNotExists):
|
|
services.update_tribe(
|
|
name="azerty", level="jkl", tribe_repo=tribe_repo, conn=conn
|
|
)
|
|
|
|
assert conn.committed == False
|
|
|
|
|
|
def test_delete_tribe():
|
|
tribes = build_tribes(3)
|
|
tribe_repo = FakeTribeRepository(tribes)
|
|
conn = FakeConn()
|
|
|
|
tribe = tribes.pop()
|
|
services.delete_tribe(name=tribe.name, tribe_repo=tribe_repo, conn=conn)
|
|
|
|
assert conn.committed == True
|
|
assert set(tribe_repo.list()) == set(tribes)
|
|
|
|
|
|
def test_delete_tribe_fail_not_exists():
|
|
tribes = build_tribes(3)
|
|
tribe_repo = FakeTribeRepository(tribes)
|
|
conn = FakeConn()
|
|
|
|
with pytest.raises(TribeDosNotExists):
|
|
services.delete_tribe(name="azerty", tribe_repo=tribe_repo, conn=conn)
|
|
|
|
assert conn.committed == False
|