lava_wait_jobs: Handle xmlrpc.client.ProtocolError too
Should handle both IOError and ProtocolError in all places.
Signed-off-by: Paul Sokolovsky <paul.sokolovsky@linaro.org>
Change-Id: Id0a2b2d747eba4d4e6832cbd568e977d761e9878
diff --git a/lava_helper/lava_wait_jobs.py b/lava_helper/lava_wait_jobs.py
index ae5ea5d..335008c 100755
--- a/lava_helper/lava_wait_jobs.py
+++ b/lava_helper/lava_wait_jobs.py
@@ -21,6 +21,7 @@
import argparse
import shutil
import logging
+from xmlrpc.client import ProtocolError
from jinja2 import Environment, FileSystemLoader
from lava_helper import test_lava_dispatch_credentials
from lava_submit_jobs import submit_lava_jobs
@@ -109,7 +110,7 @@
time.sleep(0.2)
lava.get_job_results(job_id, results_file)
break
- except IOError as e:
+ except (ProtocolError, IOError) as e:
if retry == 1:
raise
else: