graphene-django/graphene_django/tests/test_utils.py
2019-01-04 14:24:55 -05:00

33 lines
956 B
Python

from ..utils import get_model_fields, has_permissions
from .models import Film, Reporter
def test_get_model_fields_no_duplication():
reporter_fields = get_model_fields(Reporter)
reporter_name_set = set([field[0] for field in reporter_fields])
assert len(reporter_fields) == len(reporter_name_set)
film_fields = get_model_fields(Film)
film_name_set = set([field[0] for field in film_fields])
assert len(film_fields) == len(film_name_set)
def test_has_permissions():
class Viewer(object):
@staticmethod
def has_perm(permission):
return permission
viewer_as_perm = has_permissions(Viewer(), [False, True, False])
assert viewer_as_perm
def test_viewer_without_permissions():
class Viewer(object):
@staticmethod
def has_perm(permission):
return permission
viewer_as_perm = has_permissions(Viewer(), [False, False, False])
assert not viewer_as_perm