Index: tools/telemetry/telemetry/core/backends/form_based_credentials_backend_unittest_base.py |
diff --git a/tools/telemetry/telemetry/core/backends/form_based_credentials_backend_unittest_base.py b/tools/telemetry/telemetry/core/backends/form_based_credentials_backend_unittest_base.py |
index 3ce0c864aa0cb371ff22b8cdf9839dc948daeeae..866b240c8dfe9acfdf7ca90abeb24c6177f51fff 100644 |
--- a/tools/telemetry/telemetry/core/backends/form_based_credentials_backend_unittest_base.py |
+++ b/tools/telemetry/telemetry/core/backends/form_based_credentials_backend_unittest_base.py |
@@ -97,6 +97,7 @@ class FormBasedCredentialsBackendUnitTestBase(unittest.TestCase): |
def _LoginUsingMock(self, backend, login_page_url, email_element_id, |
password_element_id): # pylint: disable=R0201 |
tab = simple_mock.MockObject() |
+ ar = simple_mock.MockObject() |
config = {'username': 'blah', |
'password': 'blargh'} |
@@ -106,6 +107,8 @@ class FormBasedCredentialsBackendUnitTestBase(unittest.TestCase): |
tab.ExpectCall('EvaluateJavaScript', _).WillReturn(True) |
tab.ExpectCall('EvaluateJavaScript', _).WillReturn(False) |
tab.ExpectCall('WaitForDocumentReadyStateToBeInteractiveOrBetter') |
+ ar.ExpectCall('WaitForJavaScriptExpression') |
+ ar.ExpectCall('WaitForJavaScriptCondition') |
def VerifyEmail(js): |
assert email_element_id in js |
@@ -124,4 +127,4 @@ class FormBasedCredentialsBackendUnitTestBase(unittest.TestCase): |
# Checking for form still up. |
tab.ExpectCall('EvaluateJavaScript', _).WillReturn(False) |
- backend.LoginNeeded(tab, config) |
+ backend.LoginNeeded(tab, ar, config) |