def test_missing_keys(self):
self.login()
self.assertFalse(has_sudo_privileges(self.request))
def test_revoked(self):
self.login()
grant_sudo_privileges(self.request)
revoke_sudo_privileges(self.request)
self.assertFalse(has_sudo_privileges(self.request))
def test_cookie_and_token_mismatch(self):
self.login()
self.request.COOKIES[COOKIE_NAME] = 'nope'
self.request.session[COOKIE_NAME] = 'abc123'
self.assertFalse(has_sudo_privileges(self.request))
def test_granted(self):
self.login()
grant_sudo_privileges(self.request)
self.assertTrue(has_sudo_privileges(self.request))
def test_untouched(self):
self.assertFalse(has_sudo_privileges(self.request))
def test_user_logged_out(self):
self.login()
grant_sudo_privileges(self.request)
self.assertTrue(has_sudo_privileges(self.request))
user_logged_out.send_robust(sender=User, request=self.request)
self.assertFalse(has_sudo_privileges(self.request))
def test_user_logged_in(self):
self.login()
user_logged_in.send_robust(sender=User, request=self.request)
self.assertTrue(has_sudo_privileges(self.request))
def has_sudo_privileges(self, request):
# Override me to alter behavior
return has_sudo_privileges(request)