/Lib/test/test_cookie.py
http://unladen-swallow.googlecode.com/ · Python · 82 lines · 66 code · 12 blank · 4 comment · 3 complexity · 9600d074e1e865bcfaf637e24a1faa5a MD5 · raw file
- # Simple test suite for Cookie.py
- from test.test_support import run_unittest, run_doctest
- import unittest
- import Cookie
- import warnings
- warnings.filterwarnings("ignore",
- ".* class is insecure.*",
- DeprecationWarning)
- class CookieTests(unittest.TestCase):
- # Currently this only tests SimpleCookie
- def test_basic(self):
- cases = [
- { 'data': 'chips=ahoy; vienna=finger',
- 'dict': {'chips':'ahoy', 'vienna':'finger'},
- 'repr': "<SimpleCookie: chips='ahoy' vienna='finger'>",
- 'output': 'Set-Cookie: chips=ahoy\nSet-Cookie: vienna=finger',
- },
- { 'data': 'keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;"',
- 'dict': {'keebler' : 'E=mc2; L="Loves"; fudge=\012;'},
- 'repr': '''<SimpleCookie: keebler='E=mc2; L="Loves"; fudge=\\n;'>''',
- 'output': 'Set-Cookie: keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;"',
- },
- # Check illegal cookies that have an '=' char in an unquoted value
- { 'data': 'keebler=E=mc2',
- 'dict': {'keebler' : 'E=mc2'},
- 'repr': "<SimpleCookie: keebler='E=mc2'>",
- 'output': 'Set-Cookie: keebler=E=mc2',
- }
- ]
- for case in cases:
- C = Cookie.SimpleCookie()
- C.load(case['data'])
- self.assertEqual(repr(C), case['repr'])
- self.assertEqual(C.output(sep='\n'), case['output'])
- for k, v in sorted(case['dict'].iteritems()):
- self.assertEqual(C[k].value, v)
- def test_load(self):
- C = Cookie.SimpleCookie()
- C.load('Customer="WILE_E_COYOTE"; Version=1; Path=/acme')
- self.assertEqual(C['Customer'].value, 'WILE_E_COYOTE')
- self.assertEqual(C['Customer']['version'], '1')
- self.assertEqual(C['Customer']['path'], '/acme')
- self.assertEqual(C.output(['path']),
- 'Set-Cookie: Customer="WILE_E_COYOTE"; Path=/acme')
- self.assertEqual(C.js_output(), """
- <script type="text/javascript">
- <!-- begin hiding
- document.cookie = "Customer="WILE_E_COYOTE"; Path=/acme; Version=1";
- // end hiding -->
- </script>
- """)
- self.assertEqual(C.js_output(['path']), """
- <script type="text/javascript">
- <!-- begin hiding
- document.cookie = "Customer="WILE_E_COYOTE"; Path=/acme";
- // end hiding -->
- </script>
- """)
- def test_quoted_meta(self):
- # Try cookie with quoted meta-data
- C = Cookie.SimpleCookie()
- C.load('Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"')
- self.assertEqual(C['Customer'].value, 'WILE_E_COYOTE')
- self.assertEqual(C['Customer']['version'], '1')
- self.assertEqual(C['Customer']['path'], '/acme')
- def test_main():
- run_unittest(CookieTests)
- run_doctest(Cookie)
- if __name__ == '__main__':
- test_main()