Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(54)

Unified Diff: tools/perf/contrib/cluster_telemetry/screenshot_unittest.py

Issue 2923163007: Implemented telemetry benchmark that loads page and outputs screenshot. (Closed)
Patch Set: Addex pixel assert to unit test Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « tools/perf/contrib/cluster_telemetry/screenshot_ct.py ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tools/perf/contrib/cluster_telemetry/screenshot_unittest.py
diff --git a/tools/perf/contrib/cluster_telemetry/screenshot_unittest.py b/tools/perf/contrib/cluster_telemetry/screenshot_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..7eef0b3f0c69e179e00a3b337d49a93d24a59599
--- /dev/null
+++ b/tools/perf/contrib/cluster_telemetry/screenshot_unittest.py
@@ -0,0 +1,41 @@
+# Copyright 2017 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import os
+import shutil
+import tempfile
+
+from telemetry import decorators
+from telemetry.testing import options_for_unittests
+from telemetry.testing import page_test_test_case
+from telemetry.util import image_util
+from contrib.cluster_telemetry import screenshot
+
+class ScreenshotUnitTest(page_test_test_case.PageTestTestCase):
+
+ def setUp(self):
+ self._options = options_for_unittests.GetCopy()
+ self._png_outdir = tempfile.mkdtemp('_png_test')
+
+ def tearDown(self):
+ shutil.rmtree(self._png_outdir)
+
+ @decorators.Enabled('linux')
+ def testScreenshot(self):
+ # Screenshots for Cluster Telemetry purposes currently only supported on
+ # Linux platform.
+ page_set = self.CreateStorySetFromFileInUnittestDataDir(
+ 'screenshot_test.html')
+ measurement = screenshot.Screenshot(self._png_outdir)
+ self.RunMeasurement(measurement, page_set, options=self._options)
+
+ path = self._png_outdir + '/' + page_set.stories[0].file_safe_name + '.png'
+ self.assertTrue(os.path.exists(path))
+ self.assertTrue(os.path.isfile(path))
+ self.assertTrue(os.access(path, os.R_OK))
+
+ image = image_util.FromPngFile(path)
+ screenshot_pixels = image_util.Pixels(image)
+ special_colored_pixel = bytearray([217, 115, 43])
+ self.assertTrue(special_colored_pixel in screenshot_pixels)
« no previous file with comments | « tools/perf/contrib/cluster_telemetry/screenshot_ct.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698