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
|
# 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 asyncio
from typing import cast
import pytest
from playwright.async_api import Error, Page, Route, expect
from tests.server import Server, TestServerRequest
async def test_should_support_timeout_option_in_route_fetch(
server: Server, page: Page
) -> None:
def _handler(request: TestServerRequest) -> None:
request.responseHeaders.addRawHeader("Content-Length", "4096")
request.responseHeaders.addRawHeader("Content-Type", "text/html")
request.write(b"")
server.set_route("/slow", _handler)
async def handle(route: Route) -> None:
with pytest.raises(Error) as error:
await route.fetch(timeout=1000)
assert "Route.fetch: Timeout 1000ms exceeded." in error.value.message
await page.route("**/*", lambda route: handle(route))
with pytest.raises(Error) as error:
await page.goto(server.PREFIX + "/slow", timeout=2000)
assert "Timeout 2000ms exceeded" in error.value.message
async def test_should_not_follow_redirects_when_max_redirects_is_set_to_0_in_route_fetch(
server: Server, page: Page
) -> None:
server.set_redirect("/foo", "/empty.html")
async def handle(route: Route) -> None:
response = await route.fetch(max_redirects=0)
assert response.headers["location"] == "/empty.html"
assert response.status == 302
await route.fulfill(body="hello")
await page.route("**/*", lambda route: handle(route))
await page.goto(server.PREFIX + "/foo")
assert "hello" in await page.content()
async def test_should_intercept_with_url_override(server: Server, page: Page) -> None:
async def handle(route: Route) -> None:
response = await route.fetch(url=server.PREFIX + "/one-style.html")
await route.fulfill(response=response)
await page.route("**/*.html", lambda route: handle(route))
response = await page.goto(server.PREFIX + "/empty.html")
assert response
assert response.status == 200
assert "one-style.css" in (await response.body()).decode("utf-8")
async def test_should_intercept_with_post_data_override(
server: Server, page: Page
) -> None:
request_promise = asyncio.create_task(server.wait_for_request("/empty.html"))
async def handle(route: Route) -> None:
response = await route.fetch(post_data={"foo": "bar"})
await route.fulfill(response=response)
await page.route("**/*.html", lambda route: handle(route))
await page.goto(server.PREFIX + "/empty.html")
request = await request_promise
assert request.post_body
assert request.post_body.decode("utf-8") == '{"foo": "bar"}'
async def test_should_fulfill_popup_main_request_using_alias(
page: Page, server: Server
) -> None:
async def route_handler(route: Route) -> None:
response = await route.fetch()
await route.fulfill(response=response, body="hello")
await page.context.route("**/*", route_handler)
await page.set_content(f'<a target=_blank href="{server.EMPTY_PAGE}">click me</a>')
[popup, _] = await asyncio.gather(
page.wait_for_event("popup"), page.get_by_text("click me").click()
)
await expect(cast(Page, popup).locator("body")).to_have_text("hello")
|
