github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-02-22 07:03:28 +00:00
Code Issues Projects Releases Wiki Activity

test(e2e/dra): add test for pod requesting allocated and new claims

Browse Source
This commit is contained in:
Piotr Rogowski
2026-01-06 20:54:12 +00:00
parent ddfb1c1e53
commit 6af7de8d1c
1 changed files with 49 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
test/e2e/dra/dra.go
View File

@@ -942,6 +942,54 @@ var _ = framework.SIGDescribe("node")(framework.WithLabel("DRA"), func() {
}
}
singleNodeMultipleClaimsTests := func() {
nodes := drautils.NewNodes(f, 1, 1)
// Allow allocating more than one device so that multiple claims can be prepared on the same node.
maxAllocations := 2
driver := drautils.NewDriver(f, nodes, drautils.DriverResources(maxAllocations)) // All tests get their own driver instance.
driver.WithKubelet = true
b := drautils.NewBuilder(f, driver)
// https://github.com/kubernetes/kubernetes/issues/135901 was fixed in master for Kubernetes 1.35 and not backported
// so this test only passes for kubelet >= 1.35.
f.It("requests an already allocated and a new claim for a pod", f.WithLabel("KubeletMinVersion:1.35"), func(ctx context.Context) {
// This test covers a situation when a pod references a mix of already-prepared and new claims.
tCtx := f.TContext(ctx)
firstClaim := b.ExternalClaim()
secondClaim := b.ExternalClaim()
b.Create(tCtx, firstClaim, secondClaim)
// First pod uses only firstClaim
firstPod := b.PodExternal()
b.Create(tCtx, firstPod)
b.TestPod(tCtx, firstPod)
// Second pod uses firstClaim (already prepared) + secondClaim (new)
secondPod := b.PodExternal()
secondPod.Spec.ResourceClaims = []v1.PodResourceClaim{
{
Name: "first",
ResourceClaimName: &firstClaim.Name,
},
{
Name: "second",
ResourceClaimName: &secondClaim.Name,
},
}
secondPod.Spec.Containers[0].Resources.Claims = []v1.ResourceClaim{
{Name: "first"},
{Name: "second"},
}
b.Create(tCtx, secondPod)
b.TestPod(tCtx, secondPod)
})
}
// The following tests only make sense when there is more than one node.
// They get skipped when there's only one node.
multiNodeTests := func(withKubelet bool) {
@@ -1806,6 +1854,7 @@ var _ = framework.SIGDescribe("node")(framework.WithLabel("DRA"), func() {
framework.Context("control plane", framework.WithLabel("ConformanceCandidate") /* TODO: replace with framework.WithConformance() */, func() { singleNodeTests(false) })
framework.Context("kubelet", feature.DynamicResourceAllocation, "on single node", func() { singleNodeTests(true) })
framework.Context("kubelet", feature.DynamicResourceAllocation, "on single node with multiple claims allocation", singleNodeMultipleClaimsTests)
framework.Context("control plane", framework.WithLabel("ConformanceCandidate") /* TODO: replace with framework.WithConformance() */, func() { multiNodeTests(false) })
framework.Context("kubelet", feature.DynamicResourceAllocation, "on multiple nodes", func() { multiNodeTests(true) })