Misago/misago/core/tests/test_serializers.py

from rest_framework import serializers

from django.test import TestCase

from misago.categories.models import Category
from misago.core.serializers import MutableFields
from misago.threads import testutils
from misago.threads.models import Thread


class MutableFieldsSerializerTests(TestCase):
    def test_subset_fields(self):
        """classmethod subset_fields creates new serializer"""
        category = Category.objects.get(slug='first-category')
        thread = testutils.post_thread(category=category)

        fields = ('id', 'title', 'replies', 'last_poster_name')

        serializer = TestSerializer.subset_fields(*fields)
        self.assertEqual(
            serializer.__name__,
            'TestSerializerIdTitleRepliesLastPosterNameSubset'
        )
        self.assertEqual(serializer.Meta.fields, fields)

        serialized_thread = serializer(thread).data
        self.assertEqual(serialized_thread, {
            'id': thread.id,
            'title': thread.title,
            'replies': thread.replies,
            'last_poster_name': thread.last_poster_name,
        })

        self.assertFalse(TestSerializer.Meta.fields == serializer.Meta.fields)

    def test_exclude_fields(self):
        """classmethod exclude_fields creates new serializer"""
        category = Category.objects.get(slug='first-category')
        thread = testutils.post_thread(category=category)

        kept_fields = ('id', 'title', 'weight')
        removed_fields = tuple(set(TestSerializer.Meta.fields) - set(kept_fields))

        serializer = TestSerializer.exclude_fields(*removed_fields)
        self.assertEqual(serializer.__name__, 'TestSerializerIdTitleWeightSubset')
        self.assertEqual(serializer.Meta.fields, kept_fields)

        serialized_thread = serializer(thread).data
        self.assertEqual(serialized_thread, {
            'id': thread.id,
            'title': thread.title,
            'weight': thread.weight,
        })

        self.assertFalse(TestSerializer.Meta.fields == serializer.Meta.fields)

    def test_extend_fields(self):
        """classmethod extend_fields creates new serializer"""
        category = Category.objects.get(slug='first-category')
        thread = testutils.post_thread(category=category)

        added_fields = ('category',)

        serializer = TestSerializer.extend_fields(*added_fields)

        serialized_thread = serializer(thread).data
        self.assertEqual(serialized_thread['category'], category.pk)


class TestSerializer(serializers.ModelSerializer, MutableFields):
    url = serializers.SerializerMethodField()

    class Meta:
        model = Thread
        fields = (
            'id',
            'title',
            'replies',
            'has_unapproved_posts',
            'started_on',
            'last_post_on',
            'last_post_is_event',
            'last_post',
            'last_poster_name',
            'is_unapproved',
            'is_hidden',
            'is_closed',
            'weight',
        )