diff options
Diffstat (limited to 'tests/test_cookies.py')
| -rw-r--r-- | tests/test_cookies.py | 85 |
1 files changed, 65 insertions, 20 deletions
diff --git a/tests/test_cookies.py b/tests/test_cookies.py index 9a66101..b985fce 100644 --- a/tests/test_cookies.py +++ b/tests/test_cookies.py @@ -1,20 +1,21 @@ -# -*- coding: UTF-8 -*- +# -*- coding: utf-8 -*- from datetime import timedelta from webob import cookies -from nose.tools import ok_, assert_raises, eq_ +from nose.tools import eq_ -def test_cookie(): - """ - Test cookie parsing, serialization and `repr` - """ +def test_cookie_empty(): c = cookies.Cookie() # empty cookie eq_(repr(c), '<Cookie: []>') - # a cookie with one value + +def test_cookie_one_value(): c = cookies.Cookie('dismiss-top=6') eq_(repr(c), "<Cookie: [<Morsel: dismiss-top='6'>]>") + +def test_cookie_one_value_with_trailing_semi(): c = cookies.Cookie('dismiss-top=6;') eq_(repr(c), "<Cookie: [<Morsel: dismiss-top='6'>]>") - # more complex cookie, (also mixing commas and semicolons) + +def test_cookie_complex(): c = cookies.Cookie('dismiss-top=6; CP=null*, '\ 'PHPSESSID=0a539d42abc001cdc762809248d4beed, a="42,"') c_dict = dict((k,v.value) for k,v in c.items()) @@ -23,10 +24,32 @@ def test_cookie(): 'PHPSESSID': '0a539d42abc001cdc762809248d4beed', 'dismiss-top': '6' }) + +def test_cookie_complex_serialize(): + c = cookies.Cookie('dismiss-top=6; CP=null*, '\ + 'PHPSESSID=0a539d42abc001cdc762809248d4beed, a="42,"') eq_(c.serialize(), 'CP=null*, PHPSESSID=0a539d42abc001cdc762809248d4beed, a="42,", ' 'dismiss-top=6') - # reserved keys ($xx) + +def test_cookie_load_multiple(): + c = cookies.Cookie('a=1; Secure=true') + eq_(repr(c), "<Cookie: [<Morsel: a='1'>]>") + eq_(c['a']['secure'], 'true') + +def test_cookie_secure(): + c = cookies.Cookie() + c['foo'] = 'bar' + c['foo'].secure = True + eq_(c.serialize(), 'foo=bar; secure') + +def test_cookie_httponly(): + c = cookies.Cookie() + c['foo'] = 'bar' + c['foo'].httponly = True + eq_(c.serialize(), 'foo=bar; HttpOnly') + +def test_cookie_reserved_keys(): c = cookies.Cookie('dismiss-top=6; CP=null*; $version=42; a=42') assert '$version' not in c c = cookies.Cookie('$reserved=42; a=$42') @@ -37,20 +60,15 @@ def test_serialize_cookie_date(): Testing webob.cookies.serialize_cookie_date. Missing scenarios: * input value is an str, should be returned verbatim - * input value is an int, should be converted to timedelta and we should - continue the rest of the process + * input value is an int, should be converted to timedelta and we + should continue the rest of the process """ - ok_(cookies.serialize_cookie_date('Tue, 04-Jan-2011 13:43:50 GMT')==\ - 'Tue, 04-Jan-2011 13:43:50 GMT', 'We passed a string, should get the ' - 'same one') - ok_(cookies.serialize_cookie_date(None) is None, 'We passed None, should ' - 'get None') - + eq_(cookies.serialize_cookie_date('Tue, 04-Jan-2011 13:43:50 GMT'), + 'Tue, 04-Jan-2011 13:43:50 GMT') + eq_(cookies.serialize_cookie_date(None), None) cdate_delta = cookies.serialize_cookie_date(timedelta(seconds=10)) cdate_int = cookies.serialize_cookie_date(10) - eq_(cdate_delta, cdate_int, - 'Passing a int to method should return the same result as passing a timedelta' - ) + eq_(cdate_delta, cdate_int) def test_ch_unquote(): eq_(cookies._unquote(u'"hello world'), u'"hello world') @@ -66,3 +84,30 @@ def test_ch_unquote(): ]: eq_(cookies._unquote(q), unq) eq_(cookies._quote(unq), q) + +def test_cookie_setitem_needs_quoting(): + c = cookies.Cookie() + c['La Pe\xc3\xb1a'] = '1' + eq_(len(c), 0) + +def test_morsel_serialize_with_expires(): + morsel = cookies.Morsel('bleh', 'blah') + morsel.expires = 'Tue, 04-Jan-2011 13:43:50 GMT' + result = morsel.serialize() + eq_(result, 'bleh=blah; expires="Tue, 04-Jan-2011 13:43:50 GMT"') + +def test_serialize_max_age_timedelta(): + import datetime + val = datetime.timedelta(86400) + result = cookies.serialize_max_age(val) + eq_(result, '7464960000') + +def test_serialize_max_age_int(): + val = 86400 + result = cookies.serialize_max_age(val) + eq_(result, '86400') + +def test_serialize_max_age_str(): + val = '86400' + result = cookies.serialize_max_age(val) + eq_(result, '86400') |