8e80217e59
Former-commit-id:a02aeb236c[formerly9f19e3f712] [formerly06a8b51d6d[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]] Former-commit-id:06a8b51d6dFormer-commit-id:3360eb6c5f
45 lines
1.2 KiB
Python
Executable file
45 lines
1.2 KiB
Python
Executable file
from werkzeug import Request, Response, parse_cookie
|
|
from werkzeug.contrib.securecookie import SecureCookie
|
|
|
|
|
|
def test_basic_support():
|
|
"""Basid SecureCookie support"""
|
|
c = SecureCookie(secret_key='foo')
|
|
assert c.new
|
|
print c.modified, c.should_save
|
|
assert not c.modified
|
|
assert not c.should_save
|
|
c['x'] = 42
|
|
assert c.modified
|
|
assert c.should_save
|
|
s = c.serialize()
|
|
|
|
c2 = SecureCookie.unserialize(s, 'foo')
|
|
assert c is not c2
|
|
assert not c2.new
|
|
assert not c2.modified
|
|
assert not c2.should_save
|
|
assert c2 == c
|
|
|
|
c3 = SecureCookie.unserialize(s, 'wrong foo')
|
|
assert not c3.modified
|
|
assert not c3.new
|
|
assert c3 == {}
|
|
|
|
|
|
def test_wrapper_support():
|
|
"""Securecookie wrapper integration"""
|
|
req = Request.from_values()
|
|
resp = Response()
|
|
c = SecureCookie.load_cookie(req, secret_key='foo')
|
|
assert c.new
|
|
c['foo'] = 42
|
|
assert c.secret_key == 'foo'
|
|
c.save_cookie(resp)
|
|
|
|
req = Request.from_values(headers={
|
|
'Cookie': 'session="%s"' % parse_cookie(resp.headers['set-cookie'])['session']
|
|
})
|
|
c2 = SecureCookie.load_cookie(req, secret_key='foo')
|
|
assert not c2.new
|
|
assert c2 == c
|