utils: Fix test case failures due to caching (Closes: #813)

Reviewed-by: Sunil Mohan Adapa <sunil@medhas.org>
This commit is contained in:
Johannes Keyser 2017-04-27 02:28:54 +02:00 committed by Sunil Mohan Adapa
parent 9f0b20d7fa
commit 6c5953e8c2
No known key found for this signature in database
GPG Key ID: 43EA1CFF0AA7C5F2
2 changed files with 6 additions and 4 deletions

View File

@ -18,7 +18,7 @@
"""
Test module for Plinth's custom context processors.
"""
from unittest.mock import Mock
from unittest.mock import Mock, MagicMock
from django.http import HttpRequest
from django.test import TestCase
@ -38,6 +38,7 @@ class ContextProcessorsTestCase(TestCase):
request.path = '/aaa/bbb/ccc/'
request.user = Mock()
request.user.groups.filter().exists = Mock(return_value=True)
request.session = MagicMock()
response = cp.common(request)
self.assertIsNotNone(response)
@ -65,6 +66,7 @@ class ContextProcessorsTestCase(TestCase):
request.path = ''
request.user = Mock()
request.user.groups.filter().exists = Mock(return_value=True)
request.session = MagicMock()
response = cp.common(request)
self.assertEqual([], response['active_menu_urls'])

View File

@ -19,7 +19,7 @@
Test module for Plinth's custom middleware.
"""
from unittest.mock import Mock, patch
from unittest.mock import Mock, MagicMock, patch
from django.contrib.auth.models import AnonymousUser, User
from django.core.exceptions import PermissionDenied
@ -162,14 +162,14 @@ class TestAdminMiddleware(TestCase):
def test_that_admin_view_is_denied_for_usual_user(self):
"""Test that normal user is denied for an admin view"""
self.request.user.groups.filter().exists = Mock(return_value=False)
self.request.session = MagicMock()
self.assertRaises(PermissionDenied, self.middleware.process_view,
self.request, **self.kwargs)
def test_that_admin_view_is_allowed_for_admin_user(self):
"""Test that admin user is allowed for an admin view"""
self.request.user.groups.filter().exists = Mock(return_value=True)
self.request.session = MagicMock()
response = self.middleware.process_view(self.request, **self.kwargs)
self.assertIsNone(response)