forked from microsoft/playwright-python
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_proxy.py
122 lines (101 loc) · 4.08 KB
/
test_proxy.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
# Copyright (c) Microsoft Corporation.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import base64
import pytest
from playwright.async_api import Error
async def test_should_throw_for_bad_server_value(browser_factory):
with pytest.raises(Error) as exc_info:
await browser_factory(proxy={"server": 123})
assert "proxy.server: expected string, got number" in exc_info.value.message
async def test_should_use_proxy(browser_factory, server):
server.set_route(
"/target.html",
lambda r: (
r.write(b"<html><title>Served by the proxy</title></html>"),
r.finish(),
),
)
browser = await browser_factory(proxy={"server": f"localhost:{server.PORT}"})
page = await browser.new_page()
await page.goto("http://non-existent.com/target.html")
assert await page.title() == "Served by the proxy"
async def test_should_use_proxy_for_second_page(browser_factory, server):
server.set_route(
"/target.html",
lambda r: (
r.write(b"<html><title>Served by the proxy</title></html>"),
r.finish(),
),
)
browser = await browser_factory(proxy={"server": f"localhost:{server.PORT}"})
page1 = await browser.new_page()
await page1.goto("http://non-existent.com/target.html")
assert await page1.title() == "Served by the proxy"
page2 = await browser.new_page()
await page2.goto("http://non-existent.com/target.html")
assert await page2.title() == "Served by the proxy"
async def test_should_work_with_ip_port_notion(browser_factory, server):
server.set_route(
"/target.html",
lambda r: (
r.write(b"<html><title>Served by the proxy</title></html>"),
r.finish(),
),
)
browser = await browser_factory(proxy={"server": f"127.0.0.1:{server.PORT}"})
page = await browser.new_page()
await page.goto("http://non-existent.com/target.html")
assert await page.title() == "Served by the proxy"
async def test_should_authenticate(browser_factory, server):
def handler(req):
auth = req.getHeader("proxy-authorization")
if not auth:
req.setHeader(
b"Proxy-Authenticate", b'Basic realm="Access to internal site"'
)
req.setResponseCode(407)
else:
req.write(f"<html><title>{auth}</title></html>".encode("utf-8"))
req.finish()
server.set_route("/target.html", handler)
browser = await browser_factory(
proxy={
"server": f"localhost:{server.PORT}",
"username": "user",
"password": "secret",
}
)
page = await browser.new_page()
await page.goto("http://non-existent.com/target.html")
assert await page.title() == "Basic " + base64.b64encode(b"user:secret").decode(
"utf-8"
)
async def test_should_authenticate_with_empty_password(browser_factory, server):
def handler(req):
auth = req.getHeader("proxy-authorization")
if not auth:
req.setHeader(
b"Proxy-Authenticate", b'Basic realm="Access to internal site"'
)
req.setResponseCode(407)
else:
req.write(f"<html><title>{auth}</title></html>".encode("utf-8"))
req.finish()
server.set_route("/target.html", handler)
browser = await browser_factory(
proxy={"server": f"localhost:{server.PORT}", "username": "user", "password": ""}
)
page = await browser.new_page()
await page.goto("http://non-existent.com/target.html")
assert await page.title() == "Basic " + base64.b64encode(b"user:").decode("utf-8")