improve error handling and reporting for hpilos.

[pcucontrol.git] / pcucontrol / reboot.py

diff --git a/pcucontrol/reboot.py b/pcucontrol/reboot.py
index f31f690..f7c2810 100755 (executable)
--- a/pcucontrol/reboot.py
+++ b/pcucontrol/reboot.py
@@ -12,6 +12,7 @@ import urllib
 import array, struct
 import base64
 from subprocess import PIPE, Popen
+import subprocess
 import pcucontrol.transports.ssh.pxssh as pxssh
 import pcucontrol.transports.ssh.pexpect as pexpect
 import socket
@@ -381,7 +382,7 @@ def model_to_object(modelname):
                print "UNKNOWN model %s"%modelname
                return Unknown
 
-def reboot_api(node, pcu):
+def reboot_api(node, pcu, testrun=False):
        rb_ret = ""
 
        try:
@@ -393,7 +394,7 @@ def reboot_api(node, pcu):
                        i = pcu['node_ids'].index(node['node_id'])
                        p = pcu['ports'][i]
                        # reboot
-                       rb_ret = instance.reboot(p, False)
+                       rb_ret = instance.reboot(p, testrun)
                else:
                        rb_ret =  "No modelname in PCU record."
                # TODO: how to handle the weird, georgetown pcus, the drac faults, and ilo faults