# Copyright 2018 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.

from telemetry.testing import options_for_unittests
from telemetry.testing import page_test_test_case
from telemetry.util import wpr_modes

from measurements import rendering
from measurements import rendering_util

RENDERING_THREAD_GROUPS = ['GPU', 'IO', 'browser', 'display_compositor',
                           'other', 'raster', 'renderer_compositor',
                           'renderer_main', 'total_all', 'total_fast_path']

class RenderingUnitTest(page_test_test_case.PageTestTestCase):
  """Test for rendering measurement

     Runs rendering measurement on a simple page and verifies the existence of
     all metrics in the results. The correctness of metrics is not tested here.
  """

  def setUp(self):
    self._options = options_for_unittests.GetCopy()
    self._options.browser_options.wpr_mode = wpr_modes.WPR_OFF
    self._options.pageset_repeat = 2

  def testRendering(self):
    ps = self.CreateStorySetFromFileInUnittestDataDir('scrollable_page.html')
    results = self.RunMeasurement(
        rendering.Rendering(), ps, options=self._options)
    self.assertFalse(results.had_failures)
    stat = rendering_util.ExtractStat(results)

    self.assertGreater(stat['frame_times'].count, 1)
    self.assertGreater(stat['percentage_smooth'].count, 1)
    for thread_group in RENDERING_THREAD_GROUPS:
      # We should have at least two sample values for each metric, since
      # pageset_repeat is 2.
      histogram_name = 'thread_%s_cpu_time_per_frame' % thread_group
      self.assertGreater(stat[histogram_name].count, 1)