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", }