def test_superuser_required_mixin_not_supersuer(self):
"""Test for mixin requiring a superuser when user isn't a superuser."""
self.client.login(username=self.username, password=self.password)
with self.assertRaises(PermissionDenied):
request = WSGIHandler()
request.user = self.user
mixin = SuperuserRequiredViewMixin()
mixin.dispatch(request)
def test_superuser_required_mixin_not_supersuer(self):
"""Test for mixin requiring a superuser when user isn't a superuser."""
self.client.login(username=self.username, password=self.password)
with self.assertRaises(PermissionDenied):
request = WSGIHandler()
request.user = self.user
mixin = SuperuserRequiredViewMixin()
mixin.dispatch(request)
def test_staff_required_mixin_is_staff(self, dispatch):
"""Test for mixin requiring a staff user when user is staff."""
dispatch.return_value = 'worked'
self.client.login(username=self.username, password=self.password)
self.user.is_staff = True
request = WSGIHandler()
request.user = self.user
mixin = StaffRequiredViewMixin()
actual_return = mixin.dispatch(request)
self.assertEqual(actual_return, dispatch.return_value)
def test_staff_required_mixin_is_staff(self, dispatch):
"""Test for mixin requiring a staff user when user is staff."""
dispatch.return_value = 'worked'
self.client.login(username=self.username, password=self.password)
self.user.is_staff = True
request = WSGIHandler()
request.user = self.user
mixin = StaffRequiredViewMixin()
actual_return = mixin.dispatch(request)
self.assertEqual(actual_return, dispatch.return_value)
