Update test_http_cookiejar.py

Author: LamentXU123

Lib/test/test_http_cookiejar.py

@@ -1262,6 +1262,40 @@ def test_missing_final_slash(self):
         c.add_cookie_header(req)
         self.assertTrue(req.has_header("Cookie"))
 
+    def test_set_ok_port(self):
+        pol = DefaultCookiePolicy(
+            rfc2965=True, blocked_domains=[])
+        c = CookieJar(policy=pol)
+        headers = ["Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/; port=1234"]
+        req = urllib.request.Request("http://127.0.0.1:1234")
+        res = FakeResponse(headers, "http://127.0.0.1:1234")
+        self.assertTrue(pol.set_ok_port(c.make_cookies(res, req)[0], req))
+        self.assertTrue(pol.set_ok_port(c.make_cookies(res, req)[0], req))
+
+        c.clear()
+
+        req = urllib.request.Request("http://[::1]:1234")
+        res = FakeResponse(headers, "http://[::1]:1234")
+        self.assertTrue(pol.set_ok_port(c.make_cookies(res, req)[0], req))
+        self.assertTrue(pol.set_ok_port(c.make_cookies(res, req)[0], req))
+
+    def test_return_ok_port(self):
+        pol = DefaultCookiePolicy(
+            rfc2965=True, blocked_domains=[])
+        c = CookieJar(policy=pol)
+        headers = ["Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/; port=1234"]
+        req = urllib.request.Request("http://127.0.0.1:1234")
+        res = FakeResponse(headers, "http://127.0.0.1:1234")
+        self.assertTrue(pol.return_ok_port(c.make_cookies(res, req)[0], req))
+        self.assertTrue(pol.return_ok_port(c.make_cookies(res, req)[0], req))
+
+        c.clear()
+
+        req = urllib.request.Request("http://[::1]:1234")
+        res = FakeResponse(headers, "http://[::1]:1234")
+        self.assertTrue(pol.return_ok_port(c.make_cookies(res, req)[0], req))
+        self.assertTrue(pol.return_ok_port(c.make_cookies(res, req)[0], req))
+
     def test_domain_mirror(self):
         pol = DefaultCookiePolicy(rfc2965=True)