1 files changed, 55 insertions, 0 deletions
diff --git a/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskStorageDevicesTest.java b/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskStorageDevicesTest.java
new file mode 100644
index 00000000..20fd2fb6
--- /dev/null
+++ b/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationGenericDiskStorageDevicesTest.java
@@ -0,0 +1,55 @@
+package org.openslx.runvirt.plugin.qemu.configuration;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotEquals;
+
+import java.util.ArrayList;
+
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+import org.openslx.libvirt.domain.Domain;
+import org.openslx.libvirt.domain.device.Disk.StorageType;
+import org.openslx.libvirt.domain.device.DiskStorage;
+import org.openslx.runvirt.plugin.qemu.cmdln.CommandLineArgs;
+import org.openslx.virtualization.configuration.transformation.TransformationException;
+
+public class TransformationGenericDiskStorageDevicesTest
+{
+	@Test
+	@DisplayName( "Test filtering of VM disk storage devices configuration with specified input data" )
+	public void testFilterGenericDiskStorageDevices() throws TransformationException
+	{
+		final TransformationGenericDiskStorageDevices filter = new TransformationGenericDiskStorageDevices();
+		final Domain config = TransformationTestUtils.getDefaultDomain();
+		final CommandLineArgs args = TransformationTestUtils.getDefaultCmdLnArgs();
+
+		final ArrayList<DiskStorage> devicesBeforeFiltering = config.getDiskStorageDevices();
+		assertEquals( 1, devicesBeforeFiltering.size() );
+		assertNotEquals( StorageType.FILE, devicesBeforeFiltering.get( 0 ).getStorageType() );
+		assertNotEquals( TransformationTestUtils.DEFAULT_VM_HDD0, devicesBeforeFiltering.get( 0 ).getStorageSource() );
+
+		filter.transform( config, args );
+
+		final ArrayList<DiskStorage> devicesAfterFiltering = config.getDiskStorageDevices();
+		assertEquals( 1, devicesAfterFiltering.size() );
+		assertEquals( StorageType.FILE, devicesAfterFiltering.get( 0 ).getStorageType() );
+		assertEquals( TransformationTestUtils.DEFAULT_VM_HDD0, devicesAfterFiltering.get( 0 ).getStorageSource() );
+	}
+
+	@Test
+	@DisplayName( "Test filtering of VM disk storage devices configuration with unspecified input data" )
+	public void testFilterGenericDiskStorageDevicesNoData() throws TransformationException
+	{
+		final TransformationGenericDiskStorageDevices filter = new TransformationGenericDiskStorageDevices();
+		final Domain config = TransformationTestUtils.getDefaultDomain();
+		final CommandLineArgs args = TransformationTestUtils.getEmptyCmdLnArgs();
+
+		final ArrayList<DiskStorage> devicesBeforeFiltering = config.getDiskStorageDevices();
+		assertEquals( 1, devicesBeforeFiltering.size() );
+
+		filter.transform( config, args );
+
+		final ArrayList<DiskStorage> devicesAfterFiltering = config.getDiskStorageDevices();
+		assertEquals( 0, devicesAfterFiltering.size() );
+	}
+}