mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-10 19:56:59 +03:00
use actual user instead of fake user in permission composition test
This commit is contained in:
parent
fc08e0189f
commit
1f13b6f6b2
|
@ -5,11 +5,10 @@ import unittest
|
|||
import warnings
|
||||
|
||||
import django
|
||||
from django.contrib.auth.models import Group, Permission, User
|
||||
from django.contrib.auth.models import AnonymousUser, Group, Permission, User
|
||||
from django.db import models
|
||||
from django.test import TestCase
|
||||
from django.urls import ResolverMatch
|
||||
from django.utils.deprecation import CallableBool
|
||||
|
||||
from rest_framework import (
|
||||
HTTP_HEADER_ENCODING, authentication, generics, permissions, serializers,
|
||||
|
@ -543,39 +542,46 @@ class CustomPermissionsTests(TestCase):
|
|||
self.assertEqual(detail, self.custom_message)
|
||||
|
||||
|
||||
class FakeUser:
|
||||
def __init__(self, auth=False):
|
||||
self.is_authenticated = CallableBool(auth)
|
||||
|
||||
|
||||
class PermissionsCompositionTests(TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.username = 'john'
|
||||
self.email = 'lennon@thebeatles.com'
|
||||
self.password = 'password'
|
||||
self.user = User.objects.create_user(
|
||||
self.username,
|
||||
self.email,
|
||||
self.password
|
||||
)
|
||||
self.client.login(username=self.username, password=self.password)
|
||||
|
||||
def test_and_false(self):
|
||||
request = factory.get('/1', format='json')
|
||||
request.user = FakeUser(auth=False)
|
||||
request.user = AnonymousUser()
|
||||
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
||||
assert composed_perm().has_permission(request, None) is False
|
||||
|
||||
def test_and_true(self):
|
||||
request = factory.get('/1', format='json')
|
||||
request.user = FakeUser(auth=True)
|
||||
request.user = self.user
|
||||
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
||||
assert composed_perm().has_permission(request, None) is True
|
||||
|
||||
def test_or_false(self):
|
||||
request = factory.get('/1', format='json')
|
||||
request.user = FakeUser(auth=False)
|
||||
request.user = AnonymousUser()
|
||||
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
||||
assert composed_perm().has_permission(request, None) is True
|
||||
|
||||
def test_or_true(self):
|
||||
request = factory.get('/1', format='json')
|
||||
request.user = FakeUser(auth=True)
|
||||
request.user = self.user
|
||||
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
||||
assert composed_perm().has_permission(request, None) is True
|
||||
|
||||
def test_several_levels(self):
|
||||
request = factory.get('/1', format='json')
|
||||
request.user = FakeUser(auth=True)
|
||||
request.user = self.user
|
||||
composed_perm = (
|
||||
permissions.IsAuthenticated &
|
||||
permissions.IsAuthenticated &
|
||||
|
|
Loading…
Reference in New Issue
Block a user