[chromeos] pyauto tests should auto login by default on chromeos

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/env python
 # 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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
 run pydoc on this file.
@@ skipping 191 matching lines @@
     # Sometimes flash files loaded too quickly after firing browser
     # ends up getting downloaded, which seems to indicate that the plugin
     # hasn't been registered yet.
     if not self.IsChromeOS():
       self.GetPluginsInfo()
 
     # TODO(dtu): Remove this after crosbug.com/4558 is fixed.
     if self.IsChromeOS():
       self.WaitUntil(lambda: not self.GetNetworkInfo()['offline_mode'])
 
-    if (self.IsChromeOS() and self.ShouldOOBESkipToLogin() and
-        not self.GetLoginInfo()['is_logged_in'] and
-        self.GetOOBEScreenInfo()['screen_name'] != 'login'):
-      self.SkipToLogin()
+    if (self.IsChromeOS() and not self.GetLoginInfo()['is_logged_in'] and
+        self.ShouldOOBESkipToLogin()):
+      if self.GetOOBEScreenInfo()['screen_name'] != 'login':
+        self.SkipToLogin()
+      if self.ShouldAutoLogin():
+        # Login with default creds.
+        sys.path.append('/usr/local')  # to import autotest libs
+        from autotest.cros import constants
+        creds = constants.CREDENTIALS['$default']
+        self.Login(creds[0], creds[1])
+        assert self.GetLoginInfo()['is_logged_in']
+        logging.info('Logged in as %s.' % creds[0])
 
     # If we are connected to any RemoteHosts, create PyAuto
     # instances on the remote sides and set them up too.
     for remote in self.remotes:
       remote.CreateTarget(self)
       remote.setUp()
 
   def setUp(self):
     """Override this method to launch browser differently.
 
@@ skipping 69 matching lines @@
     example. Calling this function directly will have no effect.
 
     Returns:
       True, if the OOBE should be skipped and automation should
             go to the 'Add user' login screen directly
       False, if the OOBE should not be skipped.
     """
     assert self.IsChromeOS()
     return True
 
+  def ShouldAutoLogin(self):
+    """Determine if we should auto-login on ChromeOS at browser startup.
+
+    To be used for tests that expect user to be logged in before running test,
+    without caring which user. ShouldOOBESkipToLogin() should return True
+    for this to take effect.
+
+    Override and return False to not auto login, for tests where login is part
+    of the use case.
+
+    Returns:
+      True, if chrome should auto login after startup.
+      False, otherwise.
+    """
+    assert self.IsChromeOS()
+    return True
+
   def CloseChromeOnChromeOS(self):
     """Gracefully exit chrome on ChromeOS."""
 
     def _GetListOfChromePids():
       """Retrieves the list of currently-running Chrome process IDs.
 
       Returns:
         A list of strings, where each string represents a currently-running
         'chrome' process ID.
       """
@@ skipping 6345 matching lines @@
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()