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.2 KiB
86 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)
|
|
|