github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-23 19:56:01 +00:00
Code Issues Projects Releases Wiki Activity

KEP-4540: added e2e tests for strict-cpu-reservation option

Browse Source
This commit is contained in:
Mark Sasnal 2025-02-17 19:45:13 -05:00
parent 269bbac6e8
commit 5625483527
2 changed files with 102 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

110
test/e2e_node/cpu_manager_test.go
View File

@ -327,7 +327,7 @@ func configureCPUManagerInKubelet(oldCfg *kubeletconfig.KubeletConfiguration, ku
return newCfg
}
func runGuPodTest(ctx context.Context, f *framework.Framework, cpuCount int) {
func runGuPodTest(ctx context.Context, f *framework.Framework, cpuCount int, strictReservedCPUs cpuset.CPUSet) {
var pod *v1.Pod
ctnAttrs := []ctnAttribute{
@ -353,6 +353,7 @@ func runGuPodTest(ctx context.Context, f *framework.Framework, cpuCount int) {
framework.ExpectNoError(err, "parsing cpuset from logs for [%s] of pod [%s]", cnt.Name, pod.Name)
gomega.Expect(cpus.Size()).To(gomega.Equal(cpuCount), "expected cpu set size == %d, got %q", cpuCount, cpus.String())
gomega.Expect(cpus.Intersection(strictReservedCPUs).IsEmpty()).To(gomega.BeTrueBecause("cpuset %q should not contain strict reserved cpus %q", cpus.String(), strictReservedCPUs.String()))
}
ginkgo.By("by deleting the pods and waiting for container removal")
@ -360,7 +361,7 @@ func runGuPodTest(ctx context.Context, f *framework.Framework, cpuCount int) {
waitForAllContainerRemoval(ctx, pod.Name, pod.Namespace)
}
func runNonGuPodTest(ctx context.Context, f *framework.Framework, cpuCap int64) {
func runNonGuPodTest(ctx context.Context, f *framework.Framework, cpuCap int64, strictReservedCPUs cpuset.CPUSet) {
var ctnAttrs []ctnAttribute
var err error
var pod *v1.Pod
@ -377,11 +378,10 @@ func runNonGuPodTest(ctx context.Context, f *framework.Framework, cpuCap int64)
pod = e2epod.NewPodClient(f).CreateSync(ctx, pod)
ginkgo.By("checking if the expected cpuset was assigned")
expAllowedCPUsListRegex = fmt.Sprintf("^0-%d\n$", cpuCap-1)
// on the single CPU node the only possible value is 0
if cpuCap == 1 {
expAllowedCPUsListRegex = "^0\n$"
}
expAllowedCPUs, err := cpuset.Parse(fmt.Sprintf("0-%d", cpuCap-1))
framework.ExpectNoError(err)
expAllowedCPUs = expAllowedCPUs.Difference(strictReservedCPUs)
expAllowedCPUsListRegex = fmt.Sprintf("^%s\n$", expAllowedCPUs.String())
err = e2epod.NewPodClient(f).MatchContainerOutput(ctx, pod.Name, pod.Spec.Containers[0].Name, expAllowedCPUsListRegex)
framework.ExpectNoError(err, "expected log not found in container [%s] of pod [%s]",
pod.Spec.Containers[0].Name, pod.Name)
@ -890,10 +890,10 @@ func runCPUManagerTests(f *framework.Framework) {
updateKubeletConfig(ctx, f, newCfg, true)
ginkgo.By("running a non-Gu pod")
runNonGuPodTest(ctx, f, cpuCap)
runNonGuPodTest(ctx, f, cpuCap, cpuset.New())
ginkgo.By("running a Gu pod")
runGuPodTest(ctx, f, 1)
runGuPodTest(ctx, f, 1, cpuset.New())
ginkgo.By("running multiple Gu and non-Gu pods")
runMultipleGuNonGuPods(ctx, f, cpuCap, cpuAlloc)
@ -916,6 +916,56 @@ func runCPUManagerTests(f *framework.Framework) {
runAutomaticallyRemoveInactivePodsFromCPUManagerStateFile(ctx, f)
})
ginkgo.It("reservedSystemCPUs are excluded only for Gu pods (strict-cpu-reservation option not enabled by default)", func(ctx context.Context) {
cpuCap, cpuAlloc, _ = getLocalNodeCPUDetails(ctx, f)
// Skip CPU Manager tests altogether if the CPU capacity < 2.
if cpuCap < 2 {
e2eskipper.Skipf("Skipping CPU Manager tests since the CPU capacity < 2")
}
reservedSystemCPUs := cpuset.New(0)
newCfg := configureCPUManagerInKubelet(oldCfg, &cpuManagerKubeletArguments{
policyName: string(cpumanager.PolicyStatic),
reservedSystemCPUs: reservedSystemCPUs,
})
updateKubeletConfig(ctx, f, newCfg, true)
ginkgo.By("running a Gu pod - it shouldn't use reserved system CPUs")
runGuPodTest(ctx, f, 1, reservedSystemCPUs)
ginkgo.By("running a non-Gu pod - it can use reserved system CPUs")
runNonGuPodTest(ctx, f, cpuCap, cpuset.New())
})
ginkgo.It("reservedSystemCPUs are excluded for both Gu and non-Gu pods (strict-cpu-reservation option enabled)", func(ctx context.Context) {
cpuCap, cpuAlloc, _ = getLocalNodeCPUDetails(ctx, f)
// Skip CPU Manager tests altogether if the CPU capacity < 2.
if cpuCap < 2 {
e2eskipper.Skipf("Skipping CPU Manager tests since the CPU capacity < 2")
}
reservedSystemCPUs := cpuset.New(0)
cpuPolicyOptions := map[string]string{
cpumanager.StrictCPUReservationOption: "true",
}
newCfg := configureCPUManagerInKubelet(oldCfg, &cpuManagerKubeletArguments{
policyName: string(cpumanager.PolicyStatic),
reservedSystemCPUs: reservedSystemCPUs,
enableCPUManagerOptions: true,
options: cpuPolicyOptions,
})
updateKubeletConfig(ctx, f, newCfg, true)
ginkgo.By("running a Gu pod - it shouldn't use reserved system CPUs")
runGuPodTest(ctx, f, 1, reservedSystemCPUs)
ginkgo.By("running a non-Gu pod - it shouldn't use reserved system CPUs with strict-cpu-reservation option enabled")
runNonGuPodTest(ctx, f, cpuCap, reservedSystemCPUs)
})
ginkgo.It("should assign CPUs as expected with enhanced policy based on strict SMT alignment", func(ctx context.Context) {
fullCPUsOnlyOpt := fmt.Sprintf("option=%s", cpumanager.FullPCPUsOnlyOption)
_, cpuAlloc, _ = getLocalNodeCPUDetails(ctx, f)
@ -950,7 +1000,44 @@ func runCPUManagerTests(f *framework.Framework) {
// the order between negative and positive doesn't really matter
runSMTAlignmentNegativeTests(ctx, f)
runSMTAlignmentPositiveTests(ctx, f, smtLevel)
runSMTAlignmentPositiveTests(ctx, f, smtLevel, cpuset.New())
})
ginkgo.It("should assign CPUs as expected based on strict SMT alignment, reservedSystemCPUs should be excluded (both strict-cpu-reservation and full-pcpus-only options enabled)", func(ctx context.Context) {
fullCPUsOnlyOpt := fmt.Sprintf("option=%s", cpumanager.FullPCPUsOnlyOption)
_, cpuAlloc, _ = getLocalNodeCPUDetails(ctx, f)
smtLevel := getSMTLevel()
// strict SMT alignment is trivially verified and granted on non-SMT systems
if smtLevel < 2 {
e2eskipper.Skipf("Skipping CPU Manager %s tests since SMT disabled", fullCPUsOnlyOpt)
}
// our tests want to allocate a full core, so we need at last smtLevel*2 virtual cpus
if cpuAlloc < int64(smtLevel*2) {
e2eskipper.Skipf("Skipping CPU Manager %s tests since the CPU capacity < %d", fullCPUsOnlyOpt, smtLevel*2)
}
framework.Logf("SMT level %d", smtLevel)
reservedSystemCPUs := cpuset.New(0)
cpuPolicyOptions := map[string]string{
cpumanager.FullPCPUsOnlyOption: "true",
cpumanager.StrictCPUReservationOption: "true",
}
newCfg := configureCPUManagerInKubelet(oldCfg,
&cpuManagerKubeletArguments{
policyName: string(cpumanager.PolicyStatic),
reservedSystemCPUs: reservedSystemCPUs,
enableCPUManagerOptions: true,
options: cpuPolicyOptions,
},
)
updateKubeletConfig(ctx, f, newCfg, true)
// the order between negative and positive doesn't really matter
runSMTAlignmentNegativeTests(ctx, f)
runSMTAlignmentPositiveTests(ctx, f, smtLevel, reservedSystemCPUs)
})
ginkgo.It("should not enforce CFS quota for containers with static CPUs assigned", func(ctx context.Context) {
@ -1108,7 +1195,7 @@ func runSMTAlignmentNegativeTests(ctx context.Context, f *framework.Framework) {
waitForAllContainerRemoval(ctx, pod.Name, pod.Namespace)
}
func runSMTAlignmentPositiveTests(ctx context.Context, f *framework.Framework, smtLevel int) {
func runSMTAlignmentPositiveTests(ctx context.Context, f *framework.Framework, smtLevel int, strictReservedCPUs cpuset.CPUSet) {
// positive test: try to run a container whose requests are a multiple of SMT level, check allocated cores
// 1. are core siblings
// 2. take a full core
@ -1134,6 +1221,7 @@ func runSMTAlignmentPositiveTests(ctx context.Context, f *framework.Framework, s
cpus, err := cpuset.Parse(strings.TrimSpace(logs))
framework.ExpectNoError(err, "parsing cpuset from logs for [%s] of pod [%s]", cnt.Name, pod.Name)
gomega.Expect(cpus.Intersection(strictReservedCPUs).IsEmpty()).To(gomega.BeTrueBecause("cpuset %q should not contain strict reserved cpus %q", cpus.String(), strictReservedCPUs.String()))
validateSMTAlignment(cpus, smtLevel, pod, &cnt)
}

5
test/e2e_node/topology_manager_test.go
View File

@ -19,6 +19,7 @@ package e2enode
import (
"context"
"fmt"
"k8s.io/utils/cpuset"
"os"
"os/exec"
"regexp"
@ -446,10 +447,10 @@ func runTopologyManagerPolicySuiteTests(ctx context.Context, f *framework.Framew
}
ginkgo.By("running a non-Gu pod")
runNonGuPodTest(ctx, f, cpuCap)
runNonGuPodTest(ctx, f, cpuCap, cpuset.New())
ginkgo.By("running a Gu pod")
runGuPodTest(ctx, f, 1)
runGuPodTest(ctx, f, 1, cpuset.New())
// Skip rest of the tests if CPU allocatable < 3.
if cpuAlloc < 3 {