| Index: client/site_tests/login_LoginSuccess/login_LoginSuccess.py
|
| diff --git a/client/site_tests/login_LoginSuccess/login_LoginSuccess.py b/client/site_tests/login_LoginSuccess/login_LoginSuccess.py
|
| index 9e3b9816276b38821894e645cead3c91bd98aa48..3b1027d6b568f698ddcb5a678ef38724a73c169e 100644
|
| --- a/client/site_tests/login_LoginSuccess/login_LoginSuccess.py
|
| +++ b/client/site_tests/login_LoginSuccess/login_LoginSuccess.py
|
| @@ -4,9 +4,9 @@
|
|
|
| import logging
|
| from autotest_lib.client.common_lib import error
|
| -from autotest_lib.client.cros import auth_server, ui_test
|
| +from autotest_lib.client.cros import auth_server, cros_ui_test
|
|
|
| -class login_LoginSuccess(ui_test.UITest):
|
| +class login_LoginSuccess(cros_ui_test.UITest):
|
| version = 1
|
|
|
| def __creds_checker(self, handler, url_args):
|
| @@ -28,6 +28,10 @@ class login_LoginSuccess(ui_test.UITest):
|
| self.use_local_dns()
|
|
|
|
|
| + def initialize(self, creds='$default'):
|
| + super(login_LoginSuccess, self).initialize(creds)
|
| +
|
| +
|
| def ensure_login_complete(self):
|
| """Wait for login to complete, including cookie fetching."""
|
| self._authServer.wait_for_client_login()
|
|
|