Chromium Code Reviews| Index: tools/telemetry/telemetry/extension_page.py |
| diff --git a/tools/telemetry/telemetry/extension_page.py b/tools/telemetry/telemetry/extension_page.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..292aaea9aadbc57536846f78f747a647dfe7089c |
| --- /dev/null |
| +++ b/tools/telemetry/telemetry/extension_page.py |
| @@ -0,0 +1,94 @@ |
| +# Copyright (c) 2012 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 json |
| + |
| +from telemetry import util |
| + |
| +class ExtensionPage(object): |
| + """Represents a an extension page in the browser |
| + |
| + Provides wrapper functions for running extension APIs as: |
| + E.g.: |
| + # Evaluates 1+1 in the extensions's javascript context. |
| + tab.runtime.Evaluate('1+1') |
| + """ |
| + def __init__(self, tab): |
|
nduca
2012/12/07 19:23:47
When we get close to landing this, this shouldn't
|
| + self.tab = tab |
| + |
| + def __enter__(self): |
| + return self |
| + |
| + def __exit__(self, *args): |
| + self.tab.Close() |
| + |
| + def _FormatParameters(self, *parameters): |
| + json_params = '' |
| + for param in parameters: |
|
nduca
2012/12/07 19:23:47
doesn't our code do this for us?
zel
2012/12/12 21:03:42
not sure. what do you think I should be calling in
|
| + if len(json_params) > 0: |
| + json_params += ',' |
| + json_params += json.dumps(param[0]) |
| + |
| + return json_params |
| + |
| + def _RegisterEventListeners(self, event_names): |
| + self.tab.runtime.Execute('window.__lastCallEventHandler = {};') |
| + for event_name in event_names: |
| + self.tab.runtime.Execute(""" |
| + window.__lastCallEventHandler['%s'] = function() { |
| + window.__lastCallEventArguments['%s'] = arguments; |
| + }; |
| + %s.addListener(window.__lastCallEventHandler['%s']); |
| + """ % (event_name, event_name, event_name, event_name)) |
| + |
| + def _UnregisterEventListeners(self, event_names): |
| + for event_name in event_names: |
| + self.tab.runtime.Execute(""" |
| + %s.removeListener(window.__lastCallEventHandler['%s']); |
| + """ % (event_name, event_name)) |
| + |
| + self.tab.runtime.Execute('window.__lastCallEventHandler = {};') |
| + |
| + def RunMethodWithEvents(self, event_names, method_name, *parameters): |
|
nduca
2012/12/07 19:23:47
This stuff makes me uneasy that the last argument
zel
2012/12/12 21:03:42
we need to chat in person about this one too
|
| + self._RegisterEventListeners(event_names) |
| + self.tab.runtime.Execute(""" |
| + window.__lastCallEventArguments = {}; |
| + window.__lastCallDone = false; |
| + window.__lastCallArguments = false; |
| + %s(%s, function() { |
| + window.__lastCallArguments = arguments; |
| + window.__lastCallDone = true; |
| + }); |
| + """ % (method_name, self._FormatParameters(parameters))) |
| + def IsDone(): |
| + expr = 'window.__lastCallDone' |
| + for event_name in event_names: |
| + expr += " && window.__lastCallEventArguments['%s']" % event_name |
| + |
| + return self.tab.runtime.Evaluate(expr) |
| + |
| + util.WaitFor(IsDone, timeout=30) |
| + results = (self.tab.runtime.Evaluate('window.__lastCallArguments'), |
| + self.tab.runtime.Evaluate('window.__lastCallEventArguments')) |
| + self._UnregisterEventListeners(event_names) |
| + return results |
| + |
| + def RunMethod(self, method_name, *parameters): |
| + self.tab.runtime.Execute(""" |
| + window.__lastCallDone = false; |
| + window.__lastCallArguments = false; |
| + %s(%s, function() { |
| + window.__lastCallArguments = arguments; |
| + window.__lastCallDone = true; |
| + }); |
| + """ % (method_name, self._FormatParameters(parameters))) |
| + def IsDone(): |
| + return self.tab.runtime.Evaluate("window.__lastCallDone") |
| + |
| + util.WaitFor(IsDone, timeout=30) |
| + return self.tab.runtime.Evaluate("window.__lastCallArguments") |
| + |
| + def RunSyncMethod(self, method_name, *parameters): |
|
nduca
2012/12/07 19:23:47
I think I'm fine with this on runtime object
|
| + return self.tab.runtime.Evaluate("%s(%s);" % |
| + (method_name, self._FormatParameters(parameters))) |
| + |