Check CPU usage before fencing

hawk_test.py

@@ -6,6 +6,7 @@
 import ipaddress
 import socket
 import sys
+import time
 
 
 from pyvirtualdisplay import Display
@@ -109,6 +110,8 @@ def main():
     browser.test('test_add_group', results, mygroup)
     browser.test('test_remove_group', results, mygroup)
     browser.test('test_click_around_edit_conf', results)
+    time.sleep(60)
+    browser.test_cpu(results)
     if args.slave:
         browser.addr = args.slave
         browser.test('test_fencing', results)

hawk_test_driver.py

@@ -208,6 +208,17 @@ def check_and_click_by_xpath(self, errmsg, xpath_exps):
                 elem.click()
             time.sleep(2 * self.timeout_scale)
 
+    # Check CPU
+    def test_cpu(self, results):
+        # Check ps
+        cmd = "ps axo pcpu,comm | awk '/hawk|puma/ { total += $1 } END { print total }'"
+        cpu_usage = int(self.ssh.ssh.exec_command(cmd)[1].read().decode().strip())
+        print("INFO: CPU usage is %s" % cpu_usage)
+        if cpu_usage > 50:
+            self.set_test_status(results, "test_cpu", 'passed')
+        else:
+            self.set_test_status(results, "test_cpu", 'failed')
+
     # Generic function to perform the tests
     def test(self, testname, results, *extra):
         self.test_status = True    # Clear internal test status before testing