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.
87 lines
3.2 KiB
87 lines
3.2 KiB
import unittest
|
|
import os.path
|
|
import paramiko
|
|
|
|
from tornado.httputil import HTTPServerRequest
|
|
from webssh.handler import MixinHandler, IndexHandler
|
|
|
|
|
|
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 read_privatekey(self, filename):
|
|
with open(filename, 'rb') as f:
|
|
return f.read().decode('utf-8')
|
|
|
|
def test_get_specific_pkey_with_plain_key(self):
|
|
|
|
fname = 'test_rsa.key'
|
|
cls = paramiko.RSAKey
|
|
key = self.read_privatekey(os.path.join('tests', fname))
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, None)
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_specific_pkey(cls, key, b'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 = b'television'
|
|
|
|
key = self.read_privatekey(os.path.join('tests', 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(ValueError):
|
|
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 = self.read_privatekey(os.path.join('tests', fname))
|
|
pkey = IndexHandler.get_pkey_obj(key, None)
|
|
self.assertIsInstance(pkey, cls)
|
|
pkey = IndexHandler.get_pkey_obj(key, u'iginored')
|
|
self.assertIsInstance(pkey, cls)
|
|
with self.assertRaises(ValueError):
|
|
pkey = IndexHandler.get_pkey_obj('x'+key, None)
|
|
|
|
def test_get_pkey_obj_with_encrypted_key(self):
|
|
fname = 'test_ed25519_password.key'
|
|
password = 'abc123'
|
|
cls = paramiko.Ed25519Key
|
|
key = self.read_privatekey(os.path.join('tests', fname))
|
|
pkey = IndexHandler.get_pkey_obj(key, password)
|
|
self.assertIsInstance(pkey, cls)
|
|
with self.assertRaises(ValueError):
|
|
pkey = IndexHandler.get_pkey_obj(key, u'wrongpass')
|
|
with self.assertRaises(ValueError):
|
|
pkey = IndexHandler.get_pkey_obj('x'+key, password)
|