Implemented telemetry benchmark that loads page and outputs screenshot.

tools/perf/contrib/cluster_telemetry/screenshot_unittest.py

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..9e68fe023853c1d38cc840d161db6e712c142364
--- /dev/null
+++ b/tools/perf/contrib/cluster_telemetry/screenshot_unittest.py
@@ -0,0 +1,35 @@
+# 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 platform
+import shutil
+import tempfile
+
+from telemetry.testing import options_for_unittests
+from telemetry.testing import page_test_test_case
+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)
+
+  def testPixelDiff(self):
+    page_set = self.CreateStorySetFromFileInUnittestDataDir('blank.html')
+    measurement = screenshot.Screenshot(self._png_outdir)
+    results = self.RunMeasurement(measurement, page_set, options=self._options)
+
+    # Screenshots not supported if failure occurs on Linux
+    if results.failures and platform.system() == 'Linux':
+      assert 'not supported' in results.failures[0].exc_info[1].message
+      return
+
+    saved_picture_count = results.FindAllPageSpecificValuesNamed(
+        'saved_picture_count')
+    self.assertEquals(len(saved_picture_count), 1)
+    self.assertGreater(saved_picture_count[0].GetRepresentativeNumber(), 0)