class TestDateHelper(unittest.TestCase): def setUp(self): self.dh = DateHelper() self.date_str = "1999-08-01" self.datetime_str = "1998-05-02 14:08:00" self.date_obj = self.dh.make_date(self.date_str) self.datetime_obj = self.dh.make_datetime(self.datetime_str) def test_print_date(self): expected = "01 Aug 1999" actual = self.dh.print_date(self.date_obj) self.assertEqual(expected, actual) def test_print_datetime(self): expected = "14:08:00 02 May 1998" actual = self.dh.print_datetime(self.datetime_obj) self.assertEqual(expected, actual) def test_print_nice(self): expected = "2:08pm, 2 May 1998" actual = self.dh.print_nice(self.datetime_obj) self.assertEqual(expected, actual)
class TestAnimal(unittest.TestCase): def setUp(self): self.dh = DateHelper() self.vet = Vet("Mark", "Bridges", 3) self.address = Address("9", "Big Road", "Vatican City", "PO1 1PE") self.owner = Owner("Kevin", "Stevens", self.address, "015825536874", "*****@*****.**") self.animal = Animal( "Fluff", self.dh.make_date("2018-01-04"), "Dog", "Greyhound", self.owner, self.vet, self.dh.make_datetime("2020-08-09 13:52:01"), ) self.now = self.dh.make_datetime("2020-08-10 12:20:52") def test_animal_has_name(self): expected = "Fluff" actual = self.animal.name self.assertEqual(expected, actual) def test_animal_has_dob(self): expected = "04 Jan 2018" actual = self.dh.print_date(self.animal.dob) self.assertEqual(expected, actual) def test_animal_species(self): expected = "Dog" actual = self.animal.species self.assertEqual(expected, actual) def test_animal_breed(self): expected = "Greyhound" actual = self.animal.breed self.assertEqual(expected, actual) def test_animal_has_owner(self): expected = "*****@*****.**" actual = self.animal.owner.email self.assertEqual(expected, actual) def test_animal_has_vet(self): expected = "Mark" actual = self.animal.vet.first_name self.assertEqual(expected, actual) def test_registered_date(self): expected = "13:52:01 09 Aug 2020" actual = self.dh.print_datetime(self.animal.date_registered) self.assertEqual(expected, actual) def test_checked_in_default(self): expected = False actual = self.animal.checked_in self.assertEqual(expected, actual) def test_animal_has_None_id(self): actual = self.animal.id self.assertIsNone(actual) def test_animal_can_get_id(self): self.animal.id = 7816 expected = 7816 actual = self.animal.id self.assertEqual(expected, actual) def test_animal_has_num_records(self): expected = 0 actual = self.animal.records self.assertEqual(expected, actual) def test_set_num_records(self): self.animal.set_records(5) expected = 5 actual = self.animal.records self.assertEqual(expected, actual) def test_set_num_records_different_number(self): self.animal.set_records(10) expected = 10 actual = self.animal.records self.assertEqual(expected, actual) def test_printable_age(self): expected = "2 years old" actual = self.animal.get_age(self.now) self.assertEqual(expected, actual) def test_printed_dob(self): expected = "04 Jan 2018" actual = self.animal.print_dob("print_date") self.assertEqual(expected, actual) def test_nice_printed_date_registered(self): expected = "1:52pm, 9 Aug 2020" actual = self.animal.print_registered("print_nice") self.assertEqual(expected, actual)