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() |