Suspend- function for site_wifitest

server/site_wifitest.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 common, fnmatch, logging, os, re, string, threading, time
 
 from autotest_lib.server import autotest, hosts, subcommand
 from autotest_lib.server import site_bsd_router
 from autotest_lib.server import site_linux_router
 from autotest_lib.server import site_host_attributes
@@ skipping 733 matching lines @@
     def client_netdump_stop(self, params):
         if self.client_netdump_thread is not None:
             self.__destroy_netdump_dev()
             self.client.run("pkill %s" % self.client_cmd_netdump)
             self.client.get_file(self.client_netdump_file, '.')
             self.client.delete_tmp_dir(self.client_netdump_dir)
             self.client_netdump_thread.join()
             self.client_netdump_thread = None
 
 
+    def client_suspend(self, params):
+        """ Suspend the system """
+
+        script_client_file = self.install_script('site_system_suspend.py',
+                                                 '../client/common_lib/rtc.py',
+                                                 '../client/common_lib/'
+                                                 'sys_power.py')
+        result = self.client.run('python "%s" %d' %
+            (script_client_file, int(params.get("suspend_time", 5))))
+
+
+    def client_suspend_bg(self, params):
+        """ Suspend the system in the background """
+
+        script_client_file = self.install_script('site_system_suspend.py',
+                                                 '../client/common_lib/rtc.py',
+                                                 '../client/common_lib/'
+                                                 'sys_power.py')
+        cmd = ('python "%s" %d' %
+               (script_client_file, int(params.get("suspend_time", 5))))
+        self.client_suspend_thread = HelperThread(self.client, cmd)
+        self.client_suspend_thread.start()
+
+
+    def client_suspend_end(self, params):
+        """ Join the backgrounded suspend thread """
+
+        self.client_suspend_thread.join()
+        if self.client_suspend_thread.result.exit_status:
+            raise error.TestError('suspend failed')
+
 class HelperThread(threading.Thread):
     # Class that wraps a ping command in a thread so it can run in the bg.
     def __init__(self, client, cmd):
         threading.Thread.__init__(self)
         self.client = client
         self.cmd = cmd
 
     def run(self):
         # NB: set ignore_status as we're always terminated w/ pkill
-        self.client.run(self.cmd, ignore_status=True)
-
+        self.result = self.client.run(self.cmd, ignore_status=True)
 
 def __byfile(a, b):
     if a['file'] < b['file']:
         return -1
     elif a['file'] > b['file']:
         return 1
     else:
         return 0
 
 
@@ skipping 104 matching lines @@
     except error.TestFail:
       if 'expect_failure' in testcase:
         self.expect_failure(name, testcase['expect_failure'])
       else:
         raise
     except Exception, e:
       if 'expect_failure' in testcase:
         self.expect_failure(name, testcase['expect_failure'])
       else:
         raise error.TestFail(e)