diff options
| author | django-bot <ops@djangoproject.com> | 2022-02-03 20:24:19 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-02-07 20:37:05 +0100 |
| commit | 9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch) | |
| tree | f0506b668a013d0063e5fba3dbf4863b466713ba /tests/signed_cookies_tests/tests.py | |
| parent | f68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff) | |
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/signed_cookies_tests/tests.py')
| -rw-r--r-- | tests/signed_cookies_tests/tests.py | 57 |
1 files changed, 28 insertions, 29 deletions
diff --git a/tests/signed_cookies_tests/tests.py b/tests/signed_cookies_tests/tests.py index b7ea902bf4..cca30bb005 100644 --- a/tests/signed_cookies_tests/tests.py +++ b/tests/signed_cookies_tests/tests.py @@ -5,62 +5,61 @@ from django.test.utils import freeze_time class SignedCookieTest(SimpleTestCase): - def test_can_set_and_read_signed_cookies(self): response = HttpResponse() - response.set_signed_cookie('c', 'hello') - self.assertIn('c', response.cookies) - self.assertTrue(response.cookies['c'].value.startswith('hello:')) + response.set_signed_cookie("c", "hello") + self.assertIn("c", response.cookies) + self.assertTrue(response.cookies["c"].value.startswith("hello:")) request = HttpRequest() - request.COOKIES['c'] = response.cookies['c'].value - value = request.get_signed_cookie('c') - self.assertEqual(value, 'hello') + request.COOKIES["c"] = response.cookies["c"].value + value = request.get_signed_cookie("c") + self.assertEqual(value, "hello") def test_can_use_salt(self): response = HttpResponse() - response.set_signed_cookie('a', 'hello', salt='one') + response.set_signed_cookie("a", "hello", salt="one") request = HttpRequest() - request.COOKIES['a'] = response.cookies['a'].value - value = request.get_signed_cookie('a', salt='one') - self.assertEqual(value, 'hello') + request.COOKIES["a"] = response.cookies["a"].value + value = request.get_signed_cookie("a", salt="one") + self.assertEqual(value, "hello") with self.assertRaises(signing.BadSignature): - request.get_signed_cookie('a', salt='two') + request.get_signed_cookie("a", salt="two") def test_detects_tampering(self): response = HttpResponse() - response.set_signed_cookie('c', 'hello') + response.set_signed_cookie("c", "hello") request = HttpRequest() - request.COOKIES['c'] = response.cookies['c'].value[:-2] + '$$' + request.COOKIES["c"] = response.cookies["c"].value[:-2] + "$$" with self.assertRaises(signing.BadSignature): - request.get_signed_cookie('c') + request.get_signed_cookie("c") def test_default_argument_suppresses_exceptions(self): response = HttpResponse() - response.set_signed_cookie('c', 'hello') + response.set_signed_cookie("c", "hello") request = HttpRequest() - request.COOKIES['c'] = response.cookies['c'].value[:-2] + '$$' - self.assertIsNone(request.get_signed_cookie('c', default=None)) + request.COOKIES["c"] = response.cookies["c"].value[:-2] + "$$" + self.assertIsNone(request.get_signed_cookie("c", default=None)) def test_max_age_argument(self): - value = 'hello' + value = "hello" with freeze_time(123456789): response = HttpResponse() - response.set_signed_cookie('c', value) + response.set_signed_cookie("c", value) request = HttpRequest() - request.COOKIES['c'] = response.cookies['c'].value - self.assertEqual(request.get_signed_cookie('c'), value) + request.COOKIES["c"] = response.cookies["c"].value + self.assertEqual(request.get_signed_cookie("c"), value) with freeze_time(123456800): - self.assertEqual(request.get_signed_cookie('c', max_age=12), value) - self.assertEqual(request.get_signed_cookie('c', max_age=11), value) + self.assertEqual(request.get_signed_cookie("c", max_age=12), value) + self.assertEqual(request.get_signed_cookie("c", max_age=11), value) with self.assertRaises(signing.SignatureExpired): - request.get_signed_cookie('c', max_age=10) + request.get_signed_cookie("c", max_age=10) - @override_settings(SECRET_KEY=b'\xe7') + @override_settings(SECRET_KEY=b"\xe7") def test_signed_cookies_with_binary_key(self): response = HttpResponse() - response.set_signed_cookie('c', 'hello') + response.set_signed_cookie("c", "hello") request = HttpRequest() - request.COOKIES['c'] = response.cookies['c'].value - self.assertEqual(request.get_signed_cookie('c'), 'hello') + request.COOKIES["c"] = response.cookies["c"].value + self.assertEqual(request.get_signed_cookie("c"), "hello") |
