Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- This WPT test may be referenced by the following Test IDs:
- /webdriver/tests/classic/take_screenshot/iframe.py - WPT Dashboard Interop Dashboard
import pytest
from tests.support.asserts import assert_success
from tests.support.image import png_dimensions
from tests.support.screenshot import (
DEFAULT_CONTENT,
INNER_IFRAME_STYLE,
OUTER_IFRAME_STYLE,
REFERENCE_CONTENT,
REFERENCE_STYLE,
)
from . import viewport_dimensions
def take_screenshot(session):
return session.transport.send(
"GET", "session/{session_id}/screenshot".format(**vars(session)))
def test_always_captures_top_browsing_context(session, inline, iframe):
iframe_content = "{0}{1}".format(INNER_IFRAME_STYLE, DEFAULT_CONTENT)
session.url = inline("""{0}{1}""".format(OUTER_IFRAME_STYLE, iframe(iframe_content)))
response = take_screenshot(session)
reference_screenshot = assert_success(response)
assert png_dimensions(reference_screenshot) == viewport_dimensions(session)
frame = session.find.css("iframe", all=False)
session.switch_frame(frame)
response = take_screenshot(session)
screenshot = assert_success(response)
assert png_dimensions(screenshot) == png_dimensions(reference_screenshot)
assert screenshot == reference_screenshot
@pytest.mark.parametrize("domain", ["", "alt"], ids=["same_origin", "cross_origin"])
def test_source_origin(session, inline, iframe, domain):
session.url = inline("{0}{1}".format(REFERENCE_STYLE, REFERENCE_CONTENT))
response = take_screenshot(session)
reference_screenshot = assert_success(response)
assert png_dimensions(reference_screenshot) == viewport_dimensions(session)
iframe_content = "{0}{1}".format(INNER_IFRAME_STYLE, DEFAULT_CONTENT)
session.url = inline("""{0}{1}""".format(
OUTER_IFRAME_STYLE, iframe(iframe_content, domain=domain)))
response = take_screenshot(session)
screenshot = assert_success(response)
assert png_dimensions(screenshot) == viewport_dimensions(session)
assert screenshot == reference_screenshot