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)