def test_get_book(sqlite_session_factory, title, author):
session = sqlite_session_factory()
init_db(session)
repo = book_repository.SqlAlchemyBookRepository(session)
book = repo.get(title)
assert book.title == title
assert book.author == author
def __enter__(self):
self.session = self.session_factory() # type: Session
self.books = book_repository.SqlAlchemyBookRepository(self.session)
return super().__enter__()
def test_delete_book(sqlite_session_factory):
session = sqlite_session_factory()
repo = book_repository.SqlAlchemyBookRepository(session)
repo.delete(title="A Book")
def test_update_book(sqlite_session_factory):
session = sqlite_session_factory()
repo = book_repository.SqlAlchemyBookRepository(session)
book = model.Book("A Book", "An Author", "A Publisher", 1000)
repo.update(book)