mirror of https://github.com/huashengdun/webssh
Added print_version
parent
934d6dfcb6
commit
72df3b9ee9
|
@ -4,13 +4,19 @@ import paramiko
|
||||||
import tornado.options as options
|
import tornado.options as options
|
||||||
|
|
||||||
from webssh.settings import (
|
from webssh.settings import (
|
||||||
get_host_keys_settings, get_policy_setting, base_dir
|
get_host_keys_settings, get_policy_setting, base_dir, print_version
|
||||||
)
|
)
|
||||||
from webssh.policy import load_host_keys
|
from webssh.policy import load_host_keys
|
||||||
|
from webssh._version import __version__
|
||||||
|
|
||||||
|
|
||||||
class TestSettings(unittest.TestCase):
|
class TestSettings(unittest.TestCase):
|
||||||
|
|
||||||
|
def test_print_version(self):
|
||||||
|
self.assertNotEqual(print_version(False), 2, msg=__version__)
|
||||||
|
with self.assertRaises(SystemExit):
|
||||||
|
self.assertEqual(print_version(True), 2, msg=__version__)
|
||||||
|
|
||||||
def test_get_host_keys_settings(self):
|
def test_get_host_keys_settings(self):
|
||||||
options.hostFile = ''
|
options.hostFile = ''
|
||||||
options.sysHostFile = ''
|
options.sysHostFile = ''
|
||||||
|
|
|
@ -1,11 +1,19 @@
|
||||||
import logging
|
import logging
|
||||||
import os.path
|
import os.path
|
||||||
|
import sys
|
||||||
import uuid
|
import uuid
|
||||||
|
|
||||||
from tornado.options import define
|
from tornado.options import define
|
||||||
from webssh.policy import (
|
from webssh.policy import (
|
||||||
load_host_keys, get_policy_class, check_policy_setting
|
load_host_keys, get_policy_class, check_policy_setting
|
||||||
)
|
)
|
||||||
|
from webssh._version import __version__
|
||||||
|
|
||||||
|
|
||||||
|
def print_version(flag):
|
||||||
|
if flag:
|
||||||
|
print(__version__)
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
|
||||||
define('address', default='127.0.0.1', help='listen address')
|
define('address', default='127.0.0.1', help='listen address')
|
||||||
|
@ -15,6 +23,8 @@ define('policy', default='warning',
|
||||||
help='missing host key policy, reject|autoadd|warning')
|
help='missing host key policy, reject|autoadd|warning')
|
||||||
define('hostFile', default='', help='User defined host keys file')
|
define('hostFile', default='', help='User defined host keys file')
|
||||||
define('sysHostFile', default='', help='System wide host keys file')
|
define('sysHostFile', default='', help='System wide host keys file')
|
||||||
|
define('version', type=bool, help='show version information',
|
||||||
|
callback=print_version)
|
||||||
|
|
||||||
|
|
||||||
base_dir = os.path.dirname(__file__)
|
base_dir = os.path.dirname(__file__)
|
||||||
|
|
Loading…
Reference in New Issue