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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#!/usr/bin/python
import unittest
import sys
sys.path.append("..")
import fencing
import copy
class Test_join2(unittest.TestCase):
def test_single(self):
words = ["Mike"]
self.assertEqual(fencing._join2(words), "Mike")
self.assertEqual(fencing._join2(words, last_separator=" xor "), "Mike")
self.assertEqual(fencing._join2(words, normal_separator=" xor "), "Mike")
def test_double(self):
words = ["Mike", "John"]
self.assertEqual(fencing._join2(words), "Mike and John")
self.assertEqual(fencing._join2(words, last_separator=" xor "), "Mike xor John")
self.assertEqual(fencing._join2(words, normal_separator=" xor "), "Mike and John")
def test_triple(self):
words = ["Mike", "John", "Adam"]
self.assertEqual(fencing._join2(words), "Mike, John and Adam")
self.assertEqual(fencing._join2(words, last_separator=" xor "), "Mike, John xor Adam")
self.assertEqual(fencing._join2(words, normal_separator=" xor "), "Mike xor John and Adam")
def test_quadruple(self):
words = ["Eve", "Mike", "John", "Adam"]
self.assertEqual(fencing._join2(words), "Eve, Mike, John and Adam")
self.assertEqual(fencing._join2(words, last_separator=" xor "), "Eve, Mike, John xor Adam")
self.assertEqual(fencing._join2(words, normal_separator=" xor "), "Eve xor Mike xor John and Adam")
class Test_add_dependency_options(unittest.TestCase):
basic_set = fencing.DEPENDENCY_OPT["default"]
def test_add_nothing(self):
self.assertEqual(set(fencing._add_dependency_options([])), set(self.basic_set))
self.assertEqual(set(fencing._add_dependency_options(["not-exist"])), set(self.basic_set))
def test_add_single(self):
self.assertEqual(set(fencing._add_dependency_options(["passwd"])), set(self.basic_set + ["passwd_script"]))
def test_add_tuple(self):
self.assertEqual(set(fencing._add_dependency_options(["ssl", "passwd"])), \
set(self.basic_set + ["passwd_script", "ssl_secure", "ssl_insecure", "gnutlscli_path"]))
class Test_set_default_values(unittest.TestCase):
original_all_opt = None
def setUp(self):
# all_opt[*]["default"] can be changed during tests
self.original_all_opt = copy.deepcopy(fencing.all_opt)
def tearDown(self):
fencing.all_opt = copy.deepcopy(self.original_all_opt)
def _prepare_options(self, device_opts, args = {}):
device_opts = fencing._add_dependency_options(device_opts) + device_opts
arg_opts = args
options = dict(arg_opts)
options["device_opt"] = device_opts
fencing._update_metadata(options)
return fencing._set_default_values(options)
def test_status_io(self):
options = self._prepare_options([])
self.assertEqual(options["--action"], "reboot")
self.assertIsNone(options.get("--not-exist", None))
def test_status_fabric(self):
options = self._prepare_options(["fabric_fencing"])
self.assertEqual(options["--action"], "off")
def test_ipport_nothing(self):
# should fail because connection method (telnet/ssh/...) is not set at all
self.assertRaises(IndexError, self._prepare_options, ["ipaddr"])
def test_ipport_set(self):
options = self._prepare_options(["ipaddr", "telnet"], {"--ipport" : "999"})
self.assertEqual(options["--ipport"], "999")
def test_ipport_telnet(self):
options = self._prepare_options(["ipaddr", "telnet"])
self.assertEqual(options["--ipport"], "23")
def test_ipport_ssh(self):
options = self._prepare_options(["ipaddr", "secure"], {"--ssh" : "1"})
self.assertEqual(options["--ipport"], "22")
def test_ipport_sshtelnet_use_telnet(self):
options = self._prepare_options(["ipaddr", "secure", "telnet"])
self.assertEqual(options["--ipport"], "23")
def test_ipport_sshtelnet_use_ssh(self):
options = self._prepare_options(["ipaddr", "secure", "telnet"], {"--ssh" : "1"})
self.assertEqual(options["--ipport"], "22")
def test_ipport_ssl(self):
options = self._prepare_options(["ipaddr", "ssl"], {"--ssl-secure" : "1"})
self.assertEqual(options["--ipport"], "443")
def test_ipport_ssl_insecure_as_default(self):
fencing.all_opt["ssl_insecure"]["default"] = "1"
options = self._prepare_options(["ipaddr", "ssl"])
self.assertEqual(options["--ipport"], "443")
def test_ipport_snmp(self):
options = self._prepare_options(["ipaddr", "community"])
self.assertEqual(options["--ipport"], "161")
def test_ipport_web(self):
options = self._prepare_options(["ipaddr", "web", "ssl"])
self.assertEqual(options["--ipport"], "80")
def test_path_telnet(self):
options = self._prepare_options(["ipaddr", "telnet"])
self.assertTrue("--telnet-path" in options)
if __name__ == '__main__':
unittest.main()
|
