Added test_get_policy_dictionary
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
import unittest
|
||||
from paramiko.client import RejectPolicy, WarningPolicy
|
||||
from webssh.policy import AutoAddPolicy, get_policy_dictionary
|
||||
|
||||
|
||||
class TestPolicy(unittest.TestCase):
|
||||
|
||||
def test_get_policy_dictionary(self):
|
||||
classes = [AutoAddPolicy, RejectPolicy, WarningPolicy]
|
||||
dic = get_policy_dictionary()
|
||||
for cls in classes:
|
||||
val = dic[cls.__name__.lower()]
|
||||
self.assertIs(cls, val)
|
||||
Reference in New Issue
Block a user