def testSet(self): cookie_jar = SimpleCookieJar() cookie_jar.set("a=b") self.assertFalse( cookie_jar.jar, "Cookie with no domain should not be added to the jar") cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; domain=.abc") self.assertTrue(".abc" in cookie_jar.jar) cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; domain=abc") self.assertTrue(".abc" in cookie_jar.jar) self.assertTrue("abc" not in cookie_jar.jar) cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; c=d; domain=abc") self.assertEquals(cookie_jar.get("abc"), "a=b; c=d") cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; c=d; domain=abc") cookie_jar.set("e=f; domain=abc") self.assertEquals(cookie_jar.get("abc"), "e=f") cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; c=d; domain=abc") cookie_jar.set("e=f; domain=.abc") self.assertEquals(cookie_jar.get("abc"), "e=f") cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; c=d; domain=abc") cookie_jar.set("e=f; domain=xyz") self.assertEquals(cookie_jar.get("abc"), "a=b; c=d") self.assertEquals(cookie_jar.get("xyz"), "e=f") self.assertEquals(cookie_jar.get("something"), "")
def testGet(self): cookie_jar = SimpleCookieJar() cookie_jar.set("a=b; c=d; domain=abc.com") self.assertEquals(cookie_jar.get("abc.com"), "a=b; c=d") self.assertEquals(cookie_jar.get("x.abc.com"), "a=b; c=d") self.assertEquals(cookie_jar.get("abc.com.es"), "") self.assertEquals(cookie_jar.get("xabc.com"), "") cookie_jar.set("a=b; c=d; domain=.abc.com") self.assertEquals(cookie_jar.get("abc.com"), "a=b; c=d") self.assertEquals(cookie_jar.get("x.abc.com"), "a=b; c=d") self.assertEquals(cookie_jar.get("abc.com.es"), "") self.assertEquals(cookie_jar.get("xabc.com"), "")
def compare_digest(s1, s2): return s1 == s2 # websocket supported version. VERSION = 13 SUPPORTED_REDIRECT_STATUSES = ( HTTPStatus.MOVED_PERMANENTLY, HTTPStatus.FOUND, HTTPStatus.SEE_OTHER, ) SUCCESS_STATUSES = SUPPORTED_REDIRECT_STATUSES + ( HTTPStatus.SWITCHING_PROTOCOLS, ) CookieJar = SimpleCookieJar() class handshake_response(object): def __init__(self, status, headers, subprotocol): self.status = status self.headers = headers self.subprotocol = subprotocol CookieJar.add(headers.get("set-cookie")) def handshake(sock, hostname, port, resource, **options): headers, key = _get_handshake_headers(resource, hostname, port, options) header_str = "\r\n".join(headers) send(sock, header_str)