+++ /dev/null
-# Simple test suite for Cookie.py\r
-\r
-from test.test_support import run_unittest, run_doctest, check_warnings\r
-import unittest\r
-import Cookie\r
-\r
-\r
-class CookieTests(unittest.TestCase):\r
- # Currently this only tests SimpleCookie\r
- def test_basic(self):\r
- cases = [\r
- { 'data': 'chips=ahoy; vienna=finger',\r
- 'dict': {'chips':'ahoy', 'vienna':'finger'},\r
- 'repr': "<SimpleCookie: chips='ahoy' vienna='finger'>",\r
- 'output': 'Set-Cookie: chips=ahoy\nSet-Cookie: vienna=finger',\r
- },\r
-\r
- { 'data': 'keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;"',\r
- 'dict': {'keebler' : 'E=mc2; L="Loves"; fudge=\012;'},\r
- 'repr': '''<SimpleCookie: keebler='E=mc2; L="Loves"; fudge=\\n;'>''',\r
- 'output': 'Set-Cookie: keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;"',\r
- },\r
-\r
- # Check illegal cookies that have an '=' char in an unquoted value\r
- { 'data': 'keebler=E=mc2',\r
- 'dict': {'keebler' : 'E=mc2'},\r
- 'repr': "<SimpleCookie: keebler='E=mc2'>",\r
- 'output': 'Set-Cookie: keebler=E=mc2',\r
- }\r
- ]\r
-\r
- for case in cases:\r
- C = Cookie.SimpleCookie()\r
- C.load(case['data'])\r
- self.assertEqual(repr(C), case['repr'])\r
- self.assertEqual(C.output(sep='\n'), case['output'])\r
- for k, v in sorted(case['dict'].iteritems()):\r
- self.assertEqual(C[k].value, v)\r
-\r
- def test_load(self):\r
- C = Cookie.SimpleCookie()\r
- C.load('Customer="WILE_E_COYOTE"; Version=1; Path=/acme')\r
-\r
- self.assertEqual(C['Customer'].value, 'WILE_E_COYOTE')\r
- self.assertEqual(C['Customer']['version'], '1')\r
- self.assertEqual(C['Customer']['path'], '/acme')\r
-\r
- self.assertEqual(C.output(['path']),\r
- 'Set-Cookie: Customer="WILE_E_COYOTE"; Path=/acme')\r
- self.assertEqual(C.js_output(), r"""\r
- <script type="text/javascript">\r
- <!-- begin hiding\r
- document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme; Version=1";\r
- // end hiding -->\r
- </script>\r
- """)\r
- self.assertEqual(C.js_output(['path']), r"""\r
- <script type="text/javascript">\r
- <!-- begin hiding\r
- document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme";\r
- // end hiding -->\r
- </script>\r
- """)\r
-\r
- # loading 'expires'\r
- C = Cookie.SimpleCookie()\r
- C.load('Customer="W"; expires=Wed, 01-Jan-2010 00:00:00 GMT')\r
- self.assertEqual(C['Customer']['expires'],\r
- 'Wed, 01-Jan-2010 00:00:00 GMT')\r
- C = Cookie.SimpleCookie()\r
- C.load('Customer="W"; expires=Wed, 01-Jan-98 00:00:00 GMT')\r
- self.assertEqual(C['Customer']['expires'],\r
- 'Wed, 01-Jan-98 00:00:00 GMT')\r
-\r
- def test_extended_encode(self):\r
- # Issue 9824: some browsers don't follow the standard; we now\r
- # encode , and ; to keep them from tripping up.\r
- C = Cookie.SimpleCookie()\r
- C['val'] = "some,funky;stuff"\r
- self.assertEqual(C.output(['val']),\r
- 'Set-Cookie: val="some\\054funky\\073stuff"')\r
-\r
- def test_quoted_meta(self):\r
- # Try cookie with quoted meta-data\r
- C = Cookie.SimpleCookie()\r
- C.load('Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"')\r
- self.assertEqual(C['Customer'].value, 'WILE_E_COYOTE')\r
- self.assertEqual(C['Customer']['version'], '1')\r
- self.assertEqual(C['Customer']['path'], '/acme')\r
-\r
-def test_main():\r
- run_unittest(CookieTests)\r
- with check_warnings(('.+Cookie class is insecure; do not use it',\r
- DeprecationWarning)):\r
- run_doctest(Cookie)\r
-\r
-if __name__ == '__main__':\r
- test_main()\r