Browse Source

Added test_settings.py

pull/12/merge
Sheng 7 years ago
parent
commit
189a30af96
  1. 1
      tests/known_hosts_example2
  2. 30
      tests/test_settings.py

1
tests/known_hosts_example2

@ -0,0 +1 @@
192.168.1.196 ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINwZGQmNFADnAAlm5uFLQTrdxqpNxHdgg4JPbB3sR2kr

30
tests/test_settings.py

@ -0,0 +1,30 @@
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))
Loading…
Cancel
Save