Add PExpectError exception handling in _read_long_output() too
Change-Id: If6955b1b9b9433cd63138d9f52a8a7ee8e99aefd
diff --git a/drivers/CiscoSX300.py b/drivers/CiscoSX300.py
index dec39f6..0b0443e 100644
--- a/drivers/CiscoSX300.py
+++ b/drivers/CiscoSX300.py
@@ -368,6 +368,10 @@
prompt = self._prompt_name + '#'
try:
self.connection.expect(prompt)
+ except (pexect.EOF, pexect.TIMEOUT):
+ # Something went wrong; logout, log in and try again!
+ print "PEXPECT FAILURE, RECONNECT"
+ raise PExpectError("_cli failed on %s" % text)
except:
print "prompt is \"%s\"" % prompt
raise