Add E2E tests

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

.gitignore

@@ -6,3 +6,4 @@
 .idea
 [._]*.sw[a-p]
 coverage.out
+bin

Makefile

@@ -218,3 +218,9 @@ PHONY: .shell
 		-w /work \
 		--user $$(id -u):$$(id -g) \
 		$(BUILDIMAGE)
+
+GINKGO = $(CURDIR)/bin/ginkgo
+GINKGO_VERSION ?= $(shell cd $(CURDIR)/tests && go list -m -f '{{.Version}}' github.com/onsi/ginkgo/v2)
+.PHONY: ginkgo
+ginkgo: ## Download ginkgo locally if necessary.
+	@GOBIN=$(CURDIR)/bin GO111MODULE=on go install github.com/onsi/ginkgo/v2/ginkgo@$(GINKGO_VERSION)

hack/e2e-test.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+
+# Copyright 2025 The Kubernetes Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -o errexit
+set -o nounset
+set -o pipefail
+
+SOURCE_DIR="$(cd "$(dirname -- "${BASH_SOURCE[0]}")" && pwd -P)"
+ROOT_DIR="$SOURCE_DIR/.."
+
+GINKGO="$ROOT_DIR"/bin/ginkgo
+GINKGO_ARGS=${GINKGO_ARGS:-}
+
+# Check if ginkgo binary is present 
+if [ ! -f "$GINKGO" ]; then
+  echo "ginkgo binary not found at $GINKGO, building it"
+  make ginkgo
+fi
+
+# shellcheck disable=SC2086
+$GINKGO $GINKGO_ARGS -v --json-report ginkgo.json ./tests/e2e/... -- --config $ROOT_DIR/tests/e2e/config.yaml

tests/e2e/cleanup_test.go

@@ -0,0 +1,72 @@
+/*
+ * SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package e2e
+
+import (
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	. "github.com/onsi/gomega"
+)
+
+func cleanup() {
+	deployed, err := helmClient.ListDeployedReleases()
+	Expect(err).NotTo(HaveOccurred())
+
+	for _, release := range deployed {
+		switch release.Name {
+		case helmReleaseName:
+			err := helmClient.UninstallReleaseByName(helmReleaseName)
+			Expect(err).NotTo(HaveOccurred())
+		case "gfd":
+			// We check if GFD installation was required, if so, we uninstall it
+			// in cases where it was skipped (ENABLE_GFD=false) we don't need to uninstall it
+			if EnableGFD {
+				err := helmClient.UninstallReleaseByName("gfd")
+				Expect(err).NotTo(HaveOccurred())
+			}
+		default:
+			continue
+		}
+	}
+
+	// list all resourceclaimtemplates.resource.k8s.io on the testnamespace
+	resourceClaimTemplates, err := clientSet.ResourceV1beta1().ResourceClaimTemplates(testNamespace.Name).List(ctx, metav1.ListOptions{})
+	Expect(err).NotTo(HaveOccurred())
+
+	// delete all resourceclaimtemplates.resource.k8s.io
+	for _, resourceClaimTemplate := range resourceClaimTemplates.Items {
+		err := clientSet.ResourceV1beta1().ResourceClaimTemplates(testNamespace.Name).Delete(ctx, resourceClaimTemplate.Name, metav1.DeleteOptions{})
+		Expect(err).NotTo(HaveOccurred())
+	}
+}
+
+func cleanupCRDs() {
+	crds := []string{
+		"computedomains.resource.nvidia.com",
+	}
+
+	if EnableGFD {
+		crds = append(crds, "nodefeatures.nfd.k8s-sigs.io", "nodefeaturerules.nfd.k8s-sigs.io")
+	}
+
+	// Delete CRDs
+	for _, crd := range crds {
+		err := extClient.ApiextensionsV1().CustomResourceDefinitions().Delete(ctx, crd, metav1.DeleteOptions{})
+		Expect(err).NotTo(HaveOccurred())
+	}
+}

tests/e2e/data/test-pod1.yaml

@@ -0,0 +1,58 @@
+---
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: gpu-test1
+
+---
+apiVersion: resource.k8s.io/v1beta1
+kind: ResourceClaimTemplate
+metadata:
+  namespace: gpu-test1
+  name: single-gpu
+spec:
+  spec:
+    devices:
+      requests:
+      - name: gpu
+        deviceClassName: gpu.nvidia.com
+
+---
+apiVersion: resource.nvidia.com/v1beta1
+kind: ComputeDomain
+metadata:
+  namespace: gpu-test1
+  name: imex-channel-injection
+spec:
+  numNodes: 1
+  channel:
+    resourceClaimTemplate:
+      name: imex-channel-0
+
+---
+apiVersion: v1
+kind: Pod
+metadata:
+  namespace: gpu-test1
+  name: pod1
+  labels:
+    app: pod
+spec:
+  containers:
+  - name: ctr
+    image: ubuntu:22.04
+    command: ["bash", "-c"]
+    args: ["nvidia-smi -L; trap 'exit 0' TERM; sleep 9999 & wait"]
+    resources:
+      claims:
+      - name: gpu
+      - name: imex-channel-0
+  resourceClaims:
+  - name: gpu
+    resourceClaimTemplateName: single-gpu
+  - name: imex-channel-0
+    resourceClaimTemplateName: imex-channel-0
+  tolerations:
+  - key: "nvidia.com/gpu"
+    operator: "Exists"
+    effect: "NoSchedule"