Commit 189a30af by Sheng

Added test_settings.py

parent e5990f43
192.168.1.196 ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINwZGQmNFADnAAlm5uFLQTrdxqpNxHdgg4JPbB3sR2kr
import os.path
import unittest
import tornado.options as options
from settings import get_host_keys_settings, base_dir
from policy import load_host_keys
class TestSettings(unittest.TestCase):
def test_get_host_keys_settings(self):
options.hostFile = ''
options.sysHostFile = ''
dic = get_host_keys_settings(options)
filename = os.path.join(base_dir, 'known_hosts')
self.assertEqual(dic['host_keys'], load_host_keys(filename))
self.assertEqual(dic['host_keys_filename'], filename)
self.assertEqual(
dic['system_host_keys'],
load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
)
options.hostFile = 'tests/known_hosts_example'
options.sysHostFile = 'tests/known_hosts_example2'
dic2 = get_host_keys_settings(options)
self.assertEqual(dic2['host_keys'], load_host_keys(options.hostFile))
self.assertEqual(dic2['host_keys_filename'], options.hostFile)
self.assertEqual(dic2['system_host_keys'],
load_host_keys(options.sysHostFile))
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment