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): |
+ 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) |