serializers_test.py 文件源码

 复制代码def test_collection_serializer():
    """
    Test for CollectionSerializer
    """
    collection = factories.CollectionFactory()
    videos = [factories.VideoFactory(collection=collection) for _ in range(3)]
    expected = {
        'key': collection.hexkey,
        'created_at': DateTimeField().to_representation(collection.created_at),
        'title': collection.title,
        'description': collection.description,
        'videos': serializers.VideoSerializer(videos, many=True).data,
        'video_count': len(videos),
        'view_lists': [],
        'admin_lists': [],
        'is_admin': None
    }
    expected['videos'].sort(key=lambda x: x['key'])
    serialized_data = serializers.CollectionSerializer(collection).data
    serialized_data['videos'].sort(key=lambda x: x['key'])
    assert serialized_data == expected