Function TestCheckNodeSelector

internal/validator/validator_test.go:96–130

(t *testing.T)

}

func TestCheckNodeSelector(t *testing.T) {
	node := makeTestNode(labelled(map[string]string{"os-version": "ubuntu20.04"}))
	driver := makeTestDriver(nodeSelector(node.Labels))
	conflictingDriver := makeTestDriver(named("conflictingDriver"), nodeSelector(node.Labels))
	nonconflictingDriver := makeTestDriver(named("nonconflictingDriver"))

	tests := []struct {
		node            *corev1.Node
		existingDriver  *nvidiav1alpha1.NVIDIADriver
		requestedDriver *nvidiav1alpha1.NVIDIADriver
		shouldError     bool
	}{
		{node: node, existingDriver: driver, requestedDriver: conflictingDriver, shouldError: true},
		{node: node, existingDriver: driver, requestedDriver: nonconflictingDriver, shouldError: false},
	}

	for _, tc := range tests {
		s := scheme.Scheme
		err := nvidiav1alpha1.AddToScheme(s)
		require.NoError(t, err)
		c := fake.
			NewClientBuilder().
			WithScheme(s).
			WithObjects(tc.node, tc.existingDriver, tc.requestedDriver).
			Build()
		nsv := NewNodeSelectorValidator(c)

		err = nsv.Validate(context.Background(), tc.requestedDriver)
		if tc.shouldError {
			assert.Error(t, err)
		} else {
			assert.NoError(t, err)
		}
	}
}