tools/telemetry/telemetry/results/block_page_measurement_results_unittest.py

# Copyright 2014 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 StringIO
import os
import unittest

from telemetry.page import page_set
from telemetry.results import block_page_measurement_results
from telemetry.value import scalar


BlockPageMeasurementResults = \
    block_page_measurement_results.BlockPageMeasurementResults

def _MakePageSet():
  ps = page_set.PageSet(file_path=os.path.dirname(__file__))
  ps.AddPageWithDefaultRunNavigate('http://www.foo.com/')
  ps.AddPageWithDefaultRunNavigate('http://www.bar.com/')
  return ps


class NonPrintingBlockPageMeasurementResults(BlockPageMeasurementResults):
  def __init__(self, *args):
    super(NonPrintingBlockPageMeasurementResults, self).__init__(*args)

  def _PrintPerfResult(self, *args):
    pass

class BlockPageMeasurementResultsTest(unittest.TestCase):
  def setUp(self):
    self._output = StringIO.StringIO()
    self._page_set = _MakePageSet()

  @property
  def lines(self):
    lines = StringIO.StringIO(self._output.getvalue()).readlines()
    return [line.strip() for line in lines]

  @property
  def data(self):
    return [line.split(': ', 1) for line in self.lines]

  def test_with_output_after_every_page(self):
    results = NonPrintingBlockPageMeasurementResults(self._output)
    results.WillMeasurePage(self._page_set[0])
    results.AddValue(
        scalar.ScalarValue(self._page_set[0], 'foo', 'seconds', 3))
    results.DidMeasurePage()

    results.WillMeasurePage(self._page_set[1])
    results.AddValue(
        scalar.ScalarValue(self._page_set[1], 'bar', 'seconds', 4))
    results.DidMeasurePage()

    expected = [
      ['name', 'http://www.foo.com/'],
      ['foo (seconds)', '3'],
      [''],
      ['name', 'http://www.bar.com/'],
      ['bar (seconds)', '4'],
      ['']
    ]
    self.assertEquals(self.data, expected)