test_context_processor.py 629 B

123456789101112131415
  1. from django.http import HttpRequest
  2. from django.middleware.csrf import _compare_salted_tokens as equivalent_tokens
  3. from django.template.context_processors import csrf
  4. from django.test import SimpleTestCase
  5. from django.utils.encoding import force_text
  6. class TestContextProcessor(SimpleTestCase):
  7. def test_force_text_on_token(self):
  8. request = HttpRequest()
  9. test_token = '1bcdefghij2bcdefghij3bcdefghij4bcdefghij5bcdefghij6bcdefghijABCD'
  10. request.META['CSRF_COOKIE'] = test_token
  11. token = csrf(request).get('csrf_token')
  12. self.assertTrue(equivalent_tokens(force_text(token), test_token))