Revision b9e12624 test/py/ganeti.backend_unittest.py
| b/test/py/ganeti.backend_unittest.py | ||
|---|---|---|
| 33 | 33 |
from ganeti import errors |
| 34 | 34 |
from ganeti import hypervisor |
| 35 | 35 |
from ganeti import netutils |
| 36 |
from ganeti import objects |
|
| 36 | 37 |
from ganeti import utils |
| 37 | 38 |
|
| 38 | 39 |
|
| ... | ... | |
| 590 | 591 |
self._test_hv.ListInstances.assert_called_with(hvparams=fake_hvparams) |
| 591 | 592 |
|
| 592 | 593 |
|
| 594 |
class TestInstanceConsoleInfo(unittest.TestCase): |
|
| 595 |
|
|
| 596 |
def setUp(self): |
|
| 597 |
self._test_hv_a = self._TestHypervisor() |
|
| 598 |
self._test_hv_a.GetInstanceConsole = mock.Mock( |
|
| 599 |
return_value = objects.InstanceConsole(instance="inst", kind="aHy") |
|
| 600 |
) |
|
| 601 |
self._test_hv_b = self._TestHypervisor() |
|
| 602 |
self._test_hv_b.GetInstanceConsole = mock.Mock( |
|
| 603 |
return_value = objects.InstanceConsole(instance="inst", kind="bHy") |
|
| 604 |
) |
|
| 605 |
|
|
| 606 |
class _TestHypervisor(hypervisor.hv_base.BaseHypervisor): |
|
| 607 |
def __init__(self): |
|
| 608 |
hypervisor.hv_base.BaseHypervisor.__init__(self) |
|
| 609 |
|
|
| 610 |
def _GetHypervisor(self, name): |
|
| 611 |
if name == "a": |
|
| 612 |
return self._test_hv_a |
|
| 613 |
else: |
|
| 614 |
return self._test_hv_b |
|
| 615 |
|
|
| 616 |
def testRightHypervisor(self): |
|
| 617 |
dictMaker = lambda hyName: {
|
|
| 618 |
"instance":{"hypervisor":hyName},
|
|
| 619 |
"node":{},
|
|
| 620 |
"hvParams":{},
|
|
| 621 |
"beParams":{},
|
|
| 622 |
} |
|
| 623 |
|
|
| 624 |
call = {
|
|
| 625 |
'i1':dictMaker("a"),
|
|
| 626 |
'i2':dictMaker("b"),
|
|
| 627 |
} |
|
| 628 |
|
|
| 629 |
res = backend.GetInstanceConsoleInfo(call, get_hv_fn=self._GetHypervisor) |
|
| 630 |
|
|
| 631 |
self.assertTrue(res["i1"]["kind"] == "aHy") |
|
| 632 |
self.assertTrue(res["i2"]["kind"] == "bHy") |
|
| 633 |
|
|
| 634 |
|
|
| 593 | 635 |
class TestGetHvInfo(unittest.TestCase): |
| 594 | 636 |
|
| 595 | 637 |
def setUp(self): |
Also available in: Unified diff