|
3 | 3 | import base64
|
4 | 4 | import unittest
|
5 | 5 | import warnings
|
| 6 | +from unittest import mock |
6 | 7 |
|
7 | 8 | import django
|
8 | 9 | from django.contrib.auth.models import AnonymousUser, Group, Permission, User
|
9 | 10 | from django.db import models
|
10 | 11 | from django.test import TestCase
|
11 | 12 | from django.urls import ResolverMatch
|
12 | 13 |
|
| 14 | +import pytest |
| 15 | + |
13 | 16 | from rest_framework import (
|
14 | 17 | HTTP_HEADER_ENCODING, authentication, generics, permissions, serializers,
|
15 | 18 | status, views
|
16 | 19 | )
|
17 |
| -from rest_framework.compat import is_guardian_installed |
| 20 | +from rest_framework.compat import PY36, is_guardian_installed |
18 | 21 | from rest_framework.filters import DjangoObjectPermissionsFilter
|
19 | 22 | from rest_framework.routers import DefaultRouter
|
20 | 23 | from rest_framework.test import APIRequestFactory
|
@@ -600,3 +603,87 @@ def test_several_levels_and_precedence(self):
|
600 | 603 | permissions.IsAuthenticated
|
601 | 604 | )
|
602 | 605 | assert composed_perm().has_permission(request, None) is True
|
| 606 | + |
| 607 | + @pytest.mark.skipif(not PY36, reason="assert_called_once() not available") |
| 608 | + def test_or_lazyness(self): |
| 609 | + request = factory.get('/1', format='json') |
| 610 | + request.user = AnonymousUser() |
| 611 | + |
| 612 | + with mock.patch.object(permissions.AllowAny, 'has_permission', return_value=True) as mock_allow: |
| 613 | + with mock.patch.object(permissions.IsAuthenticated, 'has_permission', return_value=False) as mock_deny: |
| 614 | + composed_perm = (permissions.AllowAny | permissions.IsAuthenticated) |
| 615 | + hasperm = composed_perm().has_permission(request, None) |
| 616 | + self.assertIs(hasperm, True) |
| 617 | + mock_allow.assert_called_once() |
| 618 | + mock_deny.assert_not_called() |
| 619 | + |
| 620 | + with mock.patch.object(permissions.AllowAny, 'has_permission', return_value=True) as mock_allow: |
| 621 | + with mock.patch.object(permissions.IsAuthenticated, 'has_permission', return_value=False) as mock_deny: |
| 622 | + composed_perm = (permissions.IsAuthenticated | permissions.AllowAny) |
| 623 | + hasperm = composed_perm().has_permission(request, None) |
| 624 | + self.assertIs(hasperm, True) |
| 625 | + mock_deny.assert_called_once() |
| 626 | + mock_allow.assert_called_once() |
| 627 | + |
| 628 | + @pytest.mark.skipif(not PY36, reason="assert_called_once() not available") |
| 629 | + def test_object_or_lazyness(self): |
| 630 | + request = factory.get('/1', format='json') |
| 631 | + request.user = AnonymousUser() |
| 632 | + |
| 633 | + with mock.patch.object(permissions.AllowAny, 'has_object_permission', return_value=True) as mock_allow: |
| 634 | + with mock.patch.object(permissions.IsAuthenticated, 'has_object_permission', return_value=False) as mock_deny: |
| 635 | + composed_perm = (permissions.AllowAny | permissions.IsAuthenticated) |
| 636 | + hasperm = composed_perm().has_object_permission(request, None, None) |
| 637 | + self.assertIs(hasperm, True) |
| 638 | + mock_allow.assert_called_once() |
| 639 | + mock_deny.assert_not_called() |
| 640 | + |
| 641 | + with mock.patch.object(permissions.AllowAny, 'has_object_permission', return_value=True) as mock_allow: |
| 642 | + with mock.patch.object(permissions.IsAuthenticated, 'has_object_permission', return_value=False) as mock_deny: |
| 643 | + composed_perm = (permissions.IsAuthenticated | permissions.AllowAny) |
| 644 | + hasperm = composed_perm().has_object_permission(request, None, None) |
| 645 | + self.assertIs(hasperm, True) |
| 646 | + mock_deny.assert_called_once() |
| 647 | + mock_allow.assert_called_once() |
| 648 | + |
| 649 | + @pytest.mark.skipif(not PY36, reason="assert_called_once() not available") |
| 650 | + def test_and_lazyness(self): |
| 651 | + request = factory.get('/1', format='json') |
| 652 | + request.user = AnonymousUser() |
| 653 | + |
| 654 | + with mock.patch.object(permissions.AllowAny, 'has_permission', return_value=True) as mock_allow: |
| 655 | + with mock.patch.object(permissions.IsAuthenticated, 'has_permission', return_value=False) as mock_deny: |
| 656 | + composed_perm = (permissions.AllowAny & permissions.IsAuthenticated) |
| 657 | + hasperm = composed_perm().has_permission(request, None) |
| 658 | + self.assertIs(hasperm, False) |
| 659 | + mock_allow.assert_called_once() |
| 660 | + mock_deny.assert_called_once() |
| 661 | + |
| 662 | + with mock.patch.object(permissions.AllowAny, 'has_permission', return_value=True) as mock_allow: |
| 663 | + with mock.patch.object(permissions.IsAuthenticated, 'has_permission', return_value=False) as mock_deny: |
| 664 | + composed_perm = (permissions.IsAuthenticated & permissions.AllowAny) |
| 665 | + hasperm = composed_perm().has_permission(request, None) |
| 666 | + self.assertIs(hasperm, False) |
| 667 | + mock_allow.assert_not_called() |
| 668 | + mock_deny.assert_called_once() |
| 669 | + |
| 670 | + @pytest.mark.skipif(not PY36, reason="assert_called_once() not available") |
| 671 | + def test_object_and_lazyness(self): |
| 672 | + request = factory.get('/1', format='json') |
| 673 | + request.user = AnonymousUser() |
| 674 | + |
| 675 | + with mock.patch.object(permissions.AllowAny, 'has_object_permission', return_value=True) as mock_allow: |
| 676 | + with mock.patch.object(permissions.IsAuthenticated, 'has_object_permission', return_value=False) as mock_deny: |
| 677 | + composed_perm = (permissions.AllowAny & permissions.IsAuthenticated) |
| 678 | + hasperm = composed_perm().has_object_permission(request, None, None) |
| 679 | + self.assertIs(hasperm, False) |
| 680 | + mock_allow.assert_called_once() |
| 681 | + mock_deny.assert_called_once() |
| 682 | + |
| 683 | + with mock.patch.object(permissions.AllowAny, 'has_object_permission', return_value=True) as mock_allow: |
| 684 | + with mock.patch.object(permissions.IsAuthenticated, 'has_object_permission', return_value=False) as mock_deny: |
| 685 | + composed_perm = (permissions.IsAuthenticated & permissions.AllowAny) |
| 686 | + hasperm = composed_perm().has_object_permission(request, None, None) |
| 687 | + self.assertIs(hasperm, False) |
| 688 | + mock_allow.assert_not_called() |
| 689 | + mock_deny.assert_called_once() |
0 commit comments