switch to autox.py and robustify login/logout code

client/site_tests/desktopui_WindowManagerFocusNewWindows/desktopui_WindowManagerFocusNewWindows.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 os, time
 from autotest_lib.client.bin import site_ui_test, test
 from autotest_lib.client.common_lib import error
 
 class desktopui_WindowManagerFocusNewWindows(site_ui_test.UITest):
     version = 1
@@ skipping 26 matching lines @@
 
             self.autox.await_condition(
                 lambda: self.autox.get_top_window_id_at_point(200, 200) == id,
                 desc='Waiting for window 0x%x to be on top' % id)
 
         except self.autox.ConditionTimeoutError as exception:
             raise error.TestFail(
                 'Timed out on condition: %s' % exception.__str__())
 
     def run_once(self):
-        import autox
-
-        # TODO: Set these in a single, standard place for all tests.
-        os.environ['DISPLAY'] = ':0'
-        os.environ['XAUTHORITY'] = '/home/chronos/.Xauthority'
-        self.autox = autox.AutoX()
+        self.autox = self.get_autox()
 
         # Create a window and check that we switch to it.
         win = self.autox.create_and_map_window(
             width=200, height=200, title='test')
         info = self.autox.get_window_info(win.id)
         self.__check_active_window(win.id, info)
 
         # Create a second window.
         win2 = self.autox.create_and_map_window(
             width=200, height=200, title='test 2')
         info2 = self.autox.get_window_info(win2.id)
         self.__check_active_window(win2.id, info2)
 
         # Cycle backwards to the first window.
         self.autox.send_hotkey('Alt-Shift-Tab')
         self.__check_active_window(win.id, info)
 
         # Cycle forwards to the second window.
         self.autox.send_hotkey('Alt-Tab')
         self.__check_active_window(win2.id, info2)
 
         # Now destroy the second window and check that the WM goes back
         # to the first window.
         win2.destroy()
         self.__check_active_window(win.id, info)