Fix test_get_hypervisor: mock bhyve

This commit is contained in:
ch3ll 2020-04-21 18:46:49 -04:00 committed by Daniel Wozniak
parent 0d284b7a7e
commit 2bf7eacf6d

View file

@ -3322,9 +3322,10 @@ class VirtTestCase(TestCase, LoaderModuleMockMixin):
# pylint: enable=no-member
self.assertEqual(names, virt.pool_list_volumes("default"))
@patch("salt.modules.virt._is_bhyve_hyper", return_value=False)
@patch("salt.modules.virt._is_kvm_hyper", return_value=True)
@patch("salt.modules.virt._is_xen_hyper", return_value=False)
def test_get_hypervisor(self, isxen_mock, iskvm_mock):
def test_get_hypervisor(self, isxen_mock, iskvm_mock, is_bhyve_mock):
"""
test the virt.get_hypervisor() function
"""
@ -3333,6 +3334,9 @@ class VirtTestCase(TestCase, LoaderModuleMockMixin):
iskvm_mock.return_value = False
self.assertIsNone(virt.get_hypervisor())
is_bhyve_mock.return_value = False
self.assertIsNone(virt.get_hypervisor())
isxen_mock.return_value = True
self.assertEqual("xen", virt.get_hypervisor())