def test_remove_client(self):
repoClients = Repo("Repoclients:")
repoClients.add(Client(20, "John Wright"))
repoClients.add(Client(40, "Andrei Ivan"))
repoClients.remove(Client(40, None))
assert(repoClients.size() == 1)
assert(repoClients.get_all()[0].clientID == 20)
def test_add_client(self):
repoClients = Repo("Repoclients:")
repoClients.add(Client(20, "John Wright"))
repoClients.add(Client(40, "Andrei Ivan"))
assert(repoClients.size() == 2)
assert(repoClients.get_all()[0].clientID == 20)
assert(repoClients.get_all()[1].clientID == 40)
assert(repoClients.get_all()[0].clientName == "John Wright")
assert(repoClients.get_all()[1].clientName == "Andrei Ivan")
def test_remove_book(self):
repoBooks = Repo("Repobooks:")
repoBooks.add(Book(1, "The Secret Crusade", "Oliver Bowden"))
repoBooks.add(Book(2, "The Illustrated Man", "Ray Bradbury"))
repoBooks.remove(Book(1, None, None))
assert(repoBooks.size() == 1)
assert(repoBooks.get_all()[0].bookID == 2)
with self.assertRaises(RepositoryError) as context:
repoBooks.remove(Book(22, None, None))
self.assertTrue("\n[REPOSITORY ERROR] -> Repobooks: Inexistend ID!\n" in str(context.exception))
def test_add_book(self):
repoBooks = Repo("Repobooks:")
repoBooks.add(Book(1, "The Secret Crusade", "Oliver Bowden"))
repoBooks.add(Book(2, "The Illustrated Man", "Ray Bradbury"))
self.assertRaises(RepositoryError, lambda:repoBooks.add(Book(1, None, None)))
assert(repoBooks.size() == 2)
assert(repoBooks.get_all()[0].bookID == 1)
assert(repoBooks.get_all()[1].bookID == 2)
assert(repoBooks.get_all()[0].bookTitle == "The Secret Crusade")
assert(repoBooks.get_all()[1].bookTitle == "The Illustrated Man")
assert(repoBooks.get_all()[0].bookAuthor == "Oliver Bowden")
assert(repoBooks.get_all()[1].bookAuthor == "Ray Bradbury")
