diff options
| author | Tom Christie | 2014-08-29 12:35:53 +0100 | 
|---|---|---|
| committer | Tom Christie | 2014-08-29 12:35:53 +0100 | 
| commit | b3253b42836acd123224e88c0927f1ee6a031d94 (patch) | |
| tree | 4202bf23aaf5fd900e90a3a66a2e4f80d03e5443 /tests/test_permissions.py | |
| parent | b8c8d10a18741b76355ed7035655d0101c1d778a (diff) | |
| download | django-rest-framework-b3253b42836acd123224e88c0927f1ee6a031d94.tar.bz2 | |
Remove `.model` usage in tests.
Remove the shortcut `.model` view attribute usage from test cases.
Diffstat (limited to 'tests/test_permissions.py')
| -rw-r--r-- | tests/test_permissions.py | 24 | 
1 files changed, 19 insertions, 5 deletions
| diff --git a/tests/test_permissions.py b/tests/test_permissions.py index 93f8020f..b90ba4f1 100644 --- a/tests/test_permissions.py +++ b/tests/test_permissions.py @@ -3,7 +3,7 @@ from django.contrib.auth.models import User, Permission, Group  from django.db import models  from django.test import TestCase  from django.utils import unittest -from rest_framework import generics, status, permissions, authentication, HTTP_HEADER_ENCODING +from rest_framework import generics, serializers, status, permissions, authentication, HTTP_HEADER_ENCODING  from rest_framework.compat import guardian, get_model_name  from rest_framework.filters import DjangoObjectPermissionsFilter  from rest_framework.test import APIRequestFactory @@ -13,14 +13,21 @@ import base64  factory = APIRequestFactory() +class BasicSerializer(serializers.ModelSerializer): +    class Meta: +        model = BasicModel + +  class RootView(generics.ListCreateAPIView): -    model = BasicModel +    queryset = BasicModel.objects.all() +    serializer_class = BasicSerializer      authentication_classes = [authentication.BasicAuthentication]      permission_classes = [permissions.DjangoModelPermissions]  class InstanceView(generics.RetrieveUpdateDestroyAPIView): -    model = BasicModel +    queryset = BasicModel.objects.all() +    serializer_class = BasicSerializer      authentication_classes = [authentication.BasicAuthentication]      permission_classes = [permissions.DjangoModelPermissions] @@ -167,6 +174,11 @@ class BasicPermModel(models.Model):          ) +class BasicPermSerializer(serializers.ModelSerializer): +    class Meta: +        model = BasicPermModel + +  # Custom object-level permission, that includes 'view' permissions  class ViewObjectPermissions(permissions.DjangoObjectPermissions):      perms_map = { @@ -181,7 +193,8 @@ class ViewObjectPermissions(permissions.DjangoObjectPermissions):  class ObjectPermissionInstanceView(generics.RetrieveUpdateDestroyAPIView): -    model = BasicPermModel +    queryset = BasicPermModel.objects.all() +    serializer_class = BasicPermSerializer      authentication_classes = [authentication.BasicAuthentication]      permission_classes = [ViewObjectPermissions] @@ -189,7 +202,8 @@ object_permissions_view = ObjectPermissionInstanceView.as_view()  class ObjectPermissionListView(generics.ListAPIView): -    model = BasicPermModel +    queryset = BasicPermModel.objects.all() +    serializer_class = BasicPermSerializer      authentication_classes = [authentication.BasicAuthentication]      permission_classes = [ViewObjectPermissions] | 
