def setUp(self):
self.request = Request("http://www.example.com/page.html",
headers={"Content-Type": "text/html"})
self.wrapped = WrappedRequest(self.request)
def add_cookie_header(self, request):
super(CookieJar, self).add_cookie_header(WrappedRequest(request))
def set_cookie_if_ok(self, cookie, request):
super(CookieJar, self).set_cookie_if_ok(cookie,
WrappedRequest(request))
def extract_cookies(self, response, request):
wreq = WrappedRequest(request)
wrsp = WrappedResponse(response)
return super(CookieJar, self).extract_cookies(wrsp, wreq)
def make_cookies(self, response, request):
if not isinstance(request, WrappedRequest):
request = WrappedRequest(request)
if not isinstance(response, WrappedResponse):
response = WrappedResponse(response)
return super(CookieJar, self).make_cookies(response, request)
