def test_updateClient(self): with self.assertRaises(Exception): self.clientService.updateClient(Client(555, "Jhon")) self.clientService.updateClient(tuple([Client(1, "Jhon")])) self.clientService.updateClient(Client(1, "Jhon")) c = self.clientService.getClient(1) self.assertEqual(c.attrs, {Utils.CLIENT_ID: 1, Utils.CLIENT_NAME: "Jhon"})
def test_addClient(self): with self.assertRaises(Exception): self.clientService.addClient(Client(1, "Jhon")) self.clientService.addClient(tuple([Client(33, "Name")])) self.clientService.addClient(Client(55, "Name")) c = self.clientService.getClient(55) self.assertEqual(c.attrs, {Utils.CLIENT_ID: 55, Utils.CLIENT_NAME: "Name"}, "Error add client")
def test_Client(self): c = Client(1, "Name 1") self.assertEqual(c.ID, 1, "ID error") c.ID = 2 self.assertEqual(c.ID, 2, "ID setting error") self.assertEqual(c.clientNAME, "Name 1", "Name error") c.clientNAME = "Name 2" self.assertEqual(c.clientNAME, "Name 2", "Name setting error") self.assertEqual(c.attrs, {Utils.CLIENT_ID: 2, Utils.CLIENT_NAME: "Name 2"}, "Attrs error")
def test_queryList(self): c1 = Client(1, "Jhon") c2 = Client(2, "Jhonny") l = {1: c1, 2: c2} with self.assertRaises(Exception): Utils.queryList(l, "asdsdasda") self.assertEqual(Utils.queryList(l, "Jhonny"), {1: c2}) r1 = Rental(1, 1, 1, 10000.0, 20000.0, None) r2 = Rental(2, 2, 2, 20000.0, 30000.0, None) l2 = {1: r1, 2: r2} with self.assertRaises(Exception): Utils.queryList(l2, "iwafmieafimfae")
def test_validate(self): c1 = Client("asd", "abcd") with self.assertRaises(Exception): ClientValidator.validate(c1) c1 = Client(1, "ab") with self.assertRaises(Exception): ClientValidator.validate(c1) c1 = Client(1, "abcd") ClientValidator.validate(c1)
def readClient(): """ Reads the data for a new client or for updating an existing client ValueError will be raised if the ID is not an int :return: client attributes as a dictionary """ cID = input("Client ID:") cID = int(cID) name = input("Client name:") return Client(cID, name)
def addClientsMoviesRentals(self): for i in range(1, 10): self.__clientService.addClient(Client(i, "Name" + str(i))) self.__movieService.addMovie( Movie(i, "Title" + str(i), "Desc" + str(i), "Genre" + str(i))) for i in range(1, 10): rDate = Utils.timestampFromDate( str(randint(1, 12)) + "/" + str(randint(1, 12)) + "/" + str(randint(2015, 2019))) self.__rentalService.addRental( Rental(i, randint(1, 9), randint(1, 9), rDate, rDate + Utils.CST_RENTAL_PERIOD, None))
def setUp(self): self.dataManager = DataManager(ClientValidator) self.clientService = ClientService(self.dataManager) for i in range(1, 10): self.clientService.addClient(Client(i, "Name" + str(i)))