Skip to content

Commit 2b65b9b

Browse files
authored
catch socket.timeout for ut (#197)
1 parent 194f664 commit 2b65b9b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_ssh_context.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import os,sys,json,glob,shutil,uuid,getpass
22
import unittest
33
import pathlib
4+
import socket
45
from paramiko.ssh_exception import NoValidConnectionsError
56
from paramiko.ssh_exception import SSHException
67

@@ -17,7 +18,7 @@ def setUpClass(cls):
1718
mdata = json.load(f)
1819
try:
1920
cls.machine = Machine.load_from_dict(mdata['machine'])
20-
except SSHException:
21+
except (SSHException, socket.timeout):
2122
raise unittest.SkipTest("SSHException ssh cannot connect")
2223
cls.submission = SampleClass.get_sample_submission()
2324
cls.submission.bind_machine(cls.machine)

0 commit comments

Comments
 (0)