(self)
| 1071 | self.assertEqual(len(c), 3) |
| 1072 | |
| 1073 | def test_domain_allow(self): |
| 1074 | c = CookieJar(policy=DefaultCookiePolicy( |
| 1075 | blocked_domains=["acme.com"], |
| 1076 | allowed_domains=["www.acme.com"])) |
| 1077 | |
| 1078 | req = urllib.request.Request("http://acme.com/") |
| 1079 | headers = ["Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/"] |
| 1080 | res = FakeResponse(headers, "http://acme.com/") |
| 1081 | c.extract_cookies(res, req) |
| 1082 | self.assertEqual(len(c), 0) |
| 1083 | |
| 1084 | req = urllib.request.Request("http://www.acme.com/") |
| 1085 | res = FakeResponse(headers, "http://www.acme.com/") |
| 1086 | c.extract_cookies(res, req) |
| 1087 | self.assertEqual(len(c), 1) |
| 1088 | |
| 1089 | req = urllib.request.Request("http://www.coyote.com/") |
| 1090 | res = FakeResponse(headers, "http://www.coyote.com/") |
| 1091 | c.extract_cookies(res, req) |
| 1092 | self.assertEqual(len(c), 1) |
| 1093 | |
| 1094 | # set a cookie with non-allowed domain... |
| 1095 | req = urllib.request.Request("http://www.coyote.com/") |
| 1096 | res = FakeResponse(headers, "http://www.coyote.com/") |
| 1097 | cookies = c.make_cookies(res, req) |
| 1098 | c.set_cookie(cookies[0]) |
| 1099 | self.assertEqual(len(c), 2) |
| 1100 | # ... and check is doesn't get returned |
| 1101 | c.add_cookie_header(req) |
| 1102 | self.assertFalse(req.has_header("Cookie")) |
| 1103 | |
| 1104 | def test_domain_block(self): |
| 1105 | pol = DefaultCookiePolicy( |
nothing calls this directly
no test coverage detected