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_show_clients(self):
number = self.get_last_number() + 1
Client.add_client(number, 'Testing', 'Controller', '2003-11-11')
f = io.StringIO()
with redirect_stdout(f):
Controller().show_clients()
s = f.getvalue()
self.assertIn('Testing', s)
self.assertIn('Controller', s)
self.assertIn('2003-11-11', s)
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 add_client(self):
number = View.get_data('number')
firstname = View.get_data('firstname').capitalize()
lastname = View.get_data('lastname').capitalize()
birthday = View.get_data('birthday')
Client.add_client(number, firstname, lastname, birthday)
