def test_make_jsonable_with_single_object_as_value(self):
athlete = Athlete("user", "pass", "firstname", "lastname", "[email protected]")
data = make_jsonable(athlete)
expected = {
'achievements': [],
'first_name': 'firstname',
'last_name': 'lastname',
'about_me': '',
'avatar': '',
'address': '',
'birth_date': None,
'email': '[email protected]'
}
self.assertEqual(data, expected)
self.assertEqual(data, athlete.to_dict())
def test_make_jsonable_with_list_of_objects_as_value(self):
athlete1 = Athlete("user1", "pass", "firstname1", "lastname", "[email protected]")
athlete2 = Athlete("user2", "pass", "firstname2", "lastname", "[email protected]")
athlete3 = Athlete("user3", "pass", "firstname3", "lastname", "[email protected]")
athlete_list = [athlete1, athlete2, athlete3]
data = make_jsonable(athlete_list)
expected = [
{
'achievements': [],
'first_name': 'firstname1',
'last_name': 'lastname',
'about_me': '',
'avatar': '',
'address': '',
'birth_date': None,
'email': '[email protected]'
},
{
'achievements': [],
'first_name': 'firstname2',
'last_name': 'lastname',
'about_me': '',
'avatar': '',
'address': '',
'birth_date': None,
'email': '[email protected]'
},
{
'achievements': [],
'first_name': 'firstname3',
'last_name': 'lastname',
'about_me': '',
'avatar': '',
'address': '',
'birth_date': None,
'email': '[email protected]'
}
]
self.assertEqual(data, expected)
expected = []
expected.append(athlete1.to_dict())
expected.append(athlete2.to_dict())
expected.append(athlete3.to_dict())
self.assertEqual(data, expected)
