Check CPU after each test

hawk_test_driver.py

@@ -208,6 +208,16 @@ def check_and_click_by_xpath(self, errmsg, xpath_exps):
                 elem.click()
             time.sleep(2 * self.timeout_scale)
 
+    # Check CPU
+    def check_cpu(self):
+        # Check ps
+        cmd = 'ps axo pcpu,comm | awk \'/bash/ {total += $1} END {print "cpu_usage["total"]"}\''
+        cpu_usage = int(self.ssh.ssh.exec_command(cmd)[1].read().decode().strip()[len("cpu_usage"):][1:-1])
+        print("INFO: CPU usage is %s" % cpu_usage)
+        if cpu_usage > 50:
+            return False
+        return True
+
     # Generic function to perform the tests
     def test(self, testname, results, *extra):
         self.test_status = True    # Clear internal test status before testing
@@ -219,6 +229,7 @@ def test(self, testname, results, *extra):
             else:
                 self.set_test_status(results, testname, 'failed')
                 self.driver.save_screenshot('%s.png' % testname)
+        self.check_cpu()
         self._close()
 
     # Set STONITH/sbd in maintenance. Assumes stonith-sbd resource is the last one listed on the