mirror of https://github.com/huashengdun/webssh
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
3.3 KiB
86 lines
3.3 KiB
import unittest
|
|
import paramiko
|
|
|
|
from tornado.httputil import HTTPServerRequest
|
|
from tests.utils import read_file, make_tests_data_path
|
|
from webssh.handler import MixinHandler, IndexHandler, InvalidValueError
|
|
|
|
|
|
class TestMixinHandler(unittest.TestCase):
|
|
|
|
def test_get_real_client_addr(self):
|
|
handler = MixinHandler()
|
|
handler.request = HTTPServerRequest(uri='/')
|
|
self.assertIsNone(handler.get_real_client_addr())
|
|
|
|
ip = '127.0.0.1'
|
|
handler.request.headers.add('X-Real-Ip', ip)
|
|
self.assertEqual(handler.get_real_client_addr(), False)
|
|
|
|
handler.request.headers.add('X-Real-Port', '12345x')
|
|
self.assertEqual(handler.get_real_client_addr(), False)
|
|
|
|
handler.request.headers.update({'X-Real-Port': '12345'})
|
|
self.assertEqual(handler.get_real_client_addr(), (ip, 12345))
|
|
|
|
handler.request.headers.update({'X-Real-ip': None})
|
|
self.assertEqual(handler.get_real_client_addr(), False)
|
|
|
|
handler.request.headers.update({'X-Real-Port': '12345x'})
|
|
self.assertEqual(handler.get_real_client_addr(), False)
|
|
|
|
|
|
class TestIndexHandler(unittest.TestCase):
|
|
|
|
def test_get_specific_pkey_with_plain_key(self):
|
|
|
|
fname = 'test_rsa.key'
|
|
cls = paramiko.RSAKey
|
|
key = read_file(make_tests_data_path(fname))
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, None)
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, 'iginored')
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_specific_pkey(cls, 'x'+key, None)
|
|
self.assertIsNone(pkey)
|
|
|
|
def test_get_specific_pkey_with_encrypted_key(self):
|
|
fname = 'test_rsa_password.key'
|
|
cls = paramiko.RSAKey
|
|
password = 'television'
|
|
|
|
key = read_file(make_tests_data_path(fname))
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, password)
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_specific_pkey(cls, 'x'+key, None)
|
|
self.assertIsNone(pkey)
|
|
|
|
with self.assertRaises(paramiko.PasswordRequiredException):
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, None)
|
|
|
|
def test_get_pkey_obj_with_plain_key(self):
|
|
fname = 'test_ed25519.key'
|
|
cls = paramiko.Ed25519Key
|
|
key = read_file(make_tests_data_path(fname))
|
|
pkey = IndexHandler.get_pkey_obj(key, None, fname)
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_pkey_obj(key, 'iginored', fname)
|
|
self.assertIsInstance(pkey, cls)
|
|
with self.assertRaises(InvalidValueError) as exc:
|
|
pkey = IndexHandler.get_pkey_obj('x'+key, None, fname)
|
|
self.assertIn('Invalid private key', str(exc))
|
|
|
|
def test_get_pkey_obj_with_encrypted_key(self):
|
|
fname = 'test_ed25519_password.key'
|
|
password = 'abc123'
|
|
cls = paramiko.Ed25519Key
|
|
key = read_file(make_tests_data_path(fname))
|
|
pkey = IndexHandler.get_pkey_obj(key, password, fname)
|
|
self.assertIsInstance(pkey, cls)
|
|
with self.assertRaises(InvalidValueError) as exc:
|
|
pkey = IndexHandler.get_pkey_obj(key, 'wrongpass', fname)
|
|
self.assertIn('Wrong password', str(exc))
|
|
with self.assertRaises(InvalidValueError) as exc:
|
|
pkey = IndexHandler.get_pkey_obj('x'+key, password, fname)
|
|
self.assertIn('Invalid private key', str(exc))
|