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 |