tools/telemetry/telemetry/value/trace_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 os
import unittest

from telemetry.page import page_set
from telemetry.timeline import tracing_timeline_data
from telemetry.value import trace

class TestBase(unittest.TestCase):
  def setUp(self):
    self.page_set = page_set.PageSet(file_path=os.path.dirname(__file__))
    self.page_set.AddPageWithDefaultRunNavigate("http://www.bar.com/")
    self.page_set.AddPageWithDefaultRunNavigate("http://www.baz.com/")
    self.page_set.AddPageWithDefaultRunNavigate("http://www.foo.com/")

  @property
  def pages(self):
    return self.page_set.pages

class ValueTest(TestBase):
  def testAsDict(self):
    v = trace.TraceValue(
        None, tracing_timeline_data.TracingTimelineData({'test' : 1}))
    fh_id = v.GetAssociatedFileHandle().id

    d = v.AsDict()

    self.assertEqual(d['file_id'], fh_id)