Function TestDriverSpec

internal/state/driver_test.go:190–279

(t *testing.T)

}

func TestDriverSpec(t *testing.T) {
	const (
		testName = "driver-full-spec"
	)
	state, err := NewStateDriver(nil, "", nil, manifestDir)
	require.Nil(t, err)
	stateDriver, ok := state.(*stateDriver)
	require.True(t, ok)

	// set every field in driverSpec
	driverSpec := &nvidiav1alpha1.NVIDIADriverSpec{
		Manager: nvidiav1alpha1.DriverManagerSpec{
			Repository:       "/path/to/repository",
			Image:            "image",
			Version:          "version",
			ImagePullPolicy:  "Always",
			ImagePullSecrets: []string{"manager-secret"},
			Env: []nvidiav1alpha1.EnvVar{
				{Name: "FOO", Value: "foo"},
				{Name: "BAR", Value: "bar"},
			},
		},
		StartupProbe:     getDefaultContainerProbeSpec(),
		LivenessProbe:    getDefaultContainerProbeSpec(),
		ReadinessProbe:   getDefaultContainerProbeSpec(),
		UsePrecompiled:   new(bool),
		ImagePullPolicy:  "Always",
		ImagePullSecrets: []string{"secret-a", "secret-b"},
		Resources: &nvidiav1alpha1.ResourceRequirements{
			Requests: corev1.ResourceList{
				corev1.ResourceCPU:    resource.MustParse("200m"),
				corev1.ResourceMemory: resource.MustParse("100Mi"),
			},
			Limits: corev1.ResourceList{
				"memory": resource.MustParse("200Mi"),
				"cpu":    resource.MustParse("500m"),
			},
		},
		Args: []string{"--foo", "--bar"},
		Env: []nvidiav1alpha1.EnvVar{
			{Name: "FOO", Value: "foo"},
			{Name: "BAR", Value: "bar"},
		},
		NodeSelector: map[string]string{
			"example.com/foo": "foo",
			"example.com/bar": "bar",
		},
		Labels: map[string]string{
			"custom-label-1": "custom-value-1",
			"custom-label-2": "custom-value-2",
			// The below standard labels should not be overridden in the
			// DaemonSet that gets rendered
			"app":                       "foo",
			"app.kubernetes.io/part-of": "foo",
		},
		Annotations: map[string]string{
			"custom-annotation-1": "custom-value-1",
			"custom-annotation-2": "custom-value-2",