def test_set_cookie_encoded(self):
local.current_handler = handler = RequestHandler(self._get_app(), Request.from_values())
store = SessionStore(handler)
store.set_cookie('foo', 'bar', format='json')
store.set_cookie('baz', 'ding', format='json')
response = Response()
store.save(response)
headers = {'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))}
request = Request.from_values('/', headers=headers)
self.assertEqual(json_b64decode(request.cookies.get('foo')), 'bar')
self.assertEqual(json_b64decode(request.cookies.get('baz')), 'ding')
def test_get_save_session(self):
local.current_handler = handler = RequestHandler(self._get_app(), Request.from_values())
store = SessionStore(handler)
session = store.get_session()
self.assertEqual(isinstance(session, SecureCookieSession), True)
self.assertEqual(session, {})
session['foo'] = 'bar'
response = Response()
store.save(response)
request = Request.from_values('/', headers={'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))})
local.current_handler = handler = RequestHandler(self._get_app(), request)
store = SessionStore(handler)
session = store.get_session()
self.assertEqual(isinstance(session, SecureCookieSession), True)
self.assertEqual(session, {'foo': 'bar'})
def test_get_save_session(self):
local.current_handler = handler = RequestHandler(self._get_app(), Request.from_values())
store = SessionStore(handler)
session = store.get_session()
self.assertEqual(isinstance(session, SecureCookieSession), True)
self.assertEqual(session, {})
session['foo'] = 'bar'
response = Response()
store.save(response)
request = Request.from_values('/', headers={'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))})
local.current_handler = handler = RequestHandler(self._get_app(), request)
store = SessionStore(handler)
session = store.get_session()
self.assertEqual(isinstance(session, SecureCookieSession), True)
self.assertEqual(session, {'foo': 'bar'})
def test_set_delete_cookie(self):
local.current_handler = handler = RequestHandler(self._get_app(), Request.from_values())
store = SessionStore(handler)
store.set_cookie('foo', 'bar')
store.set_cookie('baz', 'ding')
response = Response()
store.save(response)
headers = {'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))}
request = Request.from_values('/', headers=headers)
self.assertEqual(request.cookies.get('foo'), 'bar')
self.assertEqual(request.cookies.get('baz'), 'ding')
store.delete_cookie('foo')
store.save(response)
headers = {'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))}
request = Request.from_values('/', headers=headers)
self.assertEqual(request.cookies.get('foo', None), '')
self.assertEqual(request.cookies['baz'], 'ding')
def test_set_delete_cookie(self):
local.current_handler = handler = RequestHandler(self._get_app(), Request.from_values())
store = SessionStore(handler)
store.set_cookie('foo', 'bar')
store.set_cookie('baz', 'ding')
response = Response()
store.save(response)
headers = {'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))}
request = Request.from_values('/', headers=headers)
self.assertEqual(request.cookies.get('foo'), 'bar')
self.assertEqual(request.cookies.get('baz'), 'ding')
store.delete_cookie('foo')
store.save(response)
headers = {'Cookie': '\n'.join(response.headers.getlist('Set-Cookie'))}
request = Request.from_values('/', headers=headers)
self.assertEqual(request.cookies.get('foo', None), '')
self.assertEqual(request.cookies['baz'], 'ding')
