Revision ff02b60f test/ganeti.cmdlib_unittest.py
| b/test/ganeti.cmdlib_unittest.py | ||
|---|---|---|
| 382 | 382 |
|
| 383 | 383 |
|
| 384 | 384 |
class _FakeLU: |
| 385 |
def __init__(self): |
|
| 385 |
def __init__(self, cfg=NotImplemented):
|
|
| 386 | 386 |
self.warning_log = [] |
| 387 | 387 |
self.info_log = [] |
| 388 |
self.cfg = cfg |
|
| 388 | 389 |
|
| 389 | 390 |
def LogWarning(self, text, *args): |
| 390 | 391 |
self.warning_log.append((text, args)) |
| ... | ... | |
| 678 | 679 |
self.assertEqual(ret, []) |
| 679 | 680 |
|
| 680 | 681 |
|
| 682 |
class _FakeConfigForTargetNodeIPolicy: |
|
| 683 |
def __init__(self, node_info=NotImplemented): |
|
| 684 |
self._node_info = node_info |
|
| 685 |
|
|
| 686 |
def GetNodeInfo(self, _): |
|
| 687 |
return self._node_info |
|
| 688 |
|
|
| 689 |
|
|
| 681 | 690 |
class TestCheckTargetNodeIPolicy(unittest.TestCase): |
| 682 | 691 |
def setUp(self): |
| 683 |
self.instance = objects.Instance(primary_node=objects.Node(group="foo"))
|
|
| 692 |
self.instance = objects.Instance(primary_node="blubb")
|
|
| 684 | 693 |
self.target_node = objects.Node(group="bar") |
| 685 |
self.lu = _FakeLU() |
|
| 694 |
node_info = objects.Node(group="foo") |
|
| 695 |
fake_cfg = _FakeConfigForTargetNodeIPolicy(node_info=node_info) |
|
| 696 |
self.lu = _FakeLU(cfg=fake_cfg) |
|
| 686 | 697 |
|
| 687 | 698 |
def testNoViolation(self): |
| 688 | 699 |
compute_recoder = _CallRecorder(return_value=[]) |
Also available in: Unified diff