From 3cb366a1f777a9c8ee1a1c679c65d67fe5b89db0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bosdonnat?= Date: Tue, 17 Mar 2020 11:01:48 +0100 Subject: [PATCH] virt._get_domain: don't raise an exception if there is no VM Raising an exception if there is no VM in _get_domain makes sense if looking for some VMs, but not when listing all VMs. --- tests/unit/modules/test_virt.py | 155 -------------------------------- 1 file changed, 155 deletions(-) diff --git a/tests/unit/modules/test_virt.py b/tests/unit/modules/test_virt.py index db6ba007b7..a5c876e27d 100644 --- a/tests/unit/modules/test_virt.py +++ b/tests/unit/modules/test_virt.py @@ -4898,158 +4898,3 @@ class VirtTestCase(TestCase, LoaderModuleMockMixin): "vm2", inactive=False, ) - - def test_volume_define(self): - """ - Test virt.volume_define function - """ - # Normal test case - pool_mock = MagicMock() - pool_mock.XMLDesc.return_value = "" - self.mock_conn.storagePoolLookupByName.return_value = pool_mock - - self.assertTrue( - virt.volume_define( - "testpool", - "myvm_system.qcow2", - 8192, - allocation=4096, - format="qcow2", - type="file", - ) - ) - - expected_xml = ( - "\n" - " myvm_system.qcow2\n" - " \n" - " \n" - " 8388608\n" - " 4194304\n" - " \n" - " \n" - " \n" - "" - ) - - pool_mock.createXML.assert_called_once_with(expected_xml, 0) - - # backing store test case - pool_mock.reset_mock() - self.assertTrue( - virt.volume_define( - "testpool", - "myvm_system.qcow2", - 8192, - allocation=4096, - format="qcow2", - type="file", - backing_store={"path": "/path/to/base.raw", "format": "raw"}, - ) - ) - - expected_xml = ( - "\n" - " myvm_system.qcow2\n" - " \n" - " \n" - " 8388608\n" - " 4194304\n" - " \n" - " \n" - " \n" - " \n" - " /path/to/base.raw\n" - " \n" - " \n" - "" - ) - - pool_mock.createXML.assert_called_once_with(expected_xml, 0) - - # logical pool test case - pool_mock.reset_mock() - pool_mock.XMLDesc.return_value = "" - self.mock_conn.storagePoolLookupByName.return_value = pool_mock - - self.assertTrue( - virt.volume_define( - "testVG", - "myvm_system", - 8192, - backing_store={"path": "/dev/testVG/base"}, - ) - ) - - expected_xml = ( - "\n" - " myvm_system\n" - " \n" - " \n" - " 8388608\n" - " 8388608\n" - " \n" - " \n" - " \n" - " /dev/testVG/base\n" - " \n" - "" - ) - - pool_mock.createXML.assert_called_once_with(expected_xml, 0) - - def test_volume_upload(self): - """ - Test virt.volume_upload function - """ - pool_mock = MagicMock() - vol_mock = MagicMock() - pool_mock.storageVolLookupByName.return_value = vol_mock - self.mock_conn.storagePoolLookupByName.return_value = pool_mock - stream_mock = MagicMock() - self.mock_conn.newStream.return_value = stream_mock - - open_mock = MagicMock() - close_mock = MagicMock() - with patch.dict( - os.__dict__, {"open": open_mock, "close": close_mock} - ): # pylint: disable=no-member - # Normal case - self.assertTrue(virt.volume_upload("pool0", "vol1.qcow2", "/path/to/file")) - stream_mock.sendAll.assert_called_once() - stream_mock.finish.assert_called_once() - self.mock_conn.close.assert_called_once() - vol_mock.upload.assert_called_once_with(stream_mock, 0, 0, 0) - - # Sparse upload case - stream_mock.sendAll.reset_mock() - vol_mock.upload.reset_mock() - self.assertTrue( - virt.volume_upload( - "pool0", - "vol1.qcow2", - "/path/to/file", - offset=123, - length=456, - sparse=True, - ) - ) - stream_mock.sendAll.assert_not_called() - stream_mock.sparseSendAll.assert_called_once() - vol_mock.upload.assert_called_once_with( - stream_mock, - 123, - 456, - self.mock_libvirt.VIR_STORAGE_VOL_UPLOAD_SPARSE_STREAM, - ) - - # Upload unsupported case - vol_mock.upload.side_effect = self.mock_libvirt.libvirtError("Unsupported") - self.assertRaisesRegex( - CommandExecutionError, - "Unsupported", - virt.volume_upload, - "pool0", - "vol1.qcow2", - "/path/to/file", - ) -- 2.29.2