Feat: add TribeSQLAlchemyRepository

This commit is contained in:
Bertrand Benjamin 2022-12-21 14:10:04 +01:00
parent 21821c275e
commit b3bb1f0cc8
3 changed files with 119 additions and 0 deletions

View File

@ -0,0 +1,23 @@
import abc
class AbstractRepository(abc.ABC):
@abc.abstractmethod
def add(self, element):
raise NotImplementedError
@abc.abstractmethod
def update(self, element):
raise NotImplementedError
@abc.abstractmethod
def list(self):
raise NotImplementedError
@abc.abstractmethod
def get(self, reference):
raise NotImplementedError
@abc.abstractmethod
def delete(self, reference):
raise NotImplementedError

View File

@ -0,0 +1,23 @@
from backend.model.tribe import Tribe
from backend.repository.abstract_repository import AbstractRepository
class TribeSQLAlchemyRepository(AbstractRepository):
def __init__(self, session) -> None:
self.session = session
def add(self, tribe: Tribe) -> None:
self.session.add(tribe)
def update(self, name: str, tribe: Tribe) -> None:
self.session.query(Tribe).filter_by(name=name).update(tribe.to_dict())
def get(self, name: str) -> Tribe:
return self.session.query(Tribe).filter_by(name=name).one()
def list(self) -> list[Tribe]:
return self.session.query(Tribe).all()
def delete(self, tribe: Tribe) -> None:
the_tribe = self.get(tribe.name)
self.session.delete(the_tribe)

View File

@ -0,0 +1,73 @@
from backend.model.tribe import Tribe
from backend.repository.tribe_sqlalchemy_repository import TribeSQLAlchemyRepository
def test_add_tribe(session):
tribe_infos = ("tribe1", "2nd")
tribe = Tribe(*tribe_infos)
tribe_repo = TribeSQLAlchemyRepository(session)
tribe_repo.add(tribe)
session.commit()
rows = list(session.execute("SELECT name, level FROM 'tribes'"))
assert rows == [tribe_infos]
def test_list_tribes(session):
session.execute(
"INSERT INTO tribes (name, level) VALUES "
"('tribe1', '2nd'),"
"('tribe2', '1ST')"
)
expected = [
(Tribe("tribe1", "2nd")),
(Tribe("tribe2", "1ST")),
]
tribe_repo = TribeSQLAlchemyRepository(session)
assert tribe_repo.list() == expected
def test_update_tribe(session):
session.execute(
"INSERT INTO tribes (name, level) VALUES "
"('tribe1', '2nd'),"
"('tribe2', '1ST')"
)
tribe_repo = TribeSQLAlchemyRepository(session)
tribe_repo.update("tribe1", Tribe("tribe3", "Term"))
session.commit()
expected = [
(Tribe("tribe3", "Term")),
(Tribe("tribe2", "1ST")),
]
assert tribe_repo.list() == expected
def test_get_tribes(session):
session.execute(
"INSERT INTO tribes (name, level) VALUES "
"('tribe1', '2nd'),"
"('tribe2', '1ST')"
)
expected = Tribe("tribe1", "2nd")
tribe_repo = TribeSQLAlchemyRepository(session)
assert tribe_repo.get("tribe1") == expected
def test_delete_tribes(session):
session.execute(
"INSERT INTO tribes (name, level) VALUES "
"('tribe1', '2nd'),"
"('tribe2', '1ST')"
)
expected = [
(Tribe("tribe2", "1ST")),
]
tribe_repo = TribeSQLAlchemyRepository(session)
tribe_repo.delete(Tribe("tribe1", "2nd"))
session.commit()
assert tribe_repo.list() == expected