Revision eb202c13 test/ganeti.rpc_unittest.py
b/test/ganeti.rpc_unittest.py | ||
---|---|---|
56 | 56 |
self._response_fn(req) |
57 | 57 |
|
58 | 58 |
|
59 |
def GetFakeSimpleStoreClass(fn): |
|
60 |
class FakeSimpleStore: |
|
61 |
GetNodePrimaryIPList = fn |
|
62 |
|
|
63 |
return FakeSimpleStore |
|
64 |
|
|
65 |
|
|
59 | 66 |
class TestClient(unittest.TestCase): |
67 |
def _FakeAddressLookup(self, map): |
|
68 |
return lambda node_list: [map.get(node) for node in node_list] |
|
69 |
|
|
60 | 70 |
def _GetVersionResponse(self, req): |
61 | 71 |
self.assertEqual(req.host, "localhost") |
62 | 72 |
self.assertEqual(req.port, 24094) |
... | ... | |
66 | 76 |
req.resp_body = serializer.DumpJson((True, 123)) |
67 | 77 |
|
68 | 78 |
def testVersionSuccess(self): |
69 |
client = rpc.Client("version", None, 24094) |
|
79 |
fn = self._FakeAddressLookup({"localhost": "localhost"}) |
|
80 |
client = rpc.Client("version", None, 24094, address_lookup_fn=fn) |
|
70 | 81 |
client.ConnectNode("localhost") |
71 | 82 |
pool = FakeHttpPool(self._GetVersionResponse) |
72 | 83 |
result = client.GetResults(http_pool=pool) |
... | ... | |
90 | 101 |
|
91 | 102 |
def testMultiVersionSuccess(self): |
92 | 103 |
nodes = ["node%s" % i for i in range(50)] |
93 |
client = rpc.Client("version", None, 23245) |
|
104 |
fn = self._FakeAddressLookup(dict(zip(nodes, nodes))) |
|
105 |
client = rpc.Client("version", None, 23245, address_lookup_fn=fn) |
|
94 | 106 |
client.ConnectList(nodes) |
95 | 107 |
|
96 | 108 |
pool = FakeHttpPool(self._GetMultiVersionResponse) |
... | ... | |
115 | 127 |
req.resp_body = serializer.DumpJson((False, "Unknown error")) |
116 | 128 |
|
117 | 129 |
def testVersionFailure(self): |
118 |
client = rpc.Client("version", None, 5903) |
|
130 |
lookup_map = {"aef9ur4i.example.com": "aef9ur4i.example.com"} |
|
131 |
fn = self._FakeAddressLookup(lookup_map) |
|
132 |
client = rpc.Client("version", None, 5903, address_lookup_fn=fn) |
|
119 | 133 |
client.ConnectNode("aef9ur4i.example.com") |
120 | 134 |
pool = FakeHttpPool(self._GetVersionResponseFail) |
121 | 135 |
result = client.GetResults(http_pool=pool) |
... | ... | |
152 | 166 |
|
153 | 167 |
def testHttpError(self): |
154 | 168 |
nodes = ["uaf6pbbv%s" % i for i in range(50)] |
169 |
fn = self._FakeAddressLookup(dict(zip(nodes, nodes))) |
|
155 | 170 |
|
156 | 171 |
httperrnodes = set(nodes[1::7]) |
157 | 172 |
self.assertEqual(len(httperrnodes), 7) |
... | ... | |
161 | 176 |
|
162 | 177 |
self.assertEqual(len(set(nodes) - failnodes - httperrnodes), 29) |
163 | 178 |
|
164 |
client = rpc.Client("vg_list", None, 15165) |
|
179 |
client = rpc.Client("vg_list", None, 15165, address_lookup_fn=fn)
|
|
165 | 180 |
client.ConnectList(nodes) |
166 | 181 |
|
167 | 182 |
pool = FakeHttpPool(compat.partial(self._GetHttpErrorResponse, |
... | ... | |
203 | 218 |
req.resp_body = serializer.DumpJson("invalid response") |
204 | 219 |
|
205 | 220 |
def testInvalidResponse(self): |
206 |
client = rpc.Client("version", None, 19978) |
|
221 |
lookup_map = {"oqo7lanhly.example.com": "oqo7lanhly.example.com"} |
|
222 |
fn = self._FakeAddressLookup(lookup_map) |
|
223 |
client = rpc.Client("version", None, 19978, address_lookup_fn=fn) |
|
207 | 224 |
for fn in [self._GetInvalidResponseA, self._GetInvalidResponseB]: |
208 | 225 |
client.ConnectNode("oqo7lanhly.example.com") |
209 | 226 |
pool = FakeHttpPool(fn) |
... | ... | |
218 | 235 |
self.assertRaises(errors.OpExecError, lhresp.Raise, "failed") |
219 | 236 |
self.assertEqual(pool.reqcount, 1) |
220 | 237 |
|
238 |
def testAddressLookupSimpleStore(self): |
|
239 |
addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] |
|
240 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] |
|
241 |
node_addr_list = [ " ".join(t) for t in zip(node_list, addr_list)] |
|
242 |
ssc = GetFakeSimpleStoreClass(lambda s: node_addr_list) |
|
243 |
result = rpc._AddressLookup(node_list, ssc=ssc) |
|
244 |
self.assertEqual(result, addr_list) |
|
245 |
|
|
246 |
def testAddressLookupNSLookup(self): |
|
247 |
addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] |
|
248 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] |
|
249 |
ssc = GetFakeSimpleStoreClass(lambda s: []) |
|
250 |
node_addr_map = dict(zip(node_list, addr_list)) |
|
251 |
nslookup_fn = lambda name: (None, None, [node_addr_map.get(name)]) |
|
252 |
result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn) |
|
253 |
self.assertEqual(result, addr_list) |
|
254 |
|
|
255 |
def testAddressLookupBoth(self): |
|
256 |
addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] |
|
257 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] |
|
258 |
n = len(addr_list) / 2 |
|
259 |
node_addr_list = [ " ".join(t) for t in zip(node_list[n:], addr_list[n:])] |
|
260 |
ssc = GetFakeSimpleStoreClass(lambda s: node_addr_list) |
|
261 |
node_addr_map = dict(zip(node_list[:n], addr_list[:n])) |
|
262 |
nslookup_fn = lambda name: (None, None, [node_addr_map.get(name)]) |
|
263 |
result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn) |
|
264 |
self.assertEqual(result, addr_list) |
|
265 |
|
|
221 | 266 |
|
222 | 267 |
if __name__ == "__main__": |
223 | 268 |
testutils.GanetiTestProgram() |
Also available in: Unified diff