Function TestTransformVGPUManager

controllers/transforms_test.go:4042–4144

(t *testing.T)

}

func TestTransformVGPUManager(t *testing.T) {
	node := &corev1.Node{
		ObjectMeta: metav1.ObjectMeta{
			Name: "test-node",
			Labels: map[string]string{
				nfdOSReleaseIDLabelKey: "ubuntu",
				nfdOSVersionIDLabelKey: "24.04",
				nfdKernelLabelKey:      "6.8.0-60-generic",
				commonGPULabelKey:      "true",
			},
		},
	}

	kernelModuleConfigMap := &corev1.ConfigMap{
		ObjectMeta: metav1.ObjectMeta{
			Name:      "vgpu-kernel-module-config",
			Namespace: "test-ns",
		},
		Data: map[string]string{
			"nvidia.conf": "options nvidia NVreg_EnableGpuFirmwareLogs=1\n",
		},
	}

	mockClient := fake.NewFakeClient(node, kernelModuleConfigMap)

	testCases := []struct {
		description   string
		daemonset     Daemonset
		cpSpec        *gpuv1.ClusterPolicySpec
		client        client.Client
		errorExpected bool
	}{
		{
			description: "transform vgpu manager with kernel module config",
			daemonset: NewDaemonset().
				WithInitContainer(corev1.Container{Name: "k8s-driver-manager"}).
				WithContainer(corev1.Container{Name: "nvidia-vgpu-manager-ctr"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				VGPUManager: gpuv1.VGPUManagerSpec{
					Repository:      "nvcr.io/nvidia",
					Image:           "vgpu-manager",
					Version:         "550.90.07",
					ImagePullPolicy: "IfNotPresent",
					DriverManager: gpuv1.DriverManagerSpec{
						Repository:      "nvcr.io/nvidia/cloud-native",
						Image:           "k8s-driver-manager",
						ImagePullPolicy: "IfNotPresent",
						Version:         "v0.8.0",
					},
					KernelModuleConfig: &gpuv1.KernelModuleConfigSpec{
						Name: "vgpu-kernel-module-config",
					},
				},
			},
			client:        mockClient,
			errorExpected: false,
		},
	}