Change DjangoField to PermissionField

This commit is contained in:
Olivia Rodriguez Valdes 2019-03-01 12:42:29 -05:00
parent ad2342a08b
commit 76cbad1fb2
2 changed files with 6 additions and 6 deletions

View File

@ -154,12 +154,12 @@ class DjangoConnectionField(ConnectionField):
)
class DjangoField(Field):
class PermissionField(Field):
"""Class to manage permission for fields"""
def __init__(self, type, permissions, permissions_resolver=auth_resolver, *args, **kwargs):
"""Get permissions to access a field"""
super(DjangoField, self).__init__(type, *args, **kwargs)
super(PermissionField, self).__init__(type, *args, **kwargs)
self.permissions = permissions
self.permissions_resolver = permissions_resolver

View File

@ -1,6 +1,6 @@
from unittest import TestCase
from django.core.exceptions import PermissionDenied
from graphene_django.fields import DjangoField
from graphene_django.fields import PermissionField
class MyInstance(object):
@ -10,11 +10,11 @@ class MyInstance(object):
return "resolver method"
class DjangoPermissionFieldTests(TestCase):
class PermissionFieldTests(TestCase):
def test_permission_field(self):
MyType = object()
field = DjangoField(MyType, permissions=['perm1', 'perm2'], source='resolver')
field = PermissionField(MyType, permissions=['perm1', 'perm2'], source='resolver')
resolver = field.get_resolver(None)
class Viewer(object):
@ -30,7 +30,7 @@ class DjangoPermissionFieldTests(TestCase):
def test_permission_field_without_permission(self):
MyType = object()
field = DjangoField(MyType, permissions=['perm1', 'perm2'], source='resolver')
field = PermissionField(MyType, permissions=['perm1', 'perm2'], source='resolver')
resolver = field.get_resolver(field.resolver)
class Viewer(object):