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_update_client(self): repoClients = Repo("Repoclients:") repoClients.add(Client(20, "John Wright")) repoClients.add(Client(40, "Andrei Ivan")) repoClients.update(Client(40, "New")) assert(repoClients.get_all()[1].clientName == "New")
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")
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_update_book(self): repoBooks = Repo("Repobooks:") repoBooks.add(Book(1, "The Secret Crusade", "Oliver Bowden")) repoBooks.add(Book(2, "The Illustrated Man", "Ray Bradbury")) repoBooks.update(Book(1, "New", "New2")) assert(repoBooks.get_all()[0].bookTitle == "New") assert(repoBooks.get_all()[0].bookAuthor == "New2") self.assertRaises(RepositoryError, lambda : repoBooks.update(Book(20, None, None)))
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))