diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_pagination.py | 19 | 
1 files changed, 19 insertions, 0 deletions
| diff --git a/tests/test_pagination.py b/tests/test_pagination.py index 80c33e2e..e1c2528b 100644 --- a/tests/test_pagination.py +++ b/tests/test_pagination.py @@ -412,6 +412,15 @@ class CustomPaginationSerializer(pagination.BasePaginationSerializer):      results_field = 'objects' +class CustomFooSerializer(serializers.Serializer): +    foo = serializers.CharField() + + +class CustomFooPaginationSerializer(pagination.PaginationSerializer): +    class Meta: +        object_serializer_class = CustomFooSerializer + +  class TestCustomPaginationSerializer(TestCase):      def setUp(self):          objects = ['john', 'paul', 'george', 'ringo'] @@ -434,6 +443,16 @@ class TestCustomPaginationSerializer(TestCase):          }          self.assertEqual(serializer.data, expected) +    def test_custom_pagination_serializer_with_custom_object_serializer(self): +        objects = [ +            {'foo': 'bar'}, +            {'foo': 'spam'} +        ] +        paginator = Paginator(objects, 1) +        page = paginator.page(1) +        serializer = CustomFooPaginationSerializer(page) +        serializer.data +  class NonIntegerPage(object): | 
