|
1 | 1 | # Simple test suite for http/cookies.py |
2 | | - |
| 2 | +import base64 |
3 | 3 | import copy |
4 | 4 | import unittest |
5 | 5 | import doctest |
6 | 6 | from http import cookies |
7 | 7 | import pickle |
8 | 8 | from test import support |
| 9 | +from test.test_ctypes.test_repr import base |
9 | 10 |
|
10 | 11 |
|
11 | 12 | class CookieTests(unittest.TestCase): |
@@ -175,17 +176,19 @@ def test_load(self): |
175 | 176 |
|
176 | 177 | self.assertEqual(C.output(['path']), |
177 | 178 | 'Set-Cookie: Customer="WILE_E_COYOTE"; Path=/acme') |
178 | | - self.assertEqual(C.js_output(), r""" |
| 179 | + cookie_encoded = base64.b64encode(b'Customer="WILE_E_COYOTE"; Path=/acme; Version=1').decode('ascii') |
| 180 | + self.assertEqual(C.js_output(), fr""" |
179 | 181 | <script type="text/javascript"> |
180 | 182 | <!-- begin hiding |
181 | | - document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme; Version=1"; |
| 183 | + document.cookie = atob("{cookie_encoded}"); |
182 | 184 | // end hiding --> |
183 | 185 | </script> |
184 | 186 | """) |
185 | | - self.assertEqual(C.js_output(['path']), r""" |
| 187 | + cookie_encoded = base64.b64encode(b'Customer="WILE_E_COYOTE"; Path=/acme').decode('ascii') |
| 188 | + self.assertEqual(C.js_output(['path']), fr""" |
186 | 189 | <script type="text/javascript"> |
187 | 190 | <!-- begin hiding |
188 | | - document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme"; |
| 191 | + document.cookie = atob("{cookie_encoded}"); |
189 | 192 | // end hiding --> |
190 | 193 | </script> |
191 | 194 | """) |
@@ -290,17 +293,19 @@ def test_quoted_meta(self): |
290 | 293 |
|
291 | 294 | self.assertEqual(C.output(['path']), |
292 | 295 | 'Set-Cookie: Customer="WILE_E_COYOTE"; Path=/acme') |
293 | | - self.assertEqual(C.js_output(), r""" |
| 296 | + expected_encoded_cookie = base64.b64encode(b'Customer=\"WILE_E_COYOTE\"; Path=/acme; Version=1').decode('ascii') |
| 297 | + self.assertEqual(C.js_output(), fr""" |
294 | 298 | <script type="text/javascript"> |
295 | 299 | <!-- begin hiding |
296 | | - document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme; Version=1"; |
| 300 | + document.cookie = atob("{expected_encoded_cookie}"); |
297 | 301 | // end hiding --> |
298 | 302 | </script> |
299 | 303 | """) |
300 | | - self.assertEqual(C.js_output(['path']), r""" |
| 304 | + expected_encoded_cookie = base64.b64encode(b'Customer=\"WILE_E_COYOTE\"; Path=/acme').decode('ascii') |
| 305 | + self.assertEqual(C.js_output(['path']), fr""" |
301 | 306 | <script type="text/javascript"> |
302 | 307 | <!-- begin hiding |
303 | | - document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme"; |
| 308 | + document.cookie = atob("{expected_encoded_cookie}"); |
304 | 309 | // end hiding --> |
305 | 310 | </script> |
306 | 311 | """) |
@@ -391,13 +396,16 @@ def test_setter(self): |
391 | 396 | self.assertEqual( |
392 | 397 | M.output(), |
393 | 398 | "Set-Cookie: %s=%s; Path=/foo" % (i, "%s_coded_val" % i)) |
| 399 | + expected_encoded_cookie = base64.b64encode( |
| 400 | + ("%s=%s; Path=/foo" % (i, "%s_coded_val" % i)).encode("ascii") |
| 401 | + ).decode('ascii') |
394 | 402 | expected_js_output = """ |
395 | 403 | <script type="text/javascript"> |
396 | 404 | <!-- begin hiding |
397 | | - document.cookie = "%s=%s; Path=/foo"; |
| 405 | + document.cookie = atob("%s"); |
398 | 406 | // end hiding --> |
399 | 407 | </script> |
400 | | - """ % (i, "%s_coded_val" % i) |
| 408 | + """ % (expected_encoded_cookie,) |
401 | 409 | self.assertEqual(M.js_output(), expected_js_output) |
402 | 410 | for i in ["foo bar", "foo@bar"]: |
403 | 411 | # Try some illegal characters |
|
0 commit comments