+ def ensure_cookies_to_set(self):
+ if not hasattr(self, 'cookies_to_set'):
+ self.cookies_to_set = []
+
+ def set_cookie(self, cookie_name, cookie_path, cookie_db):
+ self.ensure_cookies_to_set()
+ cookie = SimpleCookie()
+ cookie[cookie_name] = json.dumps(cookie_db)
+ cookie[cookie_name]['path'] = cookie_path
+ self.cookies_to_set += [cookie]
+
+ def unset_cookie(self, cookie_name, cookie_path):
+ self.ensure_cookies_to_set()
+ cookie = SimpleCookie()
+ cookie[cookie_name] = ''
+ cookie[cookie_name]['path'] = cookie_path
+ cookie[cookie_name]['expires'] = 'Thu, 01 Jan 1970 00:00:00 GMT'
+ self.cookies_to_set += [cookie]
+
+ def get_cookie_db(self, cookie_name):
+ cookie_db = {}
+ if 'Cookie' in self.headers:
+ cookie = SimpleCookie(self.headers['Cookie'])
+ if cookie_name in cookie:
+ cookie_db = json.loads(cookie[cookie_name].value)
+ return cookie_db
+