mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-02-03 21:24:33 +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 warnings
|
||||||
|
|
||||||
import django
|
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.db import models
|
||||||
from django.test import TestCase
|
from django.test import TestCase
|
||||||
from django.urls import ResolverMatch
|
from django.urls import ResolverMatch
|
||||||
from django.utils.deprecation import CallableBool
|
|
||||||
|
|
||||||
from rest_framework import (
|
from rest_framework import (
|
||||||
HTTP_HEADER_ENCODING, authentication, generics, permissions, serializers,
|
HTTP_HEADER_ENCODING, authentication, generics, permissions, serializers,
|
||||||
|
@ -543,39 +542,46 @@ class CustomPermissionsTests(TestCase):
|
||||||
self.assertEqual(detail, self.custom_message)
|
self.assertEqual(detail, self.custom_message)
|
||||||
|
|
||||||
|
|
||||||
class FakeUser:
|
|
||||||
def __init__(self, auth=False):
|
|
||||||
self.is_authenticated = CallableBool(auth)
|
|
||||||
|
|
||||||
|
|
||||||
class PermissionsCompositionTests(TestCase):
|
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):
|
def test_and_false(self):
|
||||||
request = factory.get('/1', format='json')
|
request = factory.get('/1', format='json')
|
||||||
request.user = FakeUser(auth=False)
|
request.user = AnonymousUser()
|
||||||
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
||||||
assert composed_perm().has_permission(request, None) is False
|
assert composed_perm().has_permission(request, None) is False
|
||||||
|
|
||||||
def test_and_true(self):
|
def test_and_true(self):
|
||||||
request = factory.get('/1', format='json')
|
request = factory.get('/1', format='json')
|
||||||
request.user = FakeUser(auth=True)
|
request.user = self.user
|
||||||
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
composed_perm = permissions.IsAuthenticated & permissions.AllowAny
|
||||||
assert composed_perm().has_permission(request, None) is True
|
assert composed_perm().has_permission(request, None) is True
|
||||||
|
|
||||||
def test_or_false(self):
|
def test_or_false(self):
|
||||||
request = factory.get('/1', format='json')
|
request = factory.get('/1', format='json')
|
||||||
request.user = FakeUser(auth=False)
|
request.user = AnonymousUser()
|
||||||
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
||||||
assert composed_perm().has_permission(request, None) is True
|
assert composed_perm().has_permission(request, None) is True
|
||||||
|
|
||||||
def test_or_true(self):
|
def test_or_true(self):
|
||||||
request = factory.get('/1', format='json')
|
request = factory.get('/1', format='json')
|
||||||
request.user = FakeUser(auth=True)
|
request.user = self.user
|
||||||
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
composed_perm = permissions.IsAuthenticated | permissions.AllowAny
|
||||||
assert composed_perm().has_permission(request, None) is True
|
assert composed_perm().has_permission(request, None) is True
|
||||||
|
|
||||||
def test_several_levels(self):
|
def test_several_levels(self):
|
||||||
request = factory.get('/1', format='json')
|
request = factory.get('/1', format='json')
|
||||||
request.user = FakeUser(auth=True)
|
request.user = self.user
|
||||||
composed_perm = (
|
composed_perm = (
|
||||||
permissions.IsAuthenticated &
|
permissions.IsAuthenticated &
|
||||||
permissions.IsAuthenticated &
|
permissions.IsAuthenticated &
|
||||||
|
|
Loading…
Reference in New Issue
Block a user