# Simple test suite for Cookie.py | |
from test.test_support import run_unittest, run_doctest, check_warnings | |
import unittest | |
import Cookie | |
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(), r""" | |
<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']), r""" | |
<script type="text/javascript"> | |
<!-- begin hiding | |
document.cookie = "Customer=\"WILE_E_COYOTE\"; Path=/acme"; | |
// end hiding --> | |
</script> | |
""") | |
# loading 'expires' | |
C = Cookie.SimpleCookie() | |
C.load('Customer="W"; expires=Wed, 01-Jan-2010 00:00:00 GMT') | |
self.assertEqual(C['Customer']['expires'], | |
'Wed, 01-Jan-2010 00:00:00 GMT') | |
C = Cookie.SimpleCookie() | |
C.load('Customer="W"; expires=Wed, 01-Jan-98 00:00:00 GMT') | |
self.assertEqual(C['Customer']['expires'], | |
'Wed, 01-Jan-98 00:00:00 GMT') | |
def test_extended_encode(self): | |
# Issue 9824: some browsers don't follow the standard; we now | |
# encode , and ; to keep them from tripping up. | |
C = Cookie.SimpleCookie() | |
C['val'] = "some,funky;stuff" | |
self.assertEqual(C.output(['val']), | |
'Set-Cookie: val="some\\054funky\\073stuff"') | |
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) | |
with check_warnings(('.+Cookie class is insecure; do not use it', | |
DeprecationWarning)): | |
run_doctest(Cookie) | |
if __name__ == '__main__': | |
test_main() |