def test_get_policy_class(self): keys = ['autoadd', 'reject', 'warning'] vals = [AutoAddPolicy, RejectPolicy, WarningPolicy] for key, val in zip(keys, vals): cls = get_policy_class(key) self.assertIs(cls, val) key = 'non-exists' with self.assertRaises(ValueError): get_policy_class(key)
def get_policy_setting(options, host_keys_settings): policy_class = get_policy_class(options.policy) logging.info(policy_class.__name__) check_policy_setting(policy_class, host_keys_settings) return policy_class()