webssh

commit 189a30af9600460e2590ea9e77db4557aafb7e49
parent e5990f43841a57bd2989cbfc9826c176e5223f00
Author: Sheng <webmaster0115@gmail.com>
Date:   Mon, 23 Apr 2018 16:46:49 +0800

Added test_settings.py

Diffstat:
A
tests/known_hosts_example2
 | 
1
+
A
tests/test_settings.py
 | 
30
++++++++++++++++++++++++++++++

2 files changed, 31 insertions(+), 0 deletions(-)
diff --git a/tests/known_hosts_example2 b/tests/known_hosts_example2
@@ -0,0 +1 @@
+192.168.1.196 ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAINwZGQmNFADnAAlm5uFLQTrdxqpNxHdgg4JPbB3sR2kr
diff --git a/tests/test_settings.py b/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))