def test_encoded_video_set_output(self):
"""
Tests for basic structure of EncodedVideoSetSerializer
"""
video = Video.objects.create(**constants.VIDEO_DICT_FISH)
EncodedVideo.objects.create(
video=video,
profile=Profile.objects.get(profile_name="desktop"),
**constants.ENCODED_VIDEO_DICT_DESKTOP
)
EncodedVideo.objects.create(
video=video,
profile=Profile.objects.get(profile_name="mobile"),
**constants.ENCODED_VIDEO_DICT_MOBILE
)
EncodedVideo.objects.create(
video=video,
profile=Profile.objects.get(profile_name="hls"),
**constants.ENCODED_VIDEO_DICT_HLS
)
result = VideoSerializer(video).data
# Check for 3 EncodedVideo entries
self.assertEqual(len(result.get("encoded_videos")), 3)
# Check for original Video data
matching_dict = {k: v for k, v in result.items() if k in constants.VIDEO_DICT_FISH}
assert constants.VIDEO_DICT_FISH == matching_dict
