diff options
Diffstat (limited to 'chrome/test/pyautolib/pyauto.py')
| -rwxr-xr-x | chrome/test/pyautolib/pyauto.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/chrome/test/pyautolib/pyauto.py b/chrome/test/pyautolib/pyauto.py index ca7f25b..66d5431 100755 --- a/chrome/test/pyautolib/pyauto.py +++ b/chrome/test/pyautolib/pyauto.py @@ -1167,12 +1167,15 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase): # Ensure that check for default search engine change has been performed. self._GetResultFromJSONRequest({'command': 'LoadSearchEngineInfo'}) - def GetProtectorState(self): + def GetProtectorState(self, window_index=0): """Returns current Protector state. This will trigger Protector's check for changed settings if it hasn't been performed yet. + Args: + window_index: The window index, default is 0. + Returns: A dictionary. Example: @@ -1181,7 +1184,7 @@ class PyUITest(pyautolib.PyUITestBase, unittest.TestCase): """ self._EnsureProtectorCheck() cmd_dict = {'command': 'GetProtectorState'} - return self._GetResultFromJSONRequest(cmd_dict) + return self._GetResultFromJSONRequest(cmd_dict, windex=window_index) def ApplyProtectorChange(self): """Applies the change shown by Protector and closes the bubble. |