Chromium Code Reviews| Index: tools/perf/page_sets/extension_profile_shared_state.py |
| diff --git a/tools/perf/page_sets/extension_profile_shared_state.py b/tools/perf/page_sets/extension_profile_shared_state.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..c15465c94c00231a99a553dd70b7ee9c5f8eff91 |
| --- /dev/null |
| +++ b/tools/perf/page_sets/extension_profile_shared_state.py |
| @@ -0,0 +1,29 @@ |
| +# Copyright 2015 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 shutil |
| + |
| +from profile_creators import profile_generator |
| +from profile_creators import extension_profile_extender |
| +from telemetry.page import shared_page_state |
| + |
| + |
| +class ExtensionProfileSharedState(shared_page_state.SharedPageState): |
| + def __init__(self, test, finder_options, story_set): |
|
robliao
2015/07/16 18:33:24
Add a docstring to this class. (See section under
sydli
2015/07/16 20:01:29
Done.
|
| + super(ExtensionProfileSharedState, self).__init__( |
| + test, finder_options, story_set) |
| + generator = profile_generator.ProfileGenerator( |
| + extension_profile_extender.ExtensionProfileExtender, |
| + "extension_profile") |
|
robliao
2015/07/16 18:33:24
Consistency: This delta appears to be preferring s
sydli
2015/07/16 20:01:29
Done.
|
| + self._out_dir = generator.Run(finder_options) |
| + if self._out_dir: |
| + finder_options.browser_options.profile_dir = self._out_dir |
| + else: |
| + finder_options.browser_options.dont_override_profile = True |
| + |
| + def TearDownState(self): |
| + """ Clean up generated profile directory. """ |
|
robliao
2015/07/16 18:33:23
Remove padding spaces in docstring.
sydli
2015/07/16 20:01:29
Done.
|
| + super(ExtensionProfileSharedState, self).TearDownState() |
| + if self._out_dir: |
| + shutil.rmtree(self._out_dir) |