| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import os | 5 import os |
| 6 import shutil | 6 import shutil |
| 7 import tempfile | 7 import tempfile |
| 8 import unittest | 8 import unittest |
| 9 | 9 |
| 10 from telemetry import page as page_module | 10 from telemetry import page as page_module |
| 11 from telemetry.page import page_set | 11 from telemetry.page import page_set |
| 12 from telemetry.timeline import trace_data | 12 from telemetry.timeline import trace_data |
| 13 from telemetry.unittest_util import system_stub | 13 from telemetry.unittest_util import system_stub |
| 14 from telemetry.value import trace | 14 from telemetry.value import trace |
| 15 | 15 |
| 16 | 16 |
| 17 class TestBase(unittest.TestCase): | 17 class TestBase(unittest.TestCase): |
| 18 | 18 |
| 19 def setUp(self): | 19 def setUp(self): |
| 20 ps = page_set.PageSet(file_path=os.path.dirname(__file__)) | 20 ps = page_set.PageSet(base_dir=os.path.dirname(__file__)) |
| 21 ps.AddUserStory(page_module.Page('http://www.bar.com/', ps, ps.base_dir)) | 21 ps.AddUserStory(page_module.Page('http://www.bar.com/', ps, ps.base_dir)) |
| 22 ps.AddUserStory(page_module.Page('http://www.baz.com/', ps, ps.base_dir)) | 22 ps.AddUserStory(page_module.Page('http://www.baz.com/', ps, ps.base_dir)) |
| 23 ps.AddUserStory(page_module.Page('http://www.foo.com/', ps, ps.base_dir)) | 23 ps.AddUserStory(page_module.Page('http://www.foo.com/', ps, ps.base_dir)) |
| 24 self.page_set = ps | 24 self.page_set = ps |
| 25 | 25 |
| 26 self._cloud_storage_stub = system_stub.Override(trace, ['cloud_storage']) | 26 self._cloud_storage_stub = system_stub.Override(trace, ['cloud_storage']) |
| 27 | 27 |
| 28 def tearDown(self): | 28 def tearDown(self): |
| 29 if self._cloud_storage_stub: | 29 if self._cloud_storage_stub: |
| 30 self._cloud_storage_stub.Restore() | 30 self._cloud_storage_stub.Restore() |
| (...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 112 | 112 |
| 113 def testNoLeakedTempFileWhenUploadingTrace(self): | 113 def testNoLeakedTempFileWhenUploadingTrace(self): |
| 114 v = trace.TraceValue(None, trace_data.TraceData({'test': 1})) | 114 v = trace.TraceValue(None, trace_data.TraceData({'test': 1})) |
| 115 v.CleanUp() | 115 v.CleanUp() |
| 116 self.assertTrue(_IsEmptyDir(self.temp_test_dir)) | 116 self.assertTrue(_IsEmptyDir(self.temp_test_dir)) |
| 117 | 117 |
| 118 def tearDown(self): | 118 def tearDown(self): |
| 119 super(NoLeakedTempfilesTests, self).tearDown() | 119 super(NoLeakedTempfilesTests, self).tearDown() |
| 120 shutil.rmtree(self.temp_test_dir) | 120 shutil.rmtree(self.temp_test_dir) |
| 121 trace.tempfile.tempdir = self.actual_tempdir | 121 trace.tempfile.tempdir = self.actual_tempdir |
| OLD | NEW |