forked from qutorial/ddwrt-config-tool
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_sshd.py
65 lines (53 loc) · 2.27 KB
/
test_sshd.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import unittest
from router import Router
class TestSSHD(unittest.TestCase):
def test_sshd_enable(self):
nvram = {'sshd_enable':'0',
'sshd_wanport': '33',
'sshd_port': '44',
'sshd_passwd_auth': '0',
'sshd_authorized_keys': '',
'sshd_forwarding': '0',
}
router = Router(nvram)
router.enableSshd(True)
self.assertEqual(router.nvram['sshd_enable'], '1')
self.assertEqual(router.nvram['sshd_wanport'], '22')
self.assertEqual(router.nvram['sshd_port'], '22')
self.assertEqual(router.nvram['sshd_passwd_auth'], '1')
self.assertEqual(router.nvram['sshd_authorized_keys'], '')
self.assertEqual(router.nvram['sshd_forwarding'], '0')
def test_sshd_disable(self):
nvram = {'sshd_enable': '1',
'sshd_wanport': '22',
'sshd_port': '22',
'sshd_passwd_auth': '1',
'sshd_authorized_keys': '',
'sshd_forwarding': '0',
}
router = Router(nvram)
router.enableSshd(False)
self.assertEqual(router.nvram['sshd_enable'], '0')
# self.assertEqual(router.nvram['sshd_wanport'], '22')
# self.assertEqual(router.nvram['sshd_port'], '22')
# self.assertEqual(router.nvram['sshd_passwd_auth'], '1')
# self.assertEqual(router.nvram['sshd_authorized_keys'], '')
# self.assertEqual(router.nvram['sshd_forwarding'], '0')
def test_sshd_no_change(self):
nvram = {'sshd_enable': '1',
'sshd_wanport': '22',
'sshd_port': '22',
'sshd_passwd_auth': '1',
'sshd_authorized_keys': '',
'sshd_forwarding': '0',
}
router = Router(nvram)
router.enableSshd(False)
self.assertEqual(router.nvram['sshd_enable'], '1')
self.assertEqual(router.nvram['sshd_wanport'], '22')
self.assertEqual(router.nvram['sshd_port'], '22')
self.assertEqual(router.nvram['sshd_passwd_auth'], '1')
self.assertEqual(router.nvram['sshd_authorized_keys'], '')
self.assertEqual(router.nvram['sshd_forwarding'], '0')
if __name__ == '__main__':
unittest.main()