src/tools/telemetry/telemetry/core/tab_unittest.py

   1 # Copyright (c) 2012 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
   7 from telemetry import test
   8 from telemetry.core import util
   9 from telemetry.core import exceptions
  10 from telemetry.unittest import tab_test_case
  11
  12
  13 def _IsDocumentVisible(tab):
  14   hidden = tab.EvaluateJavaScript('document.hidden || document.webkitHidden')
  15   # TODO(dtu): Remove when crbug.com/166243 is fixed.
  16   tab.Disconnect()
  17   return not hidden
  18
  19
  20 class TabTest(tab_test_case.TabTestCase):
  21   def testNavigateAndWaitToForCompleteState(self):
  22     self._browser.SetHTTPServerDirectories(util.GetUnittestDataDir())
  23     self._tab.Navigate(self._browser.http_server.UrlOf('blank.html'))
  24     self._tab.WaitForDocumentReadyStateToBeComplete()
  25
  26   def testNavigateAndWaitToForInteractiveState(self):
  27     self._browser.SetHTTPServerDirectories(util.GetUnittestDataDir())
  28     self._tab.Navigate(self._browser.http_server.UrlOf('blank.html'))
  29     self._tab.WaitForDocumentReadyStateToBeInteractiveOrBetter()
  30
  31   def testTabBrowserIsRightBrowser(self):
  32     self.assertEquals(self._tab.browser, self._browser)
  33
  34   def testRendererCrash(self):
  35     self.assertRaises(exceptions.TabCrashException,
  36                       lambda: self._tab.Navigate('chrome://crash',
  37                                                  timeout=5))
  38
  39   def testActivateTab(self):
  40     if not self._browser.supports_tab_control:
  41       logging.warning('Browser does not support tab control, skipping test.')
  42       return
  43
  44     self.assertTrue(_IsDocumentVisible(self._tab))
  45     new_tab = self._browser.tabs.New()
  46     new_tab.Navigate('about:blank')
  47     util.WaitFor(lambda: _IsDocumentVisible(new_tab), timeout=5)
  48     self.assertFalse(_IsDocumentVisible(self._tab))
  49     self._tab.Activate()
  50     util.WaitFor(lambda: _IsDocumentVisible(self._tab), timeout=5)
  51     self.assertFalse(_IsDocumentVisible(new_tab))
  52
  53
  54 class GpuTabTest(tab_test_case.TabTestCase):
  55   def setUp(self):
  56     self._extra_browser_args = ['--enable-gpu-benchmarking']
  57     super(GpuTabTest, self).setUp()
  58
  59   @test.Disabled('chromeos')
  60   def testScreenshot(self):
  61     if not self._tab.screenshot_supported:
  62       logging.warning('Browser does not support screenshots, skipping test.')
  63       return
  64
  65     self._browser.SetHTTPServerDirectories(util.GetUnittestDataDir())
  66     self._tab.Navigate(
  67       self._browser.http_server.UrlOf('green_rect.html'))
  68     self._tab.WaitForDocumentReadyStateToBeComplete()
  69     pixel_ratio = self._tab.EvaluateJavaScript('window.devicePixelRatio || 1')
  70
  71     screenshot = self._tab.Screenshot(5)
  72     assert screenshot
  73     screenshot.GetPixelColor(0 * pixel_ratio, 0 * pixel_ratio).AssertIsRGB(
  74         0, 255, 0, tolerance=2)
  75     screenshot.GetPixelColor(31 * pixel_ratio, 31 * pixel_ratio).AssertIsRGB(
  76         0, 255, 0, tolerance=2)
  77     screenshot.GetPixelColor(32 * pixel_ratio, 32 * pixel_ratio).AssertIsRGB(
  78         255, 255, 255, tolerance=2)