(self)
| 1102 | self.assertFalse(req.has_header("Cookie")) |
| 1103 | |
| 1104 | def test_domain_block(self): |
| 1105 | pol = DefaultCookiePolicy( |
| 1106 | rfc2965=True, blocked_domains=[".acme.com"]) |
| 1107 | c = CookieJar(policy=pol) |
| 1108 | headers = ["Set-Cookie: CUSTOMER=WILE_E_COYOTE; path=/"] |
| 1109 | |
| 1110 | req = urllib.request.Request("http://www.acme.com/") |
| 1111 | res = FakeResponse(headers, "http://www.acme.com/") |
| 1112 | c.extract_cookies(res, req) |
| 1113 | self.assertEqual(len(c), 0) |
| 1114 | |
| 1115 | p = pol.set_blocked_domains(["acme.com"]) |
| 1116 | c.extract_cookies(res, req) |
| 1117 | self.assertEqual(len(c), 1) |
| 1118 | |
| 1119 | c.clear() |
| 1120 | req = urllib.request.Request("http://www.roadrunner.net/") |
| 1121 | res = FakeResponse(headers, "http://www.roadrunner.net/") |
| 1122 | c.extract_cookies(res, req) |
| 1123 | self.assertEqual(len(c), 1) |
| 1124 | req = urllib.request.Request("http://www.roadrunner.net/") |
| 1125 | c.add_cookie_header(req) |
| 1126 | self.assertTrue(req.has_header("Cookie")) |
| 1127 | self.assertTrue(req.has_header("Cookie2")) |
| 1128 | |
| 1129 | c.clear() |
| 1130 | pol.set_blocked_domains([".acme.com"]) |
| 1131 | c.extract_cookies(res, req) |
| 1132 | self.assertEqual(len(c), 1) |
| 1133 | |
| 1134 | # set a cookie with blocked domain... |
| 1135 | req = urllib.request.Request("http://www.acme.com/") |
| 1136 | res = FakeResponse(headers, "http://www.acme.com/") |
| 1137 | cookies = c.make_cookies(res, req) |
| 1138 | c.set_cookie(cookies[0]) |
| 1139 | self.assertEqual(len(c), 2) |
| 1140 | # ... and check is doesn't get returned |
| 1141 | c.add_cookie_header(req) |
| 1142 | self.assertFalse(req.has_header("Cookie")) |
| 1143 | |
| 1144 | c.clear() |
| 1145 | |
| 1146 | pol.set_blocked_domains([]) |
| 1147 | req = urllib.request.Request("http://acme.com/") |
| 1148 | res = FakeResponse(headers, "http://acme.com/") |
| 1149 | cookies = c.make_cookies(res, req) |
| 1150 | c.extract_cookies(res, req) |
| 1151 | self.assertEqual(len(c), 1) |
| 1152 | |
| 1153 | req = urllib.request.Request("http://acme.com/") |
| 1154 | c.add_cookie_header(req) |
| 1155 | self.assertTrue(req.has_header("Cookie")) |
| 1156 | |
| 1157 | req = urllib.request.Request("http://badacme.com/") |
| 1158 | c.add_cookie_header(req) |
| 1159 | self.assertFalse(pol.return_ok(cookies[0], req)) |
| 1160 | self.assertFalse(req.has_header("Cookie")) |
| 1161 |
nothing calls this directly
no test coverage detected