diff --git a/tests/test_settings.py b/tests/test_settings.py index fc5cb7c..3a83315 100644 --- a/tests/test_settings.py +++ b/tests/test_settings.py @@ -1,8 +1,9 @@ import os.path import unittest +import paramiko import tornado.options as options -from settings import get_host_keys_settings, base_dir +from settings import get_host_keys_settings, get_policy_setting, base_dir from policy import load_host_keys @@ -28,3 +29,29 @@ class TestSettings(unittest.TestCase): self.assertEqual(dic2['host_keys_filename'], options.hostFile) self.assertEqual(dic2['system_host_keys'], load_host_keys(options.sysHostFile)) + + def test_get_policy_setting(self): + options.policy = 'warning' + options.hostFile = '' + options.sysHostFile = '' + settings = get_host_keys_settings(options) + instance = get_policy_setting(options, settings) + self.assertIsInstance(instance, paramiko.client.WarningPolicy) + + options.policy = 'autoadd' + options.hostFile = '' + options.sysHostFile = '' + settings = get_host_keys_settings(options) + instance = get_policy_setting(options, settings) + self.assertIsInstance(instance, paramiko.client.AutoAddPolicy) + + options.policy = 'reject' + options.hostFile = '' + options.sysHostFile = '' + settings = get_host_keys_settings(options) + try: + instance = get_policy_setting(options, settings) + except ValueError: + pass + else: + self.assertIsInstance(instance, paramiko.client.RejectPolicy)