ssh: catch ConnectionError in communicate

It seems like asyncssh is not handling ConnectionError internally when
connecting to the target, so we catch it inside kirk. Also catch the
asyncio.Error generic exception for all the other cases.

Signed-off-by: Andrea Cervesato <andrea.cervesato@suse.com>
Closes: #75

Author: acerv

libkirk/ssh.py

@@ -19,7 +19,6 @@
 
 try:
     import asyncssh
-    import asyncssh.misc
 
     class MySSHClientSession(asyncssh.SSHClientSession):
         """
@@ -239,7 +238,7 @@ async def communicate(self, iobuffer: Optional[IOBuffer] = None) -> None:
 
             self._logger.info("Maximum SSH sessions: %d", max_sessions)
             self._session_sem = asyncio.Semaphore(max_sessions)
-        except asyncssh.misc.Error as err:
+        except (asyncssh.Error, ConnectionError) as err:
             if not self._stop:
                 raise SUTError(err) from err
 
@@ -332,7 +331,7 @@ async def run_command(
 
                 panic = session.kernel_panic()
                 stdout = session.get_output()
-            except asyncssh.misc.ChannelOpenError as err:
+            except asyncssh.Error as err:
                 if not self._stop:
                     raise SUTError(err)
             finally: