Updated tests

2018-04-23 15:49:43 +08:00
parent 8e31cf5259
commit e5990f4384
3 changed files with 11 additions and 2 deletions
@@ -3,8 +3,8 @@ import unittest
 import paramiko

 from paramiko.client import RejectPolicy, WarningPolicy
-from webssh.policy import AutoAddPolicy, get_policy_dictionary, load_host_keys
-from webssh.policy import get_policy_class, check_policy_setting
+from policy import (AutoAddPolicy, get_policy_dictionary, load_host_keys,
+                    get_policy_class, check_policy_setting)


 class TestPolicy(unittest.TestCase):
@@ -25,6 +25,10 @@ class TestPolicy(unittest.TestCase):
        host_keys = load_host_keys(path)
        self.assertFalse(host_keys)

+        path = 'tests/known_hosts_example'
+        host_keys = load_host_keys(path)
+        self.assertEqual(host_keys, paramiko.hostkeys.HostKeys(path))
+
    def test_get_policy_class(self):
        keys = ['autoadd', 'reject', 'warning']
        vals = [AutoAddPolicy, RejectPolicy, WarningPolicy]