def test_delete_client(self): number = self.get_last_number() + 1 Client.add_client(number, 'A', 'B', '2000-01-01') count = self.check_count() Client.delete_client(number) self.assertEqual(self.check_count(), count - 1) for client in Client.get_clients(): if client.number == number: self.assertFalse('Client was not deleted')
def test_add_client(self): count = self.check_count() number = self.get_last_number() + 1 Client.add_client(number, 'First', 'Last', '2020-02-02') self.assertEqual(self.check_count(), count + 1) for client in Client.get_clients(): if client.number == number: self.assertEqual(client.firstname, 'First') self.assertEqual(client.lastname, 'Last') self.assertEqual(client.birthday, date.fromisoformat('2020-02-02'))
def test_update_client(self): number = self.get_last_number() + 1 Client.add_client(number, 'Abc', 'Def', '2000-01-01') count = self.check_count() Client.update_client(number, 'Abb', 'Dee', '2001-02-03') self.assertEqual(self.check_count(), count) for client in Client.get_clients(): if client.number == number: self.assertEqual(client.firstname, 'Abb') self.assertEqual(client.lastname, 'Dee') self.assertEqual(client.birthday, date.fromisoformat('2001-02-03'))
def test_get_clients(self): self.assertEqual(self.check_count(), len(Client.get_clients()))
def get_last_number(self): return max(Client.get_clients(), key=lambda client: client.number).number
def show_clients(self): clients = Client.get_clients() View.show_clients(clients)