mirror of
https://github.com/huashengdun/webssh.git
synced 2026-08-23 22:26:28 +00:00
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