mozilla-central/testing/web-platform/tests/webdriver/tests/bidi/browsing_context/capture_screenshot/capture_screenshot.py

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

import pytest
from math import ceil, floor
from tests.support.image import png_dimensions
from . import get_physical_viewport_dimensions
from ... import get_device_pixel_ratio, get_viewport_dimensions
@pytest.mark.asyncio
@pytest.mark.parametrize("activate", [True, False],
ids=["with activate", "without activate"])
async def test_capture(bidi_session, top_context, inline, compare_png_bidi,
activate):
expected_size = await get_physical_viewport_dimensions(bidi_session, top_context)
await bidi_session.browsing_context.navigate(
context=top_context["context"], url="about:blank", wait="complete"
)
if activate:
await bidi_session.browsing_context.activate(
context=top_context["context"])
reference_data = await bidi_session.browsing_context.capture_screenshot(
context=top_context["context"])
assert png_dimensions(reference_data) == expected_size
await bidi_session.browsing_context.navigate(
context=top_context["context"], url=inline("<div>foo</div>"), wait="complete"
)
if activate:
await bidi_session.browsing_context.activate(
context=top_context["context"])
data = await bidi_session.browsing_context.capture_screenshot(
context=top_context["context"])
comparison = await compare_png_bidi(data, reference_data)
assert not comparison.equal()
# Take a second screenshot that should be identical to validate that
# we don't just always return false here
await bidi_session.browsing_context.navigate(
context=top_context["context"], url=inline("<div>foo</div>"), wait="complete"
)
if activate:
await bidi_session.browsing_context.activate(
context=top_context["context"])
new_data = await bidi_session.browsing_context.capture_screenshot(
context=top_context["context"])
comparison = await compare_png_bidi(new_data, data)
assert comparison.equal()
@pytest.mark.parametrize("delta_width", [-10, +20], ids=["width smaller", "width larger"])
@pytest.mark.parametrize("delta_height", [-30, +40], ids=["height smaller", "height larger"])
@pytest.mark.asyncio
async def test_capture_with_viewport(bidi_session, new_tab, delta_width, delta_height):
original_viewport = await get_viewport_dimensions(bidi_session, new_tab)
dpr = await get_device_pixel_ratio(bidi_session, new_tab)
test_viewport = {
"width": original_viewport["width"] + delta_width,
"height": original_viewport["height"] + delta_height
}
await bidi_session.browsing_context.set_viewport(
context=new_tab["context"],
viewport=test_viewport)
expected_size = {
"width": floor(test_viewport["width"] * dpr),
"height": floor(test_viewport["height"] * dpr)
}
await bidi_session.browsing_context.navigate(
context=new_tab["context"], url="about:blank", wait="complete"
)
result = await bidi_session.browsing_context.capture_screenshot(
context=new_tab["context"])
assert png_dimensions(result) == (expected_size["width"], expected_size["height"])
@pytest.mark.parametrize("dpr", [0.5, 2])
@pytest.mark.asyncio
async def test_capture_with_different_dpr(bidi_session, new_tab, inline, dpr):
page = inline("<div style='background-color: black; width: 100px; height: 100px;'></div>")
await bidi_session.browsing_context.navigate(
context=new_tab["context"], url=page, wait="complete"
)
original_viewport = await get_viewport_dimensions(bidi_session, new_tab)
await bidi_session.browsing_context.set_viewport(
context=new_tab["context"],
device_pixel_ratio=dpr)
expected_width = original_viewport["width"] * dpr
expected_height = original_viewport["height"] * dpr
data = await bidi_session.browsing_context.capture_screenshot(context=new_tab["context"])
(actual_width, actual_height) = png_dimensions(data)
# The rounding is implementation-specific and can be either floor, ceil or round depending on the browser
# implementation. Tolerate any value between floor and ceil.
assert floor(expected_width) <= actual_width <= ceil(expected_width)
assert floor(expected_height) <= actual_height <= ceil(expected_height)