def test_serialize_course_related_models(offered_by):
"""
Verify that a serialized course contains attributes for related objects
"""
course = CourseFactory(offered_by=offered_by,
topics=CourseTopicFactory.create_batch(3))
serializer = CourseSerializer(course)
assert len(serializer.data["topics"]) == 3
assert "name" in serializer.data["topics"][0].keys()
assert len(serializer.data["runs"]) == 3
def test_serialize_course_related_models():
"""
Verify that a serialized course contains attributes for related objects
"""
course = CourseFactory(
topics=CourseTopicFactory.create_batch(3),
prices=CoursePriceFactory.create_batch(2),
instructors=CourseInstructorFactory.create_batch(2),
)
serializer = CourseSerializer(course)
assert len(serializer.data["prices"]) == 2
for attr in ("mode", "price"):
assert attr in serializer.data["prices"][0].keys()
assert len(serializer.data["instructors"]) == 2
for attr in ("first_name", "last_name"):
assert attr in serializer.data["instructors"][0].keys()
assert len(serializer.data["topics"]) == 3
assert "name" in serializer.data["topics"][0].keys()