class TestDomain(unittest.TestCase): def setUp(self): self.client = Client(23, 'alex') self.client2 = Client(12, 'diana') self.book = Book(21, 'titlu', 'descriere', 'author') self.book2 = Book(22, 'titlu2', 'descriere2', 'author2') self.rental = Rental(21236, 21, 23, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) self.rental2 = Rental(21236, 21, 23, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) def tearDown(self): pass def test_create_Client(self): # CLIENTS TESTS self.assertEqual(23, self.client.getID()) self.assertEqual('alex', self.client.getName()) self.client.setName('da') self.assertEqual('da', self.client.getName()) self.assertFalse(self.client == self.client2) self.assertEqual(str(self.client), '023. da\n') self.assertEqual(self.client.create_client(23, 'alex'), self.client) def test_create_book(self): self.assertEqual(21, self.book.getID()) self.assertEqual('titlu', self.book.getTitle()) self.assertEqual('descriere', self.book.getDescription()) self.assertEqual('author', self.book.getAuthor()) self.assertEqual(True, self.book.get_available()) self.book.setTitle('lol') self.assertEqual('lol', self.book.getTitle()) self.book.setAuthor('da') self.assertEqual('da', self.book.getAuthor()) self.book.setDescription('da') self.assertEqual('da', self.book.getDescription()) self.assertEqual(str(self.book), '021. lol by da\n->da\nStatus: Available\n') self.book.set_available(False) self.assertEqual(False, self.book.get_available()) self.assertEqual(str(self.book), '021. lol by da\n->da\nStatus: Unavailable\n') self.assertNotEqual(self.book, self.book2) self.assertEqual(self.book.create_book(21, 'titlu', 'descriere', 'author'), self.book) def test_create_rental(self): self.assertEqual(21236, self.rental.getID()) self.assertEqual(21, self.rental.getBookID()) self.assertEqual(23, self.rental.getClientID()) self.assertEqual(date(2017, 12, 6), self.rental.getDueDate()) self.assertEqual(date(2017, 11, 5), self.rental.getRentedDate()) self.assertEqual(date(2017, 12, 5), self.rental.getReturnedDate()) self.assertEqual('Rental 21236: book: 21 rented by client 23 on 05.11.2017 and is due by 06.12.2017', str(self.rental)) self.rental.setBookID(21) self.rental.setClientID(23) self.rental.setReturnedDate(date(2017, 12, 12)) self.rental.setDueDate(date(2017, 11, 21)) self.rental.setRentedDate(date(2017, 9, 23)) self.assertEqual(len(self.rental), 80) self.rental.setReturnedDate(None) self.assertEqual(len(self.rental), 0) self.assertEqual(self.rental.getDelay(), (date.today()-self.rental.getDueDate()).days) self.assertEqual(self.rental, self.rental2) self.rental.setReturnedDate(date(2017, 12, 12)) self.assertEqual(self.rental.create_rental(21, 23, 12, 5).getBookID(), 21) self.assertEqual(self.rental.printReturned(self.rental), 'Rental 21236: book: 21 rented by client 23 on 23.09.2017 was returned on 12.12.2017')
class TestRentalController(unittest.TestCase): def setUp(self): self.repo = Repository() self.brepo = Repository() self.crepo = Repository() self.Ucontroller = UndoController() self.book = Book(21, 'titlu', 'descriere', 'author') self.book2 = Book(22, 'titlu2', 'descriere2', 'author') self.brepo.add(self.book) self.brepo.add(self.book2) self.client = Client(23, 'alex') self.client2 = Client(24, 'ana') self.client3 = Client(29, 'ana') self.crepo.add(self.client) self.crepo.add(self.client2) self.rental = Rental(21236, 21, 24, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) self.rental2 = Rental(21238, 22, 24, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) self.rental3 = Rental(21238, 23, 24, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) self.rental4 = Rental(21238, 21, 29, date(2017, 11, 5), date(2017, 12, 6), date(2017, 12, 5)) self.rental5 = Rental(21231, 21, 23, date(2017, 11, 5), date(2017, 10, 6), None) self.controller = RentalController(self.repo, self.brepo, self.crepo, self.Ucontroller) self.ceva = LateRentalCount(12, 32) self.altceva = BookRentalCount(12, 23) self.nu = AuthorRentalCount('da', 23) self.da = ClientRentalCount(32, 12) def test_controller(self): self.Ucontroller.newOperation() self.controller.addRental(self.rental2) self.assertEqual(len(self.repo), 1) self.assertEqual(self.controller.getRental(21238).getID(), 21238) self.assertRaises(ControllerException, self.controller.addRental, self.rental3) self.assertRaises(ControllerException, self.controller.addRental, self.rental4) self.book2.set_available(False) self.assertRaises(ControllerException, self.controller.addRental, self.rental2) self.assertRaises(ControllerException, self.controller.returnBook, 13123124) self.Ucontroller.newOperation() self.controller.addRental(self.rental) self.assertEqual(self.controller.returnBook(21236), self.rental) self.assertEqual(len(self.controller.getAllRental()), 2) self.assertEqual(len(self.controller.mostRentedBooks()), 2) self.assertEqual(len(self.controller.mostActiveClients()), 2) self.assertEqual(len(self.controller.mostRentedAuthors()), 1) self.controller.deleteRental(self.rental.getID()) self.Ucontroller.newOperation() self.controller.addRental(self.rental5) self.assertEqual(len(self.controller.lateRentals()), 1) self.assertEqual(self.ceva.getBook(), 12) self.assertEqual(self.ceva.getCount(), 32) self.assertFalse(self.ceva > self.ceva) self.assertEqual(str(self.ceva), 'Book 12 has been overdue for 32 days') self.assertEqual(self.altceva.getCount(), 23) self.assertEqual(self.altceva.getBook(), 12) self.assertFalse(self.altceva > self.altceva) self.assertEqual(str(self.altceva), 'Book 12 was rented 23 times') self.assertEqual(self.nu.getBook(), 'da') self.assertEqual(self.nu.getCount(), 23) self.assertEqual(str(self.nu), 'Author da was rented for 23 times') self.assertFalse(self.nu > self.nu) self.assertFalse(self.da > self.da) self.assertEqual(str(self.da), 'Client 32 has rented books for 12 days') self.assertEqual(self.da.getCount(), 12) self.assertEqual(self.da.getBook(), 32)