Skip to content

Commit 2db6afb

Browse files
author
Pan
committed
Added authentication exceptions. Updated tests.
1 parent d1eab7a commit 2db6afb

File tree

3 files changed

+11
-7
lines changed

3 files changed

+11
-7
lines changed

ssh/exceptions.pyx

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,3 +33,11 @@ class FatalError(BaseSSHError):
3333

3434
class OtherError(BaseSSHError):
3535
"""Raised on other non-specific fatal errors"""
36+
37+
38+
class AuthenticationDenied(BaseSSHError):
39+
"""Raised on authentication denied errors"""
40+
41+
42+
class AuthenticationError(BaseSSHError):
43+
"""Raised on fatal errors authenticating"""

tests/base_test.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,7 @@ def setUp(self):
5858
self.session.options_set(options.HOST, self.host)
5959
self.session.options_set_port(self.port)
6060
self.session.options_set(options.USER, self.user)
61-
# self.session.handshake(self.sock)
61+
# self.session.options_set(options.LOG_VERBOSITY, '1')
6262

6363
def tearDown(self):
6464
del self.session
65-
66-
# def _auth(self):
67-
# return self.session.userauth_publickey_fromfile(
68-
# self.user, self.user_pub_key, self.user_key,
69-
# '')

tests/test_session.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,5 @@ class SessionTest(SSHTestCase):
2828
def test_connect(self):
2929
self.assertEqual(self.session.connect(), 0)
3030
self.assertRaises(RequestDenied, self.session.userauth_none)
31-
31+
self.assertRaises(
32+
RequestDenied, self.session.userauth_publickey_auto, self.user, '')

0 commit comments

Comments
 (0)