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..2aa1dad3c1e3d730859e461aa2b537f2f166a229 |
--- /dev/null |
+++ b/tools/perf/page_sets/extension_profile_shared_state.py |
@@ -0,0 +1,34 @@ |
+# 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 extension_profile_extender |
+from profile_creators import profile_generator |
+from telemetry.page import shared_page_state |
+ |
+ |
+class ExtensionProfileSharedState(shared_page_state.SharedPageState): |
+ """Shared state tied with extension profile. |
+ |
+ Generates extension profile on initialization. |
+ """ |
+ |
+ def __init__(self, test, finder_options, story_set): |
+ super(ExtensionProfileSharedState, self).__init__( |
+ test, finder_options, story_set) |
+ generator = profile_generator.ProfileGenerator( |
+ extension_profile_extender.ExtensionProfileExtender, |
+ 'extension_profile') |
+ 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.""" |
+ super(ExtensionProfileSharedState, self).TearDownState() |
+ if self._out_dir: |
+ shutil.rmtree(self._out_dir) |