Merge pull request #208 from DeliZhangX/private/deliz/CA-382842

CA-382842: Fix issue of "Connection refused"
xenserver · Sep 21, 2023 · aafc94c · aafc94c
2 parents 1bc5e48 + 021f6f4
commit aafc94c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autocertkit/common.py b/autocertkit/common.py
@@ -104,7 +104,10 @@ def run_cmd_ext(self, cmd):
 
         with open(f'/tmp/{fcmd}', 'w') as f:
             f.write(cmd)
-        self.put_file(f'/tmp/{fcmd}')
+        res = self.put_file(f'/tmp/{fcmd}')
+        if res['returncode']:
+            os.remove(f'/tmp/{fcmd}')
+            raise Exception(f'{SCP} failed, maybe {self.ip} is not yet ready.')
 
         self.run_cmd(fr'sh {fcmd} >{fout} 2>{ferr}; echo "$?" >{frc}')