зеркало из https://github.com/mozilla/gecko-dev.git
Bug 1243415 - Improve screenshot unit tests and add tests for chrome scope. r=ato
MozReview-Commit-ID: 4p62N4HgnvO --HG-- extra : rebase_source : 04929b4c56f307cd8f6cec7a2190f2e15f9c2523
This commit is contained in:
Henrik Skupin
Родитель
f68eddf3dd
Коммит
07af072122
|
|
@ -1,19 +1,31 @@
|
||||||
<?xml version="1.0"?>
|
<?xml version="1.0"?>
|
||||||
|
|
||||||
<!-- This Source Code Form is subject to the terms of the Mozilla Public
|
<!-- This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
- License, v. 2.0. If a copy of the MPL was not distributed with this
|
- License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
- file, You can obtain one at http://mozilla.org/MPL/2.0/. -->
|
- file, You can obtain one at http://mozilla.org/MPL/2.0/. -->
|
||||||
|
|
||||||
<!DOCTYPE dialog [
|
<?xml-stylesheet href="chrome://global/skin/" type="text/css"?>
|
||||||
]>
|
|
||||||
|
|
||||||
<dialog id="dialogTest"
|
|
||||||
buttons="accept, cancel"
|
|
||||||
xmlns="http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul">
|
|
||||||
|
|
||||||
|
|
||||||
<vbox id="things">
|
<dialog id="testDialog"
|
||||||
<checkbox id="testBox" label="box" />
|
xmlns="http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul"
|
||||||
|
title="Test Dialog"
|
||||||
|
buttons="accept,cancel">
|
||||||
|
|
||||||
|
<vbox flex="1" style="min-width: 300px; min-height: 500px;">
|
||||||
|
<label>Settings</label>
|
||||||
|
<separator class="thin"/>
|
||||||
|
<richlistbox id="test-list" flex="1">
|
||||||
|
<richlistitem id="item-choose" orient="horizontal" selected="true">
|
||||||
|
<label id="choose-label" value="First Entry" flex="1"/>
|
||||||
|
<button id="choose-button" oncommand="" label="Choose..."/>
|
||||||
|
</richlistitem>
|
||||||
|
</richlistbox>
|
||||||
|
<separator class="thin"/>
|
||||||
|
<checkbox id="check-box" label="Test Mode 2" />
|
||||||
|
<hbox align="center">
|
||||||
|
<label id="text-box-label" control="text-box">Name:</label>
|
||||||
|
<textbox id="text-box" flex="1" />
|
||||||
|
</hbox>
|
||||||
</vbox>
|
</vbox>
|
||||||
|
|
||||||
</dialog>
|
</dialog>
|
||||||
|
|
|
||||||
|
|
@ -10,54 +10,104 @@ import urllib
|
||||||
|
|
||||||
from unittest import skip
|
from unittest import skip
|
||||||
|
|
||||||
from marionette_driver.by import By
|
from marionette_driver import By
|
||||||
|
from marionette_driver.errors import JavascriptException, NoSuchWindowException
|
||||||
from marionette_harness import MarionetteTestCase, WindowManagerMixin
|
from marionette_harness import MarionetteTestCase, skip_if_mobile, WindowManagerMixin
|
||||||
|
|
||||||
|
|
||||||
def inline(doc, mime="text/html;charset=utf-8"):
|
def inline(doc, mime="text/html;charset=utf-8"):
|
||||||
return "data:{0},{1}".format(mime, urllib.quote(doc))
|
return "data:{0},{1}".format(mime, urllib.quote(doc))
|
||||||
|
|
||||||
|
|
||||||
ELEMENT = "iVBORw0KGgoAAAANSUhEUgAAADIAAAAyCAYAAAAeP4ixAAAAVklEQVRoge3PMQ0AMAzAsPJHVWYbjEWTj/zx7O75oXk9AAISD6QWSC2QWiC1QGqB1AKpBVILpBZILZBaILVAaoHUAqkFUgukFkgtkFogtUBqgdT6BnIBMKa1DtYxhPkAAAAASUVORK5CYII="
|
box = inline("<body><div id='box'><p id='green' style='width: 50px; height: 50px; "
|
||||||
HIGHLIGHTED_ELEMENT = "iVBORw0KGgoAAAANSUhEUgAAADIAAAAyCAYAAAAeP4ixAAAAVklEQVRoge3PQRHAQAgAMfyrwhm1sb3JIwIyN3MvmJu53f01kRqRGpEakRqRGpEakRqRGpEakRqRGpEakRqRGpEakRqRGpEakRqRGpEakRqRmvciL/gAQgW/OxTpMPwAAAAASUVORK5CYII="
|
"background: silver;'></p></div></body>")
|
||||||
|
input = inline("<body><input id='text-input'></input></body>")
|
||||||
|
long = inline("<body style='height: 300vh'><p style='margin-top: 100vh'>foo</p></body>")
|
||||||
box = inline(
|
short = inline("<body style='height: 10vh'></body>")
|
||||||
"<div id=green style='width: 50px; height: 50px; background: silver;'></div>")
|
svg = inline("""
|
||||||
long = inline("<body style='height: 300vh'><p style='margin-top: 100vh'>foo")
|
<svg xmlns="http://www.w3.org/2000/svg" height="20" width="20">
|
||||||
short = inline("<body style='height: 10vh'>")
|
<rect height="20" width="20"/>
|
||||||
svg = inline("""<svg xmlns="http://www.w3.org/2000/svg" height="20" width="20">
|
</svg>""", mime="image/svg+xml")
|
||||||
<rect height="20" width="20"/>
|
|
||||||
</svg>""", mime="image/svg+xml")
|
|
||||||
|
|
||||||
|
|
||||||
class ScreenCaptureTestCase(MarionetteTestCase):
|
class ScreenCaptureTestCase(MarionetteTestCase):
|
||||||
def assert_png(self, string):
|
|
||||||
"""Test that string is a Base64 encoded PNG file."""
|
def setUp(self):
|
||||||
image = base64.decodestring(string)
|
super(ScreenCaptureTestCase, self).setUp()
|
||||||
|
|
||||||
|
self._device_pixel_ratio = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def device_pixel_ratio(self):
|
||||||
|
if self._device_pixel_ratio is None:
|
||||||
|
self._device_pixel_ratio = self.marionette.execute_script("""
|
||||||
|
return window.devicePixelRatio
|
||||||
|
""")
|
||||||
|
return self._device_pixel_ratio
|
||||||
|
|
||||||
|
@property
|
||||||
|
def document_element(self):
|
||||||
|
return self.marionette.find_element(By.CSS_SELECTOR, ":root")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def page_y_offset(self):
|
||||||
|
return self.marionette.execute_script("return window.pageYOffset")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def viewport_dimensions(self):
|
||||||
|
return self.marionette.execute_script("""
|
||||||
|
return [arguments[0].clientWidth,
|
||||||
|
arguments[0].clientHeight];
|
||||||
|
""", script_args=[self.document_element])
|
||||||
|
|
||||||
|
def assert_png(self, screenshot):
|
||||||
|
"""Test that screenshot is a Base64 encoded PNG file."""
|
||||||
|
image = base64.decodestring(screenshot)
|
||||||
self.assertEqual(imghdr.what("", image), "png")
|
self.assertEqual(imghdr.what("", image), "png")
|
||||||
|
|
||||||
def get_image_dimensions(self, string):
|
def assert_formats(self, element=None):
|
||||||
self.assert_png(string)
|
if element is None:
|
||||||
image = base64.decodestring(string)
|
element = self.document_element
|
||||||
|
|
||||||
|
screenshot_default = self.marionette.screenshot(element=element)
|
||||||
|
screenshot_image = self.marionette.screenshot(element=element, format="base64")
|
||||||
|
binary1 = self.marionette.screenshot(element=element, format="binary")
|
||||||
|
binary2 = self.marionette.screenshot(element=element, format="binary")
|
||||||
|
hash1 = self.marionette.screenshot(element=element, format="hash")
|
||||||
|
hash2 = self.marionette.screenshot(element=element, format="hash")
|
||||||
|
|
||||||
|
# Valid data should have been returned
|
||||||
|
self.assert_png(screenshot_image)
|
||||||
|
self.assertEqual(imghdr.what("", binary1), "png")
|
||||||
|
self.assertEqual(screenshot_image, base64.b64encode(binary1))
|
||||||
|
self.assertEqual(hash1, hashlib.sha256(screenshot_image).hexdigest())
|
||||||
|
|
||||||
|
# Different formats produce different data
|
||||||
|
self.assertNotEqual(screenshot_image, binary1)
|
||||||
|
self.assertNotEqual(screenshot_image, hash1)
|
||||||
|
self.assertNotEqual(binary1, hash1)
|
||||||
|
|
||||||
|
# A second capture should be identical
|
||||||
|
self.assertEqual(screenshot_image, screenshot_default)
|
||||||
|
self.assertEqual(binary1, binary2)
|
||||||
|
self.assertEqual(hash1, hash2)
|
||||||
|
|
||||||
|
def get_element_dimensions(self, element):
|
||||||
|
rect = element.rect
|
||||||
|
return rect["width"], rect["height"]
|
||||||
|
|
||||||
|
def get_image_dimensions(self, screenshot):
|
||||||
|
self.assert_png(screenshot)
|
||||||
|
image = base64.decodestring(screenshot)
|
||||||
width, height = struct.unpack(">LL", image[16:24])
|
width, height = struct.unpack(">LL", image[16:24])
|
||||||
return int(width), int(height)
|
return int(width), int(height)
|
||||||
|
|
||||||
|
def scale(self, rect):
|
||||||
|
return (int(rect[0] * self.device_pixel_ratio),
|
||||||
|
int(rect[1] * self.device_pixel_ratio))
|
||||||
|
|
||||||
|
|
||||||
class TestScreenCaptureChrome(WindowManagerMixin, ScreenCaptureTestCase):
|
class TestScreenCaptureChrome(WindowManagerMixin, ScreenCaptureTestCase):
|
||||||
@property
|
|
||||||
def primary_window_dimensions(self):
|
|
||||||
current_window = self.marionette.current_chrome_window_handle
|
|
||||||
self.marionette.switch_to_window(self.start_window)
|
|
||||||
with self.marionette.using_context("chrome"):
|
|
||||||
rv = tuple(self.marionette.execute_script("""
|
|
||||||
let el = document.documentElement;
|
|
||||||
let rect = el.getBoundingClientRect();
|
|
||||||
return [rect.width, rect.height];
|
|
||||||
"""))
|
|
||||||
self.marionette.switch_to_window(current_window)
|
|
||||||
return rv
|
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
super(TestScreenCaptureChrome, self).setUp()
|
super(TestScreenCaptureChrome, self).setUp()
|
||||||
|
|
@ -67,90 +117,245 @@ class TestScreenCaptureChrome(WindowManagerMixin, ScreenCaptureTestCase):
|
||||||
self.close_all_windows()
|
self.close_all_windows()
|
||||||
super(TestScreenCaptureChrome, self).tearDown()
|
super(TestScreenCaptureChrome, self).tearDown()
|
||||||
|
|
||||||
# A full chrome window screenshot is not the outer dimensions of
|
@property
|
||||||
# the window, but instead the bounding box of the <window> inside
|
def window_dimensions(self):
|
||||||
# <browser>.
|
return tuple(self.marionette.execute_script("""
|
||||||
def test_window(self):
|
let el = document.documentElement;
|
||||||
ss = self.marionette.screenshot()
|
let rect = el.getBoundingClientRect();
|
||||||
self.assert_png(ss)
|
return [rect.width, rect.height];
|
||||||
self.assertEqual(self.primary_window_dimensions,
|
"""))
|
||||||
self.get_image_dimensions(ss))
|
|
||||||
|
|
||||||
def test_chrome_delegation(self):
|
def open_dialog(self, url=None, width=None, height=None):
|
||||||
with self.marionette.using_context("content"):
|
if url is None:
|
||||||
content = self.marionette.screenshot()
|
url = "chrome://marionette/content/test_dialog.xul"
|
||||||
chrome = self.marionette.screenshot()
|
|
||||||
self.assertNotEqual(content, chrome)
|
def opener():
|
||||||
|
features = "chrome"
|
||||||
|
if height is not None:
|
||||||
|
features += ",height={}".format(height)
|
||||||
|
if width is not None:
|
||||||
|
features += ",width={}".format(width)
|
||||||
|
|
||||||
# This tests that GeckoDriver#takeScreenshot uses
|
|
||||||
# currentContext.document.documentElement instead of looking for a
|
|
||||||
# <window> element, which does not exist for all windows.
|
|
||||||
def test_secondary_windows(self):
|
|
||||||
def open_window_with_js():
|
|
||||||
self.marionette.execute_script("""
|
self.marionette.execute_script("""
|
||||||
window.open('chrome://marionette/content/test_dialog.xul', 'foo',
|
window.open(arguments[0], "", arguments[1]);
|
||||||
'dialog,height=200,width=300');
|
""", script_args=[url, features])
|
||||||
""")
|
|
||||||
|
|
||||||
new_window = self.open_window(open_window_with_js)
|
return self.open_window(opener)
|
||||||
self.marionette.switch_to_window(new_window)
|
|
||||||
|
|
||||||
ss = self.marionette.screenshot()
|
def test_capture_different_context(self):
|
||||||
size = self.get_image_dimensions(ss)
|
"""Check that screenshots in content and chrome are different."""
|
||||||
self.assert_png(ss)
|
with self.marionette.using_context("content"):
|
||||||
self.assertNotEqual(self.primary_window_dimensions, size)
|
screenshot_content = self.marionette.screenshot()
|
||||||
|
screenshot_chrome = self.marionette.screenshot()
|
||||||
|
self.assertNotEqual(screenshot_content, screenshot_chrome)
|
||||||
|
|
||||||
|
@skip_if_mobile
|
||||||
|
def test_capture_element(self):
|
||||||
|
dialog = self.open_dialog()
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
|
||||||
|
# Ensure we only capture the element
|
||||||
|
el = self.marionette.find_element(By.ID, "test-list")
|
||||||
|
screenshot_element = self.marionette.screenshot(element=el)
|
||||||
|
self.assertEqual(self.scale(self.get_element_dimensions(el)),
|
||||||
|
self.get_image_dimensions(screenshot_element))
|
||||||
|
|
||||||
|
# Ensure we do not capture the full window
|
||||||
|
screenshot_dialog = self.marionette.screenshot()
|
||||||
|
self.assertNotEqual(screenshot_dialog, screenshot_element)
|
||||||
|
|
||||||
self.marionette.close_chrome_window()
|
self.marionette.close_chrome_window()
|
||||||
self.marionette.switch_to_window(self.start_window)
|
self.marionette.switch_to_window(self.start_window)
|
||||||
|
|
||||||
|
def test_capture_flags(self):
|
||||||
|
dialog = self.open_dialog()
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
|
||||||
class Content(ScreenCaptureTestCase):
|
textbox = self.marionette.find_element(By.ID, "text-box")
|
||||||
@property
|
textbox.send_keys("")
|
||||||
def body_scroll_dimensions(self):
|
screenshot_focus = self.marionette.screenshot()
|
||||||
return tuple(self.marionette.execute_script(
|
|
||||||
"return [document.body.scrollWidth, document.body.scrollHeight]"))
|
|
||||||
|
|
||||||
@property
|
self.marionette.execute_script("arguments[0].blur();", script_args=[textbox])
|
||||||
def viewport_dimensions(self):
|
screenshot_no_focus = self.marionette.screenshot()
|
||||||
return tuple(self.marionette.execute_script("""
|
|
||||||
let docEl = document.documentElement;
|
|
||||||
return [docEl.clientWidth, docEl.clientHeight];"""))
|
|
||||||
|
|
||||||
@property
|
self.marionette.close_chrome_window()
|
||||||
def document_element(self):
|
self.marionette.switch_to_window(self.start_window)
|
||||||
return self.marionette.execute_script("return document.documentElement")
|
|
||||||
|
|
||||||
@property
|
self.assertNotEqual(screenshot_focus, screenshot_no_focus)
|
||||||
def page_y_offset(self):
|
|
||||||
return self.marionette.execute_script("return window.pageYOffset")
|
def test_capture_full_area(self):
|
||||||
|
# A full capture is not the outer dimensions of the window,
|
||||||
|
# but instead the bounding box of the window's root node (documentElement).
|
||||||
|
screenshot_full = self.marionette.screenshot()
|
||||||
|
screenshot_root = self.marionette.screenshot(element=self.document_element)
|
||||||
|
|
||||||
|
self.assert_png(screenshot_full)
|
||||||
|
self.assert_png(screenshot_root)
|
||||||
|
self.assertEqual(screenshot_root, screenshot_full)
|
||||||
|
self.assertEqual(self.scale(self.get_element_dimensions(self.document_element)),
|
||||||
|
self.get_image_dimensions(screenshot_full))
|
||||||
|
|
||||||
|
def test_capture_viewport(self):
|
||||||
|
# Load a HTML test page into the chrome window to get scrollbars
|
||||||
|
test_page = self.marionette.absolute_url("test.html")
|
||||||
|
dialog = self.open_dialog(url=test_page, width=50, height=50)
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
|
||||||
|
# Size of screenshot has to match viewport size
|
||||||
|
screenshot = self.marionette.screenshot(full=False)
|
||||||
|
self.assert_png(screenshot)
|
||||||
|
self.assertEqual(self.scale(self.viewport_dimensions),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
self.assertNotEqual(self.scale(self.window_dimensions),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
|
||||||
|
self.marionette.close_chrome_window()
|
||||||
|
self.marionette.switch_to_window(self.start_window)
|
||||||
|
|
||||||
|
@skip("https://bugzilla.mozilla.org/show_bug.cgi?id=1213875")
|
||||||
|
def test_capture_scroll_element_into_view(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_capture_window_already_closed(self):
|
||||||
|
dialog = self.open_dialog()
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
self.marionette.close_chrome_window()
|
||||||
|
|
||||||
|
self.assertRaises(NoSuchWindowException, self.marionette.screenshot)
|
||||||
|
self.marionette.switch_to_window(self.start_window)
|
||||||
|
|
||||||
|
def test_formats(self):
|
||||||
|
dialog = self.open_dialog()
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
|
||||||
|
self.assert_formats()
|
||||||
|
|
||||||
|
self.marionette.close_chrome_window()
|
||||||
|
self.marionette.switch_to_window(self.start_window)
|
||||||
|
|
||||||
|
def test_format_unknown(self):
|
||||||
|
with self.assertRaises(ValueError):
|
||||||
|
self.marionette.screenshot(format="cheese")
|
||||||
|
|
||||||
|
@skip_if_mobile
|
||||||
|
def test_highlight_elements(self):
|
||||||
|
dialog = self.open_dialog()
|
||||||
|
self.marionette.switch_to_window(dialog)
|
||||||
|
|
||||||
|
# Highlighting the element itself shouldn't make the image larger
|
||||||
|
element = self.marionette.find_element(By.ID, "test-list")
|
||||||
|
screenshot_element = self.marionette.screenshot(element=element)
|
||||||
|
screenshot_highlight = self.marionette.screenshot(element=element,
|
||||||
|
highlights=[element])
|
||||||
|
self.assertEqual(self.scale(self.get_element_dimensions(element)),
|
||||||
|
self.get_image_dimensions(screenshot_element))
|
||||||
|
self.assertNotEqual(screenshot_element, screenshot_highlight)
|
||||||
|
|
||||||
|
# Highlighting a sub element
|
||||||
|
button = self.marionette.find_element(By.ID, "choose-button")
|
||||||
|
screenshot_highlight_button = self.marionette.screenshot(element=element,
|
||||||
|
highlights=[button])
|
||||||
|
self.assertNotEqual(screenshot_element, screenshot_highlight_button)
|
||||||
|
self.assertNotEqual(screenshot_highlight, screenshot_highlight_button)
|
||||||
|
|
||||||
|
self.marionette.close_chrome_window()
|
||||||
|
self.marionette.switch_to_window(self.start_window)
|
||||||
|
|
||||||
|
def test_highlight_element_not_seen(self):
|
||||||
|
"""Check that for not found elements an exception is raised."""
|
||||||
|
with self.marionette.using_context('content'):
|
||||||
|
self.marionette.navigate(box)
|
||||||
|
content_element = self.marionette.find_element(By.ID, "green")
|
||||||
|
|
||||||
|
self.assertRaisesRegexp(JavascriptException, "Element reference not seen before",
|
||||||
|
self.marionette.screenshot, highlights=[content_element])
|
||||||
|
|
||||||
|
chrome_document_element = self.document_element
|
||||||
|
with self.marionette.using_context('content'):
|
||||||
|
self.assertRaisesRegexp(JavascriptException, "Element reference not seen before",
|
||||||
|
self.marionette.screenshot,
|
||||||
|
highlights=[chrome_document_element])
|
||||||
|
|
||||||
|
|
||||||
|
class TestScreenCaptureContent(WindowManagerMixin, ScreenCaptureTestCase):
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
ScreenCaptureTestCase.setUp(self)
|
super(TestScreenCaptureContent, self).setUp()
|
||||||
self.marionette.set_context("content")
|
self.marionette.set_context("content")
|
||||||
|
|
||||||
def test_html_document_element(self):
|
def tearDown(self):
|
||||||
|
self.close_all_tabs()
|
||||||
|
super(TestScreenCaptureContent, self).tearDown()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def scroll_dimensions(self):
|
||||||
|
return tuple(self.marionette.execute_script("""
|
||||||
|
return [document.body.scrollWidth, document.body.scrollHeight]
|
||||||
|
"""))
|
||||||
|
|
||||||
|
@skip_if_mobile # Needs application independent method to open a new tab
|
||||||
|
def test_capture_tab_already_closed(self):
|
||||||
|
tab = self.open_tab()
|
||||||
|
self.marionette.switch_to_window(tab)
|
||||||
|
self.marionette.close()
|
||||||
|
|
||||||
|
self.assertRaises(NoSuchWindowException, self.marionette.screenshot)
|
||||||
|
self.marionette.switch_to_window(self.start_tab)
|
||||||
|
|
||||||
|
def test_capture_element(self):
|
||||||
|
self.marionette.navigate(box)
|
||||||
|
el = self.marionette.find_element(By.TAG_NAME, "div")
|
||||||
|
screenshot = self.marionette.screenshot(element=el)
|
||||||
|
self.assert_png(screenshot)
|
||||||
|
self.assertEqual(self.scale(self.get_element_dimensions(el)),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
|
||||||
|
@skip("https://bugzilla.mozilla.org/show_bug.cgi?id=1213875")
|
||||||
|
def test_capture_element_scrolled_into_view(self):
|
||||||
self.marionette.navigate(long)
|
self.marionette.navigate(long)
|
||||||
string = self.marionette.screenshot()
|
el = self.marionette.find_element(By.TAG_NAME, "p")
|
||||||
self.assert_png(string)
|
screenshot = self.marionette.screenshot(element=el)
|
||||||
self.assertEqual(
|
self.assert_png(screenshot)
|
||||||
self.body_scroll_dimensions, self.get_image_dimensions(string))
|
self.assertEqual(self.scale(self.get_element_dimensions(el)),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
self.assertGreater(self.page_y_offset, 0)
|
||||||
|
|
||||||
def test_svg_document_element(self):
|
def test_capture_flags(self):
|
||||||
|
self.marionette.navigate(input)
|
||||||
|
|
||||||
|
textbox = self.marionette.find_element(By.ID, "text-input")
|
||||||
|
textbox.send_keys("")
|
||||||
|
screenshot_focus = self.marionette.screenshot()
|
||||||
|
|
||||||
|
self.marionette.execute_script("arguments[0].blur();", script_args=[textbox])
|
||||||
|
screenshot_no_focus = self.marionette.screenshot()
|
||||||
|
|
||||||
|
self.assertNotEqual(screenshot_focus, screenshot_no_focus)
|
||||||
|
|
||||||
|
def test_capture_html_document_element(self):
|
||||||
|
self.marionette.navigate(long)
|
||||||
|
screenshot = self.marionette.screenshot()
|
||||||
|
self.assert_png(screenshot)
|
||||||
|
self.assertEqual(self.scale(self.scroll_dimensions),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
|
||||||
|
def test_capture_svg_document_element(self):
|
||||||
self.marionette.navigate(svg)
|
self.marionette.navigate(svg)
|
||||||
doc_el = self.document_element
|
screenshot = self.marionette.screenshot()
|
||||||
string = self.marionette.screenshot()
|
self.assert_png(screenshot)
|
||||||
self.assert_png(string)
|
self.assertEqual(self.scale(self.get_element_dimensions(self.document_element)),
|
||||||
self.assertEqual((doc_el.rect["width"], doc_el.rect["height"]),
|
self.get_image_dimensions(screenshot))
|
||||||
self.get_image_dimensions(string))
|
|
||||||
|
|
||||||
def test_viewport(self):
|
def test_capture_viewport(self):
|
||||||
|
url = self.marionette.absolute_url("clicks.html")
|
||||||
self.marionette.navigate(short)
|
self.marionette.navigate(short)
|
||||||
string = self.marionette.screenshot(full=False)
|
self.marionette.navigate(url)
|
||||||
self.assert_png(string)
|
screenshot = self.marionette.screenshot(full=False)
|
||||||
self.assertEqual(
|
self.assert_png(screenshot)
|
||||||
self.viewport_dimensions, self.get_image_dimensions(string))
|
self.assertEqual(self.scale(self.viewport_dimensions),
|
||||||
|
self.get_image_dimensions(screenshot))
|
||||||
|
|
||||||
def test_viewport_after_scroll(self):
|
def test_capture_viewport_after_scroll(self):
|
||||||
self.marionette.navigate(long)
|
self.marionette.navigate(long)
|
||||||
before = self.marionette.screenshot()
|
before = self.marionette.screenshot()
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "p")
|
el = self.marionette.find_element(By.TAG_NAME, "p")
|
||||||
|
|
@ -160,48 +365,32 @@ class Content(ScreenCaptureTestCase):
|
||||||
self.assertNotEqual(before, after)
|
self.assertNotEqual(before, after)
|
||||||
self.assertGreater(self.page_y_offset, 0)
|
self.assertGreater(self.page_y_offset, 0)
|
||||||
|
|
||||||
def test_element(self):
|
def test_formats(self):
|
||||||
self.marionette.navigate(box)
|
self.marionette.navigate(box)
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "div")
|
|
||||||
string = self.marionette.screenshot(element=el)
|
|
||||||
self.assert_png(string)
|
|
||||||
self.assertEqual(
|
|
||||||
(el.rect["width"], el.rect["height"]), self.get_image_dimensions(string))
|
|
||||||
self.assertEqual(ELEMENT, string)
|
|
||||||
|
|
||||||
@skip("https://bugzilla.mozilla.org/show_bug.cgi?id=1213875")
|
# Use a smaller region to speed up the test
|
||||||
def test_element_scrolled_into_view(self):
|
element = self.marionette.find_element(By.TAG_NAME, "div")
|
||||||
self.marionette.navigate(long)
|
self.assert_formats(element=element)
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "p")
|
|
||||||
string = self.marionette.screenshot(element=el)
|
|
||||||
self.assert_png(string)
|
|
||||||
self.assertEqual(
|
|
||||||
(el.rect["width"], el.rect["height"]), self.get_image_dimensions(string))
|
|
||||||
self.assertGreater(self.page_y_offset, 0)
|
|
||||||
|
|
||||||
def test_element_with_highlight(self):
|
def test_format_unknown(self):
|
||||||
self.marionette.navigate(box)
|
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "div")
|
|
||||||
string = self.marionette.screenshot(element=el, highlights=[el])
|
|
||||||
self.assert_png(string)
|
|
||||||
self.assertEqual(
|
|
||||||
(el.rect["width"], el.rect["height"]), self.get_image_dimensions(string))
|
|
||||||
self.assertEqual(HIGHLIGHTED_ELEMENT, string)
|
|
||||||
|
|
||||||
def test_binary_element(self):
|
|
||||||
self.marionette.navigate(box)
|
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "div")
|
|
||||||
bin = self.marionette.screenshot(element=el, format="binary")
|
|
||||||
enc = base64.b64encode(bin)
|
|
||||||
self.assertEqual(ELEMENT, enc)
|
|
||||||
|
|
||||||
def test_unknown_format(self):
|
|
||||||
with self.assertRaises(ValueError):
|
with self.assertRaises(ValueError):
|
||||||
self.marionette.screenshot(format="cheese")
|
self.marionette.screenshot(format="cheese")
|
||||||
|
|
||||||
def test_hash_format(self):
|
def test_highlight_elements(self):
|
||||||
self.marionette.navigate(box)
|
self.marionette.navigate(box)
|
||||||
el = self.marionette.find_element(By.TAG_NAME, "div")
|
element = self.marionette.find_element(By.TAG_NAME, "div")
|
||||||
content = self.marionette.screenshot(element=el, format="hash")
|
|
||||||
hash = hashlib.sha256(ELEMENT).hexdigest()
|
# Highlighting the element itself shouldn't make the image larger
|
||||||
self.assertEqual(content, hash)
|
screenshot_element = self.marionette.screenshot(element=element)
|
||||||
|
screenshot_highlight = self.marionette.screenshot(element=element,
|
||||||
|
highlights=[element])
|
||||||
|
self.assertEqual(self.scale(self.get_element_dimensions(element)),
|
||||||
|
self.get_image_dimensions(screenshot_highlight))
|
||||||
|
self.assertNotEqual(screenshot_element, screenshot_highlight)
|
||||||
|
|
||||||
|
# Highlighting a sub element
|
||||||
|
paragraph = self.marionette.find_element(By.ID, "green")
|
||||||
|
screenshot_highlight_paragraph = self.marionette.screenshot(element=element,
|
||||||
|
highlights=[paragraph])
|
||||||
|
self.assertNotEqual(screenshot_element, screenshot_highlight_paragraph)
|
||||||
|
self.assertNotEqual(screenshot_highlight, screenshot_highlight_paragraph)
|
||||||
|
|
|
||||||
Загрузка…
Ссылка в новой задаче