tools/perf/measurements/screenshot_unittest.py

   1 # Copyright 2014 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import logging
   6 import shutil
   7 import tempfile
   8
   9 from measurements import screenshot
  10 from telemetry import benchmark
  11 from telemetry.unittest_util import options_for_unittests
  12 from telemetry.unittest_util import page_test_test_case
  13
  14
  15 class ScreenshotUnitTest(page_test_test_case.PageTestTestCase):
  16   def setUp(self):
  17     self._options = options_for_unittests.GetCopy()
  18     self._png_outdir = tempfile.mkdtemp('_png_test')
  19
  20   def tearDown(self):
  21     shutil.rmtree(self._png_outdir)
  22
  23   @benchmark.Disabled('win')  # http://crbug.com/386572
  24   def testScreenshot(self):
  25     ps = self.CreatePageSetFromFileInUnittestDataDir('blank.html')
  26     measurement = screenshot.Screenshot(self._png_outdir)
  27     results = self.RunMeasurement(measurement, ps, options=self._options)
  28     if results.failures:
  29       logging.warning(str(results.failures))
  30       return
  31
  32     saved_picture_count = results.FindAllPageSpecificValuesNamed(
  33         'saved_picture_count')
  34     self.assertEquals(len(saved_picture_count), 1)
  35     self.assertGreater(saved_picture_count[0].GetRepresentativeNumber(), 0)