class TestResponse(unittest.TestCase): def setUp(self): self.resp = Response() def tearDown(self): pass def test_set_cookie(self): self.resp.set_cookie('token', 'XXXXXXXX') self.assertEqual('token=XXXXXXXX', self.resp.environ['HTTP_SET_COOKIE']) self.assertEqual('token=XXXXXXXX', self.resp.headers['set-cookie']) self.resp.set_cookie('token', 'XXXXXXXX', path='/', comment='foo', domain='example.tld', max_age=10, secure=True, version='1', httponly=True) self.assertEqual('token=XXXXXXXX; Comment=foo; Domain=example.tld; httponly; Max-Age=10; Path=/; secure; Version=1', self.resp.environ['HTTP_SET_COOKIE'])
class TestResponse(unittest.TestCase): def setUp(self): self.resp = Response() def tearDown(self): pass def test_set_cookie(self): self.resp.set_cookie('token', 'XXXXXXXX') self.assertEqual('token=XXXXXXXX', self.resp.environ['HTTP_SET_COOKIE']) self.assertEqual('token=XXXXXXXX', self.resp.headers['set-cookie']) self.resp.set_cookie('token', 'XXXXXXXX', path='/', comment='foo', domain='example.tld', max_age=10, secure=True, version='1', httponly=True) self.assertEqual( 'token=XXXXXXXX; Comment=foo; Domain=example.tld; httponly; Max-Age=10; Path=/; secure; Version=1', self.resp.environ['HTTP_SET_COOKIE'])
return self.page_cont_list(req, storage_url, token) if path_type == 3: # container return self.page_obj_list(req, storage_url, token) if path_type == 4: # object try: (obj_status, objct) = get_object(storage_url, token, cont, obj) except ClientException, e: resp = Response(charset='utf8') resp.status = e.http_status return resp except err: pass resp = Response() resp.set_cookie('_token', token, path=self.page_path, max_age=self.cookie_max_age, secure=self.secure) resp.status = HTTP_OK resp.headers = obj_status resp.body = objct self.token_bank[token].update({'msg': ''}) self.memcache_update(token) return resp return HTTPFound(location=self.add_prefix(storage_url)) def page_cont_list(self, req, storage_url, token, template=None): """ """ if template is None: tmpl = self.tmpl path = urlparse(self.del_prefix(req.url)).path
if path_type == 2: # account return self.page_cont_list(req, storage_url, token) if path_type == 3: # container return self.page_obj_list(req, storage_url, token) if path_type == 4: # object try: (obj_status, objct) = get_object(storage_url, token, cont, obj) except ClientException, e: resp = Response(charset='utf8') resp.status = e.http_status return resp except err: pass resp = Response() resp.set_cookie('_token', token, path=self.page_path, max_age=self.cookie_max_age, secure=self.secure) resp.status = HTTP_OK resp.headers = obj_status resp.body = objct self.token_bank[token].update({'msg': ''}) self.memcache_update(token) return resp return HTTPFound(location=self.add_prefix(storage_url)) def page_cont_list(self, req, storage_url, token, template=None): """ """ if template is None: tmpl = self.tmpl path = urlparse(self.del_prefix(req.url)).path if len(path.split('/')) <= 2: