mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-22 11:53:54 +08:00
342bb13bff
42428013 websocket patch from commaai/websocket-client.git 0fda5bb7 add jsonrpc 8139b06b add websocket_client git-subtree-dir: pyextra git-subtree-split: 4242801316e12c55e5b7c626331fbefad2e15e0c
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
import unittest
|
|
|
|
from websocket._cookiejar import SimpleCookieJar
|
|
|
|
try:
|
|
import Cookie
|
|
except:
|
|
import http.cookies as Cookie
|
|
|
|
|
|
class CookieJarTest(unittest.TestCase):
|
|
def testAdd(self):
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("")
|
|
self.assertFalse(cookie_jar.jar, "Cookie with no domain should not be added to the jar")
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b")
|
|
self.assertFalse(cookie_jar.jar, "Cookie with no domain should not be added to the jar")
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; domain=.abc")
|
|
self.assertTrue(".abc" in cookie_jar.jar)
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; domain=abc")
|
|
self.assertTrue(".abc" in cookie_jar.jar)
|
|
self.assertTrue("abc" not in cookie_jar.jar)
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; c=d; domain=abc")
|
|
self.assertEquals(cookie_jar.get("abc"), "a=b; c=d")
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; c=d; domain=abc")
|
|
cookie_jar.add("e=f; domain=abc")
|
|
self.assertEquals(cookie_jar.get("abc"), "a=b; c=d; e=f")
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; c=d; domain=abc")
|
|
cookie_jar.add("e=f; domain=.abc")
|
|
self.assertEquals(cookie_jar.get("abc"), "a=b; c=d; e=f")
|
|
|
|
cookie_jar = SimpleCookieJar()
|
|
cookie_jar.add("a=b; c=d; domain=abc")
|
|
cookie_jar.add("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 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"), "")
|