Add tests for special cookies

curl_cffi/requests/session.py

@@ -5,7 +5,6 @@
 import warnings
 from enum import Enum
 from http.cookiejar import Cookie
-from http.cookies import SimpleCookie
 from functools import partialmethod
 from io import BytesIO
 from json import dumps
@@ -415,7 +414,7 @@ def _set_curl_options(
         interface = interface or self.interface
         if interface:
             c.setopt(CurlOpt.INTERFACE, interface.encode())
-        
+
         return Request(url, h, method), buffer, header_buffer
 
     def _parse_response(self, curl, buffer, header_buffer):

tests/unittest/conftest.py

@@ -93,6 +93,8 @@ async def app(scope, receive, send):
         await set_headers(scope, receive, send)
     elif scope["path"].startswith("/set_cookies"):
         await set_cookies(scope, receive, send)
+    elif scope["path"].startswith("/set_special_cookies"):
+        await set_special_cookies(scope, receive, send)
     elif scope["path"].startswith("/redirect_301"):
         await redirect_301(scope, receive, send)
     elif scope["path"].startswith("/redirect_then_echo_cookies"):
@@ -301,6 +303,20 @@ async def set_cookies(scope, receive, send):
     await send({"type": "http.response.body", "body": b"Hello, world!"})
 
 
+async def set_special_cookies(scope, receive, send):
+    await send(
+        {
+            "type": "http.response.start",
+            "status": 200,
+            "headers": [
+                [b"content-type", b"text/plain"],
+                [b"set-cookie", b"foo=bar space"],
+            ],
+        }
+    )
+    await send({"type": "http.response.body", "body": b"Hello, world!"})
+
+
 async def redirect_301(scope, receive, send):
     await send(
         {"type": "http.response.start", "status": 301, "headers": [[b"location", b"/"]]}

tests/unittest/test_requests.py

@@ -292,6 +292,14 @@ def test_cookies_after_redirect(server):
     assert r.json()["foo"] == "bar"
 
 
+def test_cookies_with_special_chars(server):
+    s = requests.Session(debug=True)
+    r = s.get(str(server.url.copy_with(path="/set_special_cookies")))
+    assert s.cookies["foo"] == "bar space"
+    r = s.get(str(server.url.copy_with(path="/echo_cookies")))
+    assert r.json()["foo"] == "bar space"
+
+
 # https://github.com/yifeikong/curl_cffi/issues/39
 def test_post_body_cleaned(server):
     s = requests.Session()