1 files changed, 134 insertions, 0 deletions
diff --git a/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationSpecificQemuGpuPassthroughNvidiaTest.java b/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationSpecificQemuGpuPassthroughNvidiaTest.java
new file mode 100644
index 00000000..8f50b7df
--- /dev/null
+++ b/core/modules/qemu/runvirt-plugin-qemu/src/test/java/org/openslx/runvirt/plugin/qemu/configuration/TransformationSpecificQemuGpuPassthroughNvidiaTest.java
@@ -0,0 +1,134 @@
+package org.openslx.runvirt.plugin.qemu.configuration;
+
+import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
+
+import java.io.InputStream;
+import java.math.BigInteger;
+import java.util.List;
+
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+import org.openslx.libvirt.capabilities.Capabilities;
+import org.openslx.libvirt.domain.Domain;
+import org.openslx.libvirt.domain.device.HostdevPci;
+import org.openslx.libvirt.domain.device.HostdevPciDeviceAddress;
+import org.openslx.libvirt.domain.device.Shmem;
+import org.openslx.libvirt.domain.device.Video;
+import org.openslx.libvirt.xml.LibvirtXmlDocumentException;
+import org.openslx.libvirt.xml.LibvirtXmlSerializationException;
+import org.openslx.libvirt.xml.LibvirtXmlTestResources;
+import org.openslx.libvirt.xml.LibvirtXmlValidationException;
+import org.openslx.runvirt.plugin.qemu.cmdln.CommandLineArgs;
+import org.openslx.virtualization.configuration.transformation.TransformationException;
+
+class TransformationSpecificQemuGpuPassthroughNvidiaStub extends TransformationSpecificQemuGpuPassthroughNvidia
+{
+	final String capabilityFileName;
+
+	public TransformationSpecificQemuGpuPassthroughNvidiaStub( String capabilityFileName )
+	{
+		super( null );
+
+		this.capabilityFileName = capabilityFileName;
+	}
+
+	@Override
+	protected Capabilities getCapabilities() throws TransformationException
+	{
+		final InputStream capabilityContent = LibvirtXmlTestResources.getLibvirtXmlStream( this.capabilityFileName );
+		Capabilities capabilites = null;
+
+		try {
+			capabilites = new Capabilities( capabilityContent );
+		} catch ( LibvirtXmlDocumentException | LibvirtXmlSerializationException | LibvirtXmlValidationException e ) {
+			fail( "Could not create stub for getCapabilities(): " + e.getLocalizedMessage() );
+		}
+
+		return capabilites;
+	}
+}
+
+public class TransformationSpecificQemuGpuPassthroughNvidiaTest
+{
+	@Test
+	@DisplayName( "Test transformation of VM GPU passthrough configuration if NVIDIA GPU passthrouh is required" )
+	public void testTransformationSpecificQemuGpuPassthroughNvidia() throws TransformationException
+	{
+		final TransformationSpecificQemuGpuPassthroughNvidiaStub transformation;
+		transformation = new TransformationSpecificQemuGpuPassthroughNvidiaStub( "qemu-kvm_capabilities_default.xml" );
+		final Domain config = TransformationTestUtils.getDefaultDomain();
+		final CommandLineArgs args = TransformationTestUtils.getDefaultCmdLnArgs();
+
+		transformation.transform( config, args );
+
+		final List<HostdevPci> pciDevices = config.getHostdevPciDevices();
+		assertNotNull( pciDevices );
+		assertEquals( 1, pciDevices.size() );
+
+		final HostdevPci pciDevice = pciDevices.get( 0 );
+		assertTrue( pciDevice.isManaged() );
+		assertEquals( HostdevPciDeviceAddress.valueOf( TransformationTestUtils.DEFAULT_VM_GPU0_ADDR ),
+				pciDevice.getSource() );
+
+		final List<Shmem> shmemDevices = config.getShmemDevices();
+		assertNotNull( shmemDevices );
+		assertEquals( 1, shmemDevices.size() );
+
+		final Shmem shmemDevice = shmemDevices.get( 0 );
+		assertEquals( "looking-glass", shmemDevice.getName() );
+		assertEquals( Shmem.Model.IVSHMEM_PLAIN, shmemDevice.getModel() );
+		assertEquals( BigInteger.valueOf( 67108864 ).toString(), shmemDevice.getSize().toString() );
+
+		assertEquals( TransformationSpecificQemuGpuPassthroughNvidia.HYPERV_VENDOR_ID,
+				config.getFeatureHypervVendorIdValue() );
+		assertTrue( config.isFeatureHypervVendorIdStateOn() );
+		assertTrue( config.isFeatureKvmHiddenStateOn() );
+
+		final List<Video> videoDevices = config.getVideoDevices();
+		assertNotNull( videoDevices );
+		for ( final Video videoDevice : videoDevices ) {
+			assertEquals( Video.Model.NONE, videoDevice.getModel() );
+		}
+
+		assertDoesNotThrow( () -> config.validateXml() );
+	}
+
+	@Test
+	@DisplayName( "Test transformation of VM GPU passthrough configuration if NVIDIA GPU passthrouh is not specified" )
+	public void testTransformationSpecificQemuGpuPassthroughNvidiaNoGpu() throws TransformationException
+	{
+		final TransformationSpecificQemuGpuPassthroughNvidiaStub transformation;
+		transformation = new TransformationSpecificQemuGpuPassthroughNvidiaStub( "qemu-kvm_capabilities_default.xml" );
+		final Domain config = TransformationTestUtils.getDefaultDomain();
+		final CommandLineArgs args = TransformationTestUtils.getEmptyCmdLnArgs();
+
+		transformation.transform( config, args );
+
+		final List<HostdevPci> pciDevices = config.getHostdevPciDevices();
+		assertNotNull( pciDevices );
+		assertEquals( 0, pciDevices.size() );
+
+		final List<Shmem> shmemDevices = config.getShmemDevices();
+		assertNotNull( shmemDevices );
+		assertEquals( 0, shmemDevices.size() );
+
+		assertNotEquals( TransformationSpecificQemuGpuPassthroughNvidia.HYPERV_VENDOR_ID,
+				config.getFeatureHypervVendorIdValue() );
+		assertFalse( config.isFeatureHypervVendorIdStateOn() );
+		assertFalse( config.isFeatureKvmHiddenStateOn() );
+
+		final List<Video> videoDevices = config.getVideoDevices();
+		assertNotNull( videoDevices );
+		for ( final Video videoDevice : videoDevices ) {
+			assertNotEquals( Video.Model.NONE, videoDevice.getModel() );
+		}
+
+		assertDoesNotThrow( () -> config.validateXml() );
+	}
+}