switch to autox.py and robustify login/logout code

client/site_tests/login_ChromeProfileSanitary/login_ChromeProfileSanitary.py

 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import logging, os, stat, time, utils
 from autotest_lib.client.bin import chromeos_constants, site_cryptohome
-from autotest_lib.client.bin import site_login, test
+from autotest_lib.client.bin import site_login, site_ui_test
 from autotest_lib.client.common_lib import error, site_httpd, site_ui
 
 def respond_with_cookies(handler, url_args):
     """Responds with a Set-Cookie header to any GET request, and redirects
     to a chosen URL.
     """
     handler.send_response(303)
     handler.send_header('Set-Cookie', 'name=value')
     handler.send_header('Location', url_args['continue'][0])
     handler.end_headers()
     handler.wfile.write('Got form data:\n')
     handler.wfile.write('%s:\n' % url_args)
 
 
-class login_ChromeProfileSanitary(test.test):
+class login_ChromeProfileSanitary(site_ui_test.UITest):
     version = 1
 
-    def __wait_for_login_profile(self, timeout = 10):
+    def __wait_for_login_profile(self, timeout=10):
         start_time = time.time()
         while time.time() - start_time < timeout:
             if os.path.exists(chromeos_constants.LOGIN_PROFILE + '/Cookies'):
                 break;
             time.sleep(1)
         else:
             raise error.TestError('Login Profile took too long to populate')
 
 
-    def initialize(self):
+    def initialize(self, creds='$default'):
         spec = 'http://localhost:8000'
         path = '/set_cookie'
         self._wait_path = '/test_over'
         self._test_url = spec + path + '?continue=' + spec + self._wait_path
         self._testServer = site_httpd.HTTPListener(8000, docroot=self.srcdir)
         self._testServer.add_url_handler('/set_cookie', respond_with_cookies)
         self._testServer.run()
 
-
-    def setup(self):
-        site_login.setup_autox(self)
+        site_ui_test.UITest.initialize(self, creds)
 
 
     def cleanup(self):
         self._testServer.stop()
+        site_ui_test.UITest.cleanup(self)
 
 
-    def run_once(self, script = 'autox_script.json', timeout = 10):
-        logged_in = site_login.logged_in()
-
-        if not logged_in:
-            # Test account information embedded into json file.
-            site_login.attempt_login(self, script)
-
+    def run_once(self, timeout = 10):
         # Get Default/Cookies mtime.
         cookies_info = os.stat(chromeos_constants.LOGIN_PROFILE + '/Cookies')
         cookies_mtime = cookies_info[stat.ST_MTIME]
 
         # "crash" chrome.
         site_login.wait_for_initial_chrome_window()
         site_login.nuke_process_by_name(chromeos_constants.BROWSER,
                                         with_prejudice = True)
         site_login.wait_for_browser()
 
         # Navigate to site that leaves cookies.
         latch = self._testServer.add_wait_url(self._wait_path)
         cookie_fetch_args = ("--user-data-dir=" +
                              chromeos_constants.USER_DATA_DIR + ' ' +
                              self._test_url)
         session = site_ui.ChromeSession(args=cookie_fetch_args,
                                         clean_state=False)
         logging.debug('Chrome session started.')
         latch.wait(timeout)
 
         # Ensure chrome writes state to disk.
-        site_login.attempt_logout()
-        site_login.attempt_login(self, script)
+        self.logout()
+        self.login()
 
         if not latch.is_set():
             raise error.TestError('Never received callback from browser.')
 
         # Check mtime of Default/Cookies.  If changed, KABLOOEY.
         self.__wait_for_login_profile()
         cookies_info = os.stat(chromeos_constants.LOGIN_PROFILE + '/Cookies')
         # TODO: Note that this currently (20100329) fails and will continue to
         # do so until http://crosbug.com/1967 is fixed.
         if cookies_mtime != cookies_info[stat.ST_MTIME]:
             raise error.TestFail('Cookies in Default profile changed!')
-
-        # If we started logged out, log back out.
-        if not logged_in:
-            site_login.attempt_logout()