def test_removeClient(self): self.assertEqual(len(self._repo), 0) self._repo.addClient(client(2, "Mihai")) self._repo.addClient(client(45, "Andrei")) self.assertEqual(len(self._repo), 2) self._repo.removeClient(1) self.assertEqual(len(self._repo), 1) self._repo.removeClient(0) self.assertEqual(len(self._repo), 0)
def __loadFromFile(self): try: f = open(self.__fName, "r") line = f.readline().strip() while line != "": attrs = line.split(",") clients = client(int(attrs[0]), attrs[1]) clientRep.addClient(self, clients) line = f.readline().strip() except IOError: raise RepositoryException("Error saving file") finally: f.close()
def test_addClient(self): self.assertEqual(len(self._repo), 0) self._repo.addClient(client(1, "Mihai")) self._repo.addClient(client(2, "Andrei")) self.assertEqual(len(self._repo), 2)
def test_findIdClient(self): self._repo.addClient(client(1, "Mihai")) self._repo.addClient(client(23, "Ion")) self._repo.addClient(client(99, "Raul")) assert self._repo.findIdClient(23) == 1
def test_updateClient(self): self._repo.addClient(client(1, "Mihai")) self._repo.updateClient([2, "Ion"], 1)
def readUpClient(id): while True: name = input("New name = ") return client(id, name)
def readClient(id): name = input("Name = ") return client(id, name)