# found in the LICENSE file.
import os
-from telemetry.page import page_measurement
from telemetry.page import page_test
+from telemetry.page import page_test
+from telemetry.value import scalar
-class Screenshot(page_measurement.PageMeasurement):
+class Screenshot(page_test.PageTest):
def __init__(self):
super(Screenshot, self).__init__(
action_name_to_run = 'RunPrepareForScreenshot',
parser.error('Please specify --png-outdir')
cls._png_outdir = args.png_outdir
- def MeasurePage(self, page, tab, results):
+ def ValidateAndMeasurePage(self, page, tab, results):
if not tab.screenshot_supported:
raise page_test.TestNotSupportedOnPlatformFailure(
'Browser does not support screenshotting')
saved_picture_count = 0
if os.path.exists(outpath) and os.path.getmtime(outpath) > previous_mtime:
saved_picture_count = 1
- results.Add('saved_picture_count', 'count', saved_picture_count)
+ results.AddValue(scalar.ScalarValue(
+ results.current_page, 'saved_picture_count', 'count',
+ saved_picture_count))