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)
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())