15d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)# Copyright 2014 The Chromium Authors. All rights reserved.
25d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be
35d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)# found in the LICENSE file.
46e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)
55d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)from measurements import loading_trace
65d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)from telemetry.core import wpr_modes
75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)from telemetry.unittest import options_for_unittests
86e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)from telemetry.unittest import page_test_test_case
95d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
106e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)class LoadingTraceUnitTest(page_test_test_case.PageTestTestCase):
115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  def setUp(self):
125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    self._options = options_for_unittests.GetCopy()
135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    self._options.browser_options.wpr_mode = wpr_modes.WPR_OFF
145d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)
155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  def testLoadingTraceBasic(self):
165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    ps = self.CreatePageSetFromFileInUnittestDataDir('scrollable_page.html')
175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    measurement = loading_trace.LoadingTrace()
185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    trace_options = self._options
195d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    results = self.RunMeasurement(measurement, ps, options = trace_options)
205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    self.assertEquals(0, len(results.failures))
21effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch
22effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch  def testCleanUpTrace(self):
23effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch    self.TestTracingCleanedUp(loading_trace.LoadingTrace, self._options)
24