Lib/test/test_cookie.py

   1 # Simple test suite for Cookie.py
   2
   3 from test.test_support import verify, verbose, run_doctest
   4 import Cookie
   5
   6 import warnings
   7 warnings.filterwarnings("ignore",
   8                         ".* class is insecure.*",
   9                         DeprecationWarning)
  10
  11 # Currently this only tests SimpleCookie
  12
  13 cases = [
  14     ('chips=ahoy; vienna=finger', {'chips':'ahoy', 'vienna':'finger'}),
  15     ('keebler="E=mc2; L=\\"Loves\\"; fudge=\\012;";',
  16      {'keebler' : 'E=mc2; L="Loves"; fudge=\012;'}),
  17
  18     # Check illegal cookies that have an '=' char in an unquoted value
  19     ('keebler=E=mc2;', {'keebler' : 'E=mc2'})
  20     ]
  21
  22 for data, dict in cases:
  23     C = Cookie.SimpleCookie() ; C.load(data)
  24     print repr(C)
  25     print str(C)
  26     items = dict.items()
  27     items.sort()
  28     for k, v in items:
  29         print ' ', k, repr( C[k].value ), repr(v)
  30         verify(C[k].value == v)
  31         print C[k]
  32
  33 C = Cookie.SimpleCookie()
  34 C.load('Customer="WILE_E_COYOTE"; Version=1; Path=/acme')
  35
  36 verify(C['Customer'].value == 'WILE_E_COYOTE')
  37 verify(C['Customer']['version'] == '1')
  38 verify(C['Customer']['path'] == '/acme')
  39
  40 print C.output(['path'])
  41 print C.js_output()
  42 print C.js_output(['path'])
  43
  44 # Try cookie with quoted meta-data
  45 C = Cookie.SimpleCookie()
  46 C.load('Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"')
  47 verify(C['Customer'].value == 'WILE_E_COYOTE')
  48 verify(C['Customer']['version'] == '1')
  49 verify(C['Customer']['path'] == '/acme')
  50
  51 print "If anything blows up after this line, it's from Cookie's doctest."
  52 run_doctest(Cookie)