Refactor e2e test infrastructure with unified helpers leveraging operator clientsets

[karthikvetrivel]

This PR is the second step for migrating GPU Operator e2e tests from bash to Go/Ginkgo.

Overview

This PR introduces a comprehensive set of helper utilities that leverage the GPU operator's generated Go clientsets (api/versioned).

Note: This PR contains only the helper function infrastructure. Actual test implementations using these helpers will be added in a follow-up PR.

Changes

1. Helper Infrastructure (tests/e2e/helpers/)

Created comprehensive helper utilities:

• operator.go - Helm client for deploying GPU Operator
• clusterpolicy.go - ClusterPolicy CRUD + component toggles (DCGM, GFD, etc.)
• nvidiadriver.go - NVIDIADriver CR operations (cluster-scoped)
• daemonset.go - DaemonSet queries and readiness checks
• node.go - Node labeling operations
• workload.go - GPU workload deployment and verification
• pod.go - Pod operations and namespace management

Structure:

• Flattened directory: moved operator/helm.go → operator.go, kubernetes/pod.go → pod.go
• Updated imports in gpu_operator_test.go

2. Constants (tests/e2e/constants.go)

Added shared constants:

• DefaultPollingInterval - Standard 5s interval for wait operations
• UpgradeDoneState - Driver upgrade completion state

3. Updated Existing Tests

Modified gpu_operator_test.go to use new helper package structure:

• operator.Client → helpers.OperatorClient
• k8stest.Client → helpers.PodClient
• Improved error handling for pod log retrieval

Next Steps

• Step 3: Migrate bash test scenarios to Go using these helpers
• Step 4: Update CI/CD pipelines to run Go-based e2e tests

[rajathagasthya]

LGTM. Thanks for addressing the comments!

[karthikvetrivel]

Moving this to draft until I add tests to keep the github.com/NVIDIA/gpu-operator dep (by referencing it in the e2e/tests code).

[rajathagasthya]

@karthikvetrivel Could you update helm.sh/helm/v3 dependency to address Dependabot alerts in https://github.com/NVIDIA/gpu-operator/security/dependabot? go get helm.sh/helm/v3@latest should work.

[karthikvetrivel]

@rajathagasthya This would require updating the main module's Kubernetes dependencies from v0.33.2 to v0.34.0. Should we still do it?

[rajathagasthya]

@tariq1890 updated it recently in #1805, so we should be able to contain these changes to just the e2e module.

[karthikvetrivel]

@rajathagasthya Fixed and amended. Thanks!

[coderabbitai]

Important

Review skipped

Auto reviews are disabled on this repository.

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

✨ Finishing touches

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Post copyable unit tests in a comment

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}