jocay/webssh
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Re-raise InvalidValueError for PasswordRequiredException

Browse Source
This commit is contained in:
Sheng
2019-05-19 20:27:44 +08:00
parent e1cd3efdf0
commit 786b42da7e
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_handler.py
+5 -3
View File
@@ -170,8 +170,9 @@ class TestIndexHandler(unittest.TestCase):
pkey = IndexHandler.get_specific_pkey(cls, 'x'+key, None)
self.assertIsNone(pkey)
with self.assertRaises(paramiko.PasswordRequiredException):
with self.assertRaises(InvalidValueError) as ctx:
pkey = IndexHandler.get_specific_pkey(cls, key, None)
self.assertIn('Need a password', str(ctx.exception))
def test_get_pkey_obj_with_plain_key(self):
fname = 'test_ed25519.key'
@@ -205,8 +206,9 @@ class TestIndexHandler(unittest.TestCase):
pkey = IndexHandler.get_pkey_obj('x'+key, '', fname)
self.assertIn('Invalid private key', str(ctx.exception))
with self.assertRaises(paramiko.PasswordRequiredException):
pkey = IndexHandler.get_pkey_obj(key, '', fname)
with self.assertRaises(InvalidValueError) as ctx:
pkey = IndexHandler.get_specific_pkey(cls, key, None)
self.assertIn('Need a password', str(ctx.exception))
class TestWsockHandler(unittest.TestCase):