Synnefo » snf-ganeti

import mock

from ganeti import hypervisor

class TestGetInstanceList(unittest.TestCase):

  def setUp(self):

    self._test_hv = self._TestHypervisor()

    self._test_hv.ListInstances = mock.Mock(

      return_value=["instance1", "instance2", "instance3"] )

  class _TestHypervisor(hypervisor.hv_base.BaseHypervisor):

    def __init__(self):

      hypervisor.hv_base.BaseHypervisor.__init__(self)

  def _GetHypervisor(self, name):

    return self._test_hv

  def testHvparams(self):

    fake_hvparams = {constants.HV_XEN_CMD: constants.XEN_CMD_XL}

    hvparams = {constants.HT_FAKE: fake_hvparams}

    backend.GetInstanceList([constants.HT_FAKE], all_hvparams=hvparams,

                            get_hv_fn=self._GetHypervisor)

    self._test_hv.ListInstances.assert_called_with(hvparams=fake_hvparams)