Revision bd6d1202 test/ganeti.rpc_unittest.py
b/test/ganeti.rpc_unittest.py | ||
---|---|---|
1 | 1 |
#!/usr/bin/python |
2 | 2 |
# |
3 | 3 |
|
4 |
# Copyright (C) 2010, 2011 Google Inc. |
|
4 |
# Copyright (C) 2010, 2011, 2012 Google Inc.
|
|
5 | 5 |
# |
6 | 6 |
# This program is free software; you can redistribute it and/or modify |
7 | 7 |
# it under the terms of the GNU General Public License as published by |
... | ... | |
61 | 61 |
return FakeSimpleStore |
62 | 62 |
|
63 | 63 |
|
64 |
def _RaiseNotImplemented(): |
|
65 |
"""Simple wrapper to raise NotImplementedError. |
|
66 |
|
|
67 |
""" |
|
68 |
raise NotImplementedError |
|
69 |
|
|
70 |
|
|
64 | 71 |
class TestRpcProcessor(unittest.TestCase): |
65 | 72 |
def _FakeAddressLookup(self, map): |
66 | 73 |
return lambda node_list: [map.get(node) for node in node_list] |
... | ... | |
331 | 338 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] |
332 | 339 |
node_addr_list = [" ".join(t) for t in zip(node_list, addr_list)] |
333 | 340 |
ssc = GetFakeSimpleStoreClass(lambda _: node_addr_list) |
334 |
result = rpc._SsconfResolver(node_list, NotImplemented, |
|
341 |
result = rpc._SsconfResolver(True, node_list, NotImplemented,
|
|
335 | 342 |
ssc=ssc, nslookup_fn=NotImplemented) |
336 | 343 |
self.assertEqual(result, zip(node_list, addr_list)) |
337 | 344 |
|
... | ... | |
341 | 348 |
ssc = GetFakeSimpleStoreClass(lambda _: []) |
342 | 349 |
node_addr_map = dict(zip(node_list, addr_list)) |
343 | 350 |
nslookup_fn = lambda name, family=None: node_addr_map.get(name) |
344 |
result = rpc._SsconfResolver(node_list, NotImplemented, |
|
351 |
result = rpc._SsconfResolver(True, node_list, NotImplemented, |
|
352 |
ssc=ssc, nslookup_fn=nslookup_fn) |
|
353 |
self.assertEqual(result, zip(node_list, addr_list)) |
|
354 |
|
|
355 |
def testDisabledSsconfIp(self): |
|
356 |
addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] |
|
357 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] |
|
358 |
ssc = GetFakeSimpleStoreClass(_RaiseNotImplemented) |
|
359 |
node_addr_map = dict(zip(node_list, addr_list)) |
|
360 |
nslookup_fn = lambda name, family=None: node_addr_map.get(name) |
|
361 |
result = rpc._SsconfResolver(False, node_list, NotImplemented, |
|
345 | 362 |
ssc=ssc, nslookup_fn=nslookup_fn) |
346 | 363 |
self.assertEqual(result, zip(node_list, addr_list)) |
347 | 364 |
|
... | ... | |
353 | 370 |
ssc = GetFakeSimpleStoreClass(lambda _: node_addr_list) |
354 | 371 |
node_addr_map = dict(zip(node_list[:n], addr_list[:n])) |
355 | 372 |
nslookup_fn = lambda name, family=None: node_addr_map.get(name) |
356 |
result = rpc._SsconfResolver(node_list, NotImplemented, |
|
373 |
result = rpc._SsconfResolver(True, node_list, NotImplemented,
|
|
357 | 374 |
ssc=ssc, nslookup_fn=nslookup_fn) |
358 | 375 |
self.assertEqual(result, zip(node_list, addr_list)) |
359 | 376 |
|
... | ... | |
362 | 379 |
node_list = ["node%d.example.com" % n for n in range(0, 255, 11)] |
363 | 380 |
node_addr_list = [" ".join(t) for t in zip(node_list, addr_list)] |
364 | 381 |
ssc = GetFakeSimpleStoreClass(lambda _: node_addr_list) |
365 |
result = rpc._SsconfResolver(node_list, NotImplemented, |
|
382 |
result = rpc._SsconfResolver(True, node_list, NotImplemented,
|
|
366 | 383 |
ssc=ssc, nslookup_fn=NotImplemented) |
367 | 384 |
self.assertEqual(result, zip(node_list, addr_list)) |
368 | 385 |
|
Also available in: Unified diff