4 # Copyright (C) 2011 Google Inc.
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
11 # This program is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 # General Public License for more details.
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
22 """Script for testing ganeti.ht"""
31 class TestTypeChecks(unittest.TestCase):
33 self.assertFalse(ht.TNotNone(None))
34 self.assertTrue(ht.TNone(None))
36 for val in [0, True, "", "Hello World", [], range(5)]:
37 self.assertTrue(ht.TNotNone(val))
38 self.assertFalse(ht.TNone(val))
41 self.assertTrue(ht.TBool(True))
42 self.assertTrue(ht.TBool(False))
44 for val in [0, None, "", [], "Hello"]:
45 self.assertFalse(ht.TBool(val))
47 for val in [True, -449, 1, 3, "x", "abc", [1, 2]]:
48 self.assertTrue(ht.TTrue(val))
50 for val in [False, 0, None, []]:
51 self.assertFalse(ht.TTrue(val))
54 for val in [-100, -3, 0, 16, 128, 923874]:
55 self.assertTrue(ht.TInt(val))
56 self.assertTrue(ht.TNumber(val))
58 for val in [False, True, None, "", [], "Hello", 0.0, 0.23, -3818.163]:
59 self.assertFalse(ht.TInt(val))
61 for val in range(0, 100, 4):
62 self.assertTrue(ht.TPositiveInt(val))
64 self.assertFalse(ht.TPositiveInt(neg))
65 self.assertFalse(ht.TStrictPositiveInt(neg))
67 self.assertFalse(ht.TPositiveInt(0.1 + val))
68 self.assertFalse(ht.TStrictPositiveInt(0.1 + val))
70 for val in [0, 0.1, 0.9, -0.3]:
71 self.assertFalse(ht.TStrictPositiveInt(val))
73 for val in range(1, 100, 4):
74 self.assertTrue(ht.TStrictPositiveInt(val))
75 self.assertFalse(ht.TStrictPositiveInt(0.1 + val))
78 for val in [-100.21, -3.0, 0.0, 16.12, 128.3433, 923874.928]:
79 self.assertTrue(ht.TFloat(val))
80 self.assertTrue(ht.TNumber(val))
82 for val in [False, True, None, "", [], "Hello", 0, 28, -1, -3281]:
83 self.assertFalse(ht.TFloat(val))
86 for val in [-100, -3, 0, 16, 128, 923874,
87 -100.21, -3.0, 0.0, 16.12, 128.3433, 923874.928]:
88 self.assertTrue(ht.TNumber(val))
90 for val in [False, True, None, "", [], "Hello", "1"]:
91 self.assertFalse(ht.TNumber(val))
94 for val in ["", "abc", "Hello World", "123",
95 u"", u"\u272C", u"abc"]:
96 self.assertTrue(ht.TString(val))
98 for val in [False, True, None, [], 0, 1, 5, -193, 93.8582]:
99 self.assertFalse(ht.TString(val))
101 def testElemOf(self):
102 fn = ht.TElemOf(range(10))
103 self.assertTrue(fn(0))
104 self.assertTrue(fn(3))
105 self.assertTrue(fn(9))
106 self.assertFalse(fn(-1))
107 self.assertFalse(fn(100))
110 self.assertFalse(fn(0))
111 self.assertFalse(fn(100))
112 self.assertFalse(fn(True))
114 fn = ht.TElemOf(["Hello", "World"])
115 self.assertTrue(fn("Hello"))
116 self.assertTrue(fn("World"))
117 self.assertFalse(fn("e"))
120 for val in [[], range(10), ["Hello", "World", "!"]]:
121 self.assertTrue(ht.TList(val))
123 for val in [False, True, None, {}, 0, 1, 5, -193, 93.8582]:
124 self.assertFalse(ht.TList(val))
127 for val in [{}, dict.fromkeys(range(10)), {"Hello": [], "World": "!"}]:
128 self.assertTrue(ht.TDict(val))
130 for val in [False, True, None, [], 0, 1, 5, -193, 93.8582]:
131 self.assertFalse(ht.TDict(val))
133 def testIsLength(self):
134 fn = ht.TIsLength(10)
135 self.assertTrue(fn(range(10)))
136 self.assertFalse(fn(range(1)))
137 self.assertFalse(fn(range(100)))
140 fn = ht.TAnd(ht.TNotNone, ht.TString)
141 self.assertTrue(fn(""))
142 self.assertFalse(fn(1))
143 self.assertFalse(fn(None))
146 fn = ht.TOr(ht.TNone, ht.TAnd(ht.TString, ht.TIsLength(5)))
147 self.assertTrue(fn("12345"))
148 self.assertTrue(fn(None))
149 self.assertFalse(fn(1))
150 self.assertFalse(fn(""))
151 self.assertFalse(fn("abc"))
154 self.assertTrue(ht.TMap(str, ht.TString)(123))
155 self.assertTrue(ht.TMap(int, ht.TInt)("9999"))
156 self.assertFalse(ht.TMap(lambda x: x + 100, ht.TString)(123))
158 def testNonEmptyString(self):
159 self.assertTrue(ht.TNonEmptyString("xyz"))
160 self.assertTrue(ht.TNonEmptyString("Hello World"))
161 self.assertFalse(ht.TNonEmptyString(""))
162 self.assertFalse(ht.TNonEmptyString(None))
163 self.assertFalse(ht.TNonEmptyString([]))
165 def testMaybeString(self):
166 self.assertTrue(ht.TMaybeString("xyz"))
167 self.assertTrue(ht.TMaybeString("Hello World"))
168 self.assertTrue(ht.TMaybeString(None))
169 self.assertFalse(ht.TMaybeString(""))
170 self.assertFalse(ht.TMaybeString([]))
172 def testMaybeBool(self):
173 self.assertTrue(ht.TMaybeBool(False))
174 self.assertTrue(ht.TMaybeBool(True))
175 self.assertTrue(ht.TMaybeBool(None))
176 self.assertFalse(ht.TMaybeBool([]))
177 self.assertFalse(ht.TMaybeBool("0"))
178 self.assertFalse(ht.TMaybeBool("False"))
180 def testListOf(self):
181 fn = ht.TListOf(ht.TNonEmptyString)
182 self.assertTrue(fn([]))
183 self.assertTrue(fn(["x"]))
184 self.assertTrue(fn(["Hello", "World"]))
185 self.assertFalse(fn(None))
186 self.assertFalse(fn(False))
187 self.assertFalse(fn(range(3)))
188 self.assertFalse(fn(["x", None]))
190 def testDictOf(self):
191 fn = ht.TDictOf(ht.TNonEmptyString, ht.TInt)
192 self.assertTrue(fn({}))
193 self.assertTrue(fn({"x": 123, "y": 999}))
194 self.assertFalse(fn(None))
195 self.assertFalse(fn({1: "x"}))
196 self.assertFalse(fn({"x": ""}))
197 self.assertFalse(fn({"x": None}))
198 self.assertFalse(fn({"": 8234}))
200 def testStrictDictRequireAllExclusive(self):
201 fn = ht.TStrictDict(True, True, { "a": ht.TInt, })
202 self.assertFalse(fn(1))
203 self.assertFalse(fn(None))
204 self.assertFalse(fn({}))
205 self.assertFalse(fn({"a": "Hello", }))
206 self.assertFalse(fn({"unknown": 999,}))
207 self.assertFalse(fn({"unknown": None,}))
209 self.assertTrue(fn({"a": 123, }))
210 self.assertTrue(fn({"a": -5, }))
212 fn = ht.TStrictDict(True, True, { "a": ht.TInt, "x": ht.TString, })
213 self.assertFalse(fn({}))
214 self.assertFalse(fn({"a": -5, }))
215 self.assertTrue(fn({"a": 123, "x": "", }))
216 self.assertFalse(fn({"a": 123, "x": None, }))
218 def testStrictDictExclusive(self):
219 fn = ht.TStrictDict(False, True, { "a": ht.TInt, "b": ht.TList, })
220 self.assertTrue(fn({}))
221 self.assertTrue(fn({"a": 123, }))
222 self.assertTrue(fn({"b": range(4), }))
223 self.assertFalse(fn({"b": 123, }))
225 self.assertFalse(fn({"foo": {}, }))
226 self.assertFalse(fn({"bar": object(), }))
228 def testStrictDictRequireAll(self):
229 fn = ht.TStrictDict(True, False, { "a": ht.TInt, "m": ht.TInt, })
230 self.assertTrue(fn({"a": 1, "m": 2, "bar": object(), }))
231 self.assertFalse(fn({}))
232 self.assertFalse(fn({"a": 1, "bar": object(), }))
233 self.assertFalse(fn({"a": 1, "m": [], "bar": object(), }))
235 def testStrictDict(self):
236 fn = ht.TStrictDict(False, False, { "a": ht.TInt, })
237 self.assertTrue(fn({}))
238 self.assertFalse(fn({"a": ""}))
239 self.assertTrue(fn({"a": 11}))
240 self.assertTrue(fn({"other": 11}))
241 self.assertTrue(fn({"other": object()}))
244 for i in [0, 1, 4395, 2347625220]:
245 self.assertTrue(ht.TJobId(i))
246 self.assertTrue(ht.TJobId(str(i)))
247 self.assertFalse(ht.TJobId(-(i + 1)))
249 for i in ["", "-", ".", ",", "a", "99j", "job-123", "\t", " 83 ",
250 None, [], {}, object()]:
251 self.assertFalse(ht.TJobId(i))
253 def testRelativeJobId(self):
254 for i in [-1, -93, -4395]:
255 self.assertTrue(ht.TRelativeJobId(i))
256 self.assertFalse(ht.TRelativeJobId(str(i)))
258 for i in [0, 1, 2, 10, 9289, "", "0", "-1", "-999"]:
259 self.assertFalse(ht.TRelativeJobId(i))
260 self.assertFalse(ht.TRelativeJobId(str(i)))
263 self.assertRaises(AssertionError, ht.TItems, [])
265 fn = ht.TItems([ht.TString])
266 self.assertFalse(fn([0]))
267 self.assertFalse(fn([None]))
268 self.assertTrue(fn(["Hello"]))
269 self.assertTrue(fn(["Hello", "World"]))
270 self.assertTrue(fn(["Hello", 0, 1, 2, "anything"]))
272 fn = ht.TItems([ht.TAny, ht.TInt, ht.TAny])
273 self.assertTrue(fn(["Hello", 0, []]))
274 self.assertTrue(fn(["Hello", 893782]))
275 self.assertTrue(fn([{}, -938210858947, None]))
276 self.assertFalse(fn(["Hello", []]))
279 if __name__ == "__main__":
280 testutils.GanetiTestProgram()