def test_rebooted_error():
"""
rebooted state test, error case.
"""
with patch.dict(virt.__opts__, {"test": False}):
reboot_mock = MagicMock(return_value=True)
with patch.dict(
virt.__salt__,
{
"virt.list_domains":
MagicMock(return_value=["myvm", "vm1"]),
"virt.reboot":
MagicMock(side_effect=virt.libvirt.libvirtError("Some error")),
},
):
assert virt.rebooted("myvm") == {
"name": "myvm",
"result": False,
"changes": {
"ignored": [{
"domain": "myvm",
"issue": "Some error"
}]
},
"comment": "No changes had happened",
}
def test_rebooted(test):
"""
rebooted state test
"""
with patch.dict(virt.__opts__, {"test": test}):
reboot_mock = MagicMock(return_value=True)
with patch.dict(
virt.__salt__,
{
"virt.list_domains": MagicMock(return_value=["myvm", "vm1"]),
"virt.reboot": reboot_mock,
},
):
assert virt.rebooted(
"myvm", connection="myconnection", username="******", password="******",
) == {
"name": "myvm",
"result": True if not test else None,
"changes": {"rebooted": [{"domain": "myvm", "reboot": True}]},
"comment": "Machine has been rebooted",
}
if not test:
reboot_mock.assert_called_with(
"myvm",
connection="myconnection",
username="******",
password="******",
)
else:
reboot_mock.assert_not_called()
def test_rebooted(self):
'''
rebooted state test cases.
'''
ret = {'name': 'myvm',
'changes': {},
'result': True}
reboot_mock = MagicMock(return_value=True)
with patch.dict(virt.__salt__, { # pylint: disable=no-member
'virt.list_domains': MagicMock(return_value=['myvm', 'vm1']),
'virt.reboot': reboot_mock
}):
ret.update({'changes': {
'rebooted': [{'domain': 'myvm', 'reboot': True}]
},
'comment': 'Machine has been rebooted'})
self.assertDictEqual(virt.rebooted('myvm'), ret)
reboot_mock.assert_called_with('myvm', connection=None, username=None, password=None)
with patch.dict(virt.__salt__, { # pylint: disable=no-member
'virt.list_domains': MagicMock(return_value=['myvm', 'vm1']),
'virt.reboot': reboot_mock,
}):
self.assertDictEqual(virt.rebooted('myvm',
connection='myconnection',
username='******',
password='******'), ret)
reboot_mock.assert_called_with('myvm',
connection='myconnection',
username='******',
password='******')
with patch.dict(virt.__salt__, { # pylint: disable=no-member
'virt.list_domains': MagicMock(return_value=['myvm', 'vm1']),
'virt.reboot': MagicMock(side_effect=self.mock_libvirt.libvirtError('Some error'))
}):
ret.update({'changes': {'ignored': [{'domain': 'myvm', 'issue': 'Some error'}]},
'result': False,
'comment': 'No changes had happened'})
self.assertDictEqual(virt.rebooted('myvm'), ret)
with patch.dict(virt.__salt__, {'virt.list_domains': MagicMock(return_value=[])}): # pylint: disable=no-member
ret.update({'changes': {}, 'result': False, 'comment': 'No changes had happened'})
self.assertDictEqual(virt.rebooted('myvm'), ret)
def test_rebooted_not_existing(test):
"""
rebooted state test cases.
"""
with patch.dict(virt.__opts__, {"test": test}):
with patch.dict(virt.__salt__,
{"virt.list_domains": MagicMock(return_value=[])}):
assert virt.rebooted("myvm") == {
"name": "myvm",
"changes": {},
"result": False,
"comment": "No changes had happened",
}
