mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-22 03:11:40 +00:00
e2e: node: add tests for GetAllocatableResources
Add e2e tests for the new GetAllocatableResources API. The tests are added in the `podresources_test` suite created previously in this series. Signed-off-by: Francesco Romani <fromani@redhat.com>
This commit is contained in:
parent
8afdf4f146
commit
9c69db3f04
@ -75,6 +75,15 @@ func NewCPUSet(cpus ...int) CPUSet {
|
|||||||
return b.Result()
|
return b.Result()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// NewCPUSet returns a new CPUSet containing the supplied elements, as slice of int64.
|
||||||
|
func NewCPUSetInt64(cpus ...int64) CPUSet {
|
||||||
|
b := NewBuilder()
|
||||||
|
for _, c := range cpus {
|
||||||
|
b.Add(int(c))
|
||||||
|
}
|
||||||
|
return b.Result()
|
||||||
|
}
|
||||||
|
|
||||||
// Size returns the number of elements in this set.
|
// Size returns the number of elements in this set.
|
||||||
func (s CPUSet) Size() int {
|
func (s CPUSet) Size() int {
|
||||||
return len(s.elems)
|
return len(s.elems)
|
||||||
|
@ -210,6 +210,10 @@ func disableCPUManagerInKubelet(f *framework.Framework) (oldCfg *kubeletconfig.K
|
|||||||
}
|
}
|
||||||
|
|
||||||
func enableCPUManagerInKubelet(f *framework.Framework, cleanStateFile bool) (oldCfg *kubeletconfig.KubeletConfiguration) {
|
func enableCPUManagerInKubelet(f *framework.Framework, cleanStateFile bool) (oldCfg *kubeletconfig.KubeletConfiguration) {
|
||||||
|
return configureCPUManagerInKubelet(f, cleanStateFile, cpuset.CPUSet{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func configureCPUManagerInKubelet(f *framework.Framework, cleanStateFile bool, reservedSystemCPUs cpuset.CPUSet) (oldCfg *kubeletconfig.KubeletConfiguration) {
|
||||||
// Enable CPU Manager in Kubelet with static policy.
|
// Enable CPU Manager in Kubelet with static policy.
|
||||||
oldCfg, err := getCurrentKubeletConfig()
|
oldCfg, err := getCurrentKubeletConfig()
|
||||||
framework.ExpectNoError(err)
|
framework.ExpectNoError(err)
|
||||||
@ -239,15 +243,21 @@ func enableCPUManagerInKubelet(f *framework.Framework, cleanStateFile bool) (old
|
|||||||
// Set the CPU Manager reconcile period to 1 second.
|
// Set the CPU Manager reconcile period to 1 second.
|
||||||
newCfg.CPUManagerReconcilePeriod = metav1.Duration{Duration: 1 * time.Second}
|
newCfg.CPUManagerReconcilePeriod = metav1.Duration{Duration: 1 * time.Second}
|
||||||
|
|
||||||
// The Kubelet panics if either kube-reserved or system-reserved is not set
|
if reservedSystemCPUs.Size() > 0 {
|
||||||
// when CPU Manager is enabled. Set cpu in kube-reserved > 0 so that
|
cpus := reservedSystemCPUs.String()
|
||||||
// kubelet doesn't panic.
|
framework.Logf("configureCPUManagerInKubelet: using reservedSystemCPUs=%q", cpus)
|
||||||
if newCfg.KubeReserved == nil {
|
newCfg.ReservedSystemCPUs = cpus
|
||||||
newCfg.KubeReserved = map[string]string{}
|
} else {
|
||||||
}
|
// The Kubelet panics if either kube-reserved or system-reserved is not set
|
||||||
|
// when CPU Manager is enabled. Set cpu in kube-reserved > 0 so that
|
||||||
|
// kubelet doesn't panic.
|
||||||
|
if newCfg.KubeReserved == nil {
|
||||||
|
newCfg.KubeReserved = map[string]string{}
|
||||||
|
}
|
||||||
|
|
||||||
if _, ok := newCfg.KubeReserved["cpu"]; !ok {
|
if _, ok := newCfg.KubeReserved["cpu"]; !ok {
|
||||||
newCfg.KubeReserved["cpu"] = "200m"
|
newCfg.KubeReserved["cpu"] = "200m"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
// Update the Kubelet configuration.
|
// Update the Kubelet configuration.
|
||||||
framework.ExpectNoError(setKubeletConfiguration(f, newCfg))
|
framework.ExpectNoError(setKubeletConfiguration(f, newCfg))
|
||||||
|
@ -19,6 +19,7 @@ package e2enode
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -27,6 +28,7 @@ import (
|
|||||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
kubeletpodresourcesv1 "k8s.io/kubelet/pkg/apis/podresources/v1"
|
kubeletpodresourcesv1 "k8s.io/kubelet/pkg/apis/podresources/v1"
|
||||||
"k8s.io/kubernetes/pkg/kubelet/apis/podresources"
|
"k8s.io/kubernetes/pkg/kubelet/apis/podresources"
|
||||||
|
"k8s.io/kubernetes/pkg/kubelet/cm/cpuset"
|
||||||
"k8s.io/kubernetes/pkg/kubelet/util"
|
"k8s.io/kubernetes/pkg/kubelet/util"
|
||||||
|
|
||||||
"k8s.io/kubernetes/test/e2e/framework"
|
"k8s.io/kubernetes/test/e2e/framework"
|
||||||
@ -36,9 +38,17 @@ import (
|
|||||||
"github.com/onsi/gomega"
|
"github.com/onsi/gomega"
|
||||||
)
|
)
|
||||||
|
|
||||||
func makePodResourcesTestPod(podName, cntName, devName, devCount string) *v1.Pod {
|
type podDesc struct {
|
||||||
|
podName string
|
||||||
|
cntName string
|
||||||
|
resourceName string
|
||||||
|
resourceAmount int
|
||||||
|
cpuCount int
|
||||||
|
}
|
||||||
|
|
||||||
|
func makePodResourcesTestPod(desc podDesc) *v1.Pod {
|
||||||
cnt := v1.Container{
|
cnt := v1.Container{
|
||||||
Name: cntName,
|
Name: desc.cntName,
|
||||||
Image: busyboxImage,
|
Image: busyboxImage,
|
||||||
Resources: v1.ResourceRequirements{
|
Resources: v1.ResourceRequirements{
|
||||||
Requests: v1.ResourceList{},
|
Requests: v1.ResourceList{},
|
||||||
@ -46,13 +56,20 @@ func makePodResourcesTestPod(podName, cntName, devName, devCount string) *v1.Pod
|
|||||||
},
|
},
|
||||||
Command: []string{"sh", "-c", "sleep 1d"},
|
Command: []string{"sh", "-c", "sleep 1d"},
|
||||||
}
|
}
|
||||||
if devName != "" && devCount != "" {
|
if desc.cpuCount > 0 {
|
||||||
cnt.Resources.Requests[v1.ResourceName(devName)] = resource.MustParse(devCount)
|
cnt.Resources.Requests[v1.ResourceCPU] = resource.MustParse(fmt.Sprintf("%d", desc.cpuCount))
|
||||||
cnt.Resources.Limits[v1.ResourceName(devName)] = resource.MustParse(devCount)
|
cnt.Resources.Limits[v1.ResourceCPU] = resource.MustParse(fmt.Sprintf("%d", desc.cpuCount))
|
||||||
|
// we don't really care, we only need to be in guaranteed QoS
|
||||||
|
cnt.Resources.Requests[v1.ResourceMemory] = resource.MustParse("100Mi")
|
||||||
|
cnt.Resources.Limits[v1.ResourceMemory] = resource.MustParse("100Mi")
|
||||||
|
}
|
||||||
|
if desc.resourceName != "" && desc.resourceAmount > 0 {
|
||||||
|
cnt.Resources.Requests[v1.ResourceName(desc.resourceName)] = resource.MustParse(fmt.Sprintf("%d", desc.resourceAmount))
|
||||||
|
cnt.Resources.Limits[v1.ResourceName(desc.resourceName)] = resource.MustParse(fmt.Sprintf("%d", desc.resourceAmount))
|
||||||
}
|
}
|
||||||
return &v1.Pod{
|
return &v1.Pod{
|
||||||
ObjectMeta: metav1.ObjectMeta{
|
ObjectMeta: metav1.ObjectMeta{
|
||||||
Name: podName,
|
Name: desc.podName,
|
||||||
},
|
},
|
||||||
Spec: v1.PodSpec{
|
Spec: v1.PodSpec{
|
||||||
RestartPolicy: v1.RestartPolicyNever,
|
RestartPolicy: v1.RestartPolicyNever,
|
||||||
@ -63,42 +80,44 @@ func makePodResourcesTestPod(podName, cntName, devName, devCount string) *v1.Pod
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func countPodResources(podIdx int, pr *kubeletpodresourcesv1.PodResources) int {
|
func logPodResources(podIdx int, pr *kubeletpodresourcesv1.PodResources) {
|
||||||
ns := pr.GetNamespace()
|
ns := pr.GetNamespace()
|
||||||
devCount := 0
|
cnts := pr.GetContainers()
|
||||||
for cntIdx, cnt := range pr.GetContainers() {
|
if len(cnts) == 0 {
|
||||||
if len(cnt.Devices) > 0 {
|
framework.Logf("#%02d/%02d/%02d - %s/%s/%s No containers", podIdx, 0, 0, ns, pr.GetName(), "_")
|
||||||
for devIdx, dev := range cnt.Devices {
|
return
|
||||||
framework.Logf("#%02d/%02d/%02d - %s/%s/%s %s -> %s", podIdx, cntIdx, devIdx, ns, pr.GetName(), cnt.Name, dev.ResourceName, strings.Join(dev.DeviceIds, ", "))
|
}
|
||||||
devCount++
|
|
||||||
}
|
for cntIdx, cnt := range cnts {
|
||||||
} else {
|
if len(cnt.Devices) == 0 {
|
||||||
framework.Logf("#%02d/%02d/%02d - %s/%s/%s No resources", podIdx, cntIdx, 0, ns, pr.GetName(), cnt.Name)
|
framework.Logf("#%02d/%02d/%02d - %s/%s/%s cpus -> %v resources -> none", podIdx, cntIdx, 0, ns, pr.GetName(), cnt.Name, cnt.CpuIds)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for devIdx, dev := range cnt.Devices {
|
||||||
|
framework.Logf("#%02d/%02d/%02d - %s/%s/%s cpus -> %v %s -> %s", podIdx, cntIdx, devIdx, ns, pr.GetName(), cnt.Name, cnt.CpuIds, dev.ResourceName, strings.Join(dev.DeviceIds, ", "))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return devCount
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func getPodResources(cli kubeletpodresourcesv1.PodResourcesListerClient) ([]*kubeletpodresourcesv1.PodResources, []*kubeletpodresourcesv1.PodResources) {
|
type podResMap map[string]map[string]kubeletpodresourcesv1.ContainerResources
|
||||||
|
|
||||||
|
func getPodResources(cli kubeletpodresourcesv1.PodResourcesListerClient) podResMap {
|
||||||
resp, err := cli.List(context.TODO(), &kubeletpodresourcesv1.ListPodResourcesRequest{})
|
resp, err := cli.List(context.TODO(), &kubeletpodresourcesv1.ListPodResourcesRequest{})
|
||||||
framework.ExpectNoError(err)
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
res := []*kubeletpodresourcesv1.PodResources{}
|
res := make(map[string]map[string]kubeletpodresourcesv1.ContainerResources)
|
||||||
noRes := []*kubeletpodresourcesv1.PodResources{}
|
|
||||||
for idx, podResource := range resp.GetPodResources() {
|
for idx, podResource := range resp.GetPodResources() {
|
||||||
if countPodResources(idx, podResource) > 0 {
|
// to make troubleshooting easier
|
||||||
res = append(res, podResource)
|
logPodResources(idx, podResource)
|
||||||
} else {
|
|
||||||
noRes = append(noRes, podResource)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res, noRes
|
|
||||||
}
|
|
||||||
|
|
||||||
type podDesc struct {
|
cnts := make(map[string]kubeletpodresourcesv1.ContainerResources)
|
||||||
podName string
|
for _, cnt := range podResource.GetContainers() {
|
||||||
resourceName string
|
cnts[cnt.GetName()] = *cnt
|
||||||
resourceAmount string
|
}
|
||||||
|
res[podResource.GetName()] = cnts
|
||||||
|
}
|
||||||
|
return res
|
||||||
}
|
}
|
||||||
|
|
||||||
type testPodData struct {
|
type testPodData struct {
|
||||||
@ -113,7 +132,7 @@ func newTestPodData() *testPodData {
|
|||||||
|
|
||||||
func (tpd *testPodData) createPodsForTest(f *framework.Framework, podReqs []podDesc) {
|
func (tpd *testPodData) createPodsForTest(f *framework.Framework, podReqs []podDesc) {
|
||||||
for _, podReq := range podReqs {
|
for _, podReq := range podReqs {
|
||||||
pod := makePodResourcesTestPod(podReq.podName, "cnt-0", podReq.resourceName, podReq.resourceAmount)
|
pod := makePodResourcesTestPod(podReq)
|
||||||
pod = f.PodClient().CreateSync(pod)
|
pod = f.PodClient().CreateSync(pod)
|
||||||
|
|
||||||
framework.Logf("created pod %s", podReq.podName)
|
framework.Logf("created pod %s", podReq.podName)
|
||||||
@ -136,137 +155,376 @@ func (tpd *testPodData) deletePod(f *framework.Framework, podName string) {
|
|||||||
delete(tpd.PodMap, podName)
|
delete(tpd.PodMap, podName)
|
||||||
}
|
}
|
||||||
|
|
||||||
func expectPodResources(cli kubeletpodresourcesv1.PodResourcesListerClient, expectedPodsWithResources, expectedPodsWithoutResources int) {
|
func findContainerDeviceByName(devs []*kubeletpodresourcesv1.ContainerDevices, resourceName string) *kubeletpodresourcesv1.ContainerDevices {
|
||||||
gomega.EventuallyWithOffset(1, func() error {
|
for _, dev := range devs {
|
||||||
podResources, noResources := getPodResources(cli)
|
if dev.ResourceName == resourceName {
|
||||||
if len(podResources) != expectedPodsWithResources {
|
return dev
|
||||||
return fmt.Errorf("pod with resources: expected %d found %d", expectedPodsWithResources, len(podResources))
|
|
||||||
}
|
}
|
||||||
if len(noResources) != expectedPodsWithoutResources {
|
}
|
||||||
return fmt.Errorf("pod WITHOUT resources: expected %d found %d", expectedPodsWithoutResources, len(noResources))
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func matchPodDescWithResources(expected []podDesc, found podResMap) error {
|
||||||
|
for _, podReq := range expected {
|
||||||
|
framework.Logf("matching: %#v", podReq)
|
||||||
|
|
||||||
|
podInfo, ok := found[podReq.podName]
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("no pod resources for pod %q", podReq.podName)
|
||||||
}
|
}
|
||||||
return nil
|
cntInfo, ok := podInfo[podReq.cntName]
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("no container resources for pod %q container %q", podReq.podName, podReq.cntName)
|
||||||
|
}
|
||||||
|
|
||||||
|
if podReq.cpuCount > 0 {
|
||||||
|
if len(cntInfo.CpuIds) != podReq.cpuCount {
|
||||||
|
return fmt.Errorf("pod %q container %q expected %d cpus got %v", podReq.podName, podReq.cntName, podReq.cpuCount, cntInfo.CpuIds)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if podReq.resourceName != "" && podReq.resourceAmount > 0 {
|
||||||
|
dev := findContainerDeviceByName(cntInfo.GetDevices(), podReq.resourceName)
|
||||||
|
if dev == nil {
|
||||||
|
return fmt.Errorf("pod %q container %q expected data for resource %q not found", podReq.podName, podReq.cntName, podReq.resourceName)
|
||||||
|
}
|
||||||
|
if len(dev.DeviceIds) != podReq.resourceAmount {
|
||||||
|
return fmt.Errorf("pod %q container %q resource %q expected %d items got %v", podReq.podName, podReq.cntName, podReq.resourceName, podReq.resourceAmount, dev.DeviceIds)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
devs := cntInfo.GetDevices()
|
||||||
|
if len(devs) > 0 {
|
||||||
|
return fmt.Errorf("pod %q container %q expected no resources, got %v", podReq.podName, podReq.cntName, devs)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectPodResources(offset int, cli kubeletpodresourcesv1.PodResourcesListerClient, expected []podDesc) {
|
||||||
|
gomega.EventuallyWithOffset(1+offset, func() error {
|
||||||
|
found := getPodResources(cli)
|
||||||
|
return matchPodDescWithResources(expected, found)
|
||||||
}, time.Minute, 10*time.Second).Should(gomega.BeNil())
|
}, time.Minute, 10*time.Second).Should(gomega.BeNil())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func filterOutDesc(descs []podDesc, name string) []podDesc {
|
||||||
|
var ret []podDesc
|
||||||
|
for _, desc := range descs {
|
||||||
|
if desc.podName == name {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
ret = append(ret, desc)
|
||||||
|
}
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
|
||||||
func podresourcesListTests(f *framework.Framework, cli kubeletpodresourcesv1.PodResourcesListerClient, sd *sriovData) {
|
func podresourcesListTests(f *framework.Framework, cli kubeletpodresourcesv1.PodResourcesListerClient, sd *sriovData) {
|
||||||
var podResources []*kubeletpodresourcesv1.PodResources
|
|
||||||
var noResources []*kubeletpodresourcesv1.PodResources
|
|
||||||
var tpd *testPodData
|
var tpd *testPodData
|
||||||
|
|
||||||
|
var found podResMap
|
||||||
|
var expected []podDesc
|
||||||
|
var extra podDesc
|
||||||
|
|
||||||
|
expectedBasePods := 0 /* nothing but pods we create */
|
||||||
|
if sd != nil {
|
||||||
|
expectedBasePods = 1 // sriovdp
|
||||||
|
}
|
||||||
|
|
||||||
ginkgo.By("checking the output when no pods are present")
|
ginkgo.By("checking the output when no pods are present")
|
||||||
expectPodResources(cli, 0, 1) // sriovdp
|
found = getPodResources(cli)
|
||||||
|
gomega.ExpectWithOffset(1, found).To(gomega.HaveLen(expectedBasePods), "base pod expectation mismatch")
|
||||||
|
|
||||||
tpd = newTestPodData()
|
tpd = newTestPodData()
|
||||||
ginkgo.By("checking the output when only pods which don't require resources are present")
|
ginkgo.By("checking the output when only pods which don't require resources are present")
|
||||||
tpd.createPodsForTest(f, []podDesc{
|
expected = []podDesc{
|
||||||
{
|
{
|
||||||
podName: "pod-00",
|
podName: "pod-00",
|
||||||
|
cntName: "cnt-00",
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
podName: "pod-01",
|
podName: "pod-01",
|
||||||
|
cntName: "cnt-00",
|
||||||
},
|
},
|
||||||
})
|
}
|
||||||
expectPodResources(cli, 0, 2+1) // test pods + sriovdp
|
tpd.createPodsForTest(f, expected)
|
||||||
|
expectPodResources(1, cli, expected)
|
||||||
tpd.deletePodsForTest(f)
|
tpd.deletePodsForTest(f)
|
||||||
|
|
||||||
tpd = newTestPodData()
|
tpd = newTestPodData()
|
||||||
ginkgo.By("checking the output when only a subset of pods require resources")
|
ginkgo.By("checking the output when only a subset of pods require resources")
|
||||||
tpd.createPodsForTest(f, []podDesc{
|
if sd != nil {
|
||||||
{
|
expected = []podDesc{
|
||||||
podName: "pod-00",
|
{
|
||||||
},
|
podName: "pod-00",
|
||||||
{
|
cntName: "cnt-00",
|
||||||
podName: "pod-01",
|
},
|
||||||
resourceName: sd.resourceName,
|
{
|
||||||
resourceAmount: "1",
|
podName: "pod-01",
|
||||||
},
|
cntName: "cnt-00",
|
||||||
{
|
resourceName: sd.resourceName,
|
||||||
podName: "pod-02",
|
resourceAmount: 1,
|
||||||
},
|
cpuCount: 2,
|
||||||
{
|
},
|
||||||
podName: "pod-03",
|
{
|
||||||
resourceName: sd.resourceName,
|
podName: "pod-02",
|
||||||
resourceAmount: "1",
|
cntName: "cnt-00",
|
||||||
},
|
cpuCount: 2,
|
||||||
})
|
},
|
||||||
expectPodResources(cli, 2, 2+1) // test pods + sriovdp
|
{
|
||||||
// TODO check for specific pods
|
podName: "pod-03",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
resourceName: sd.resourceName,
|
||||||
|
resourceAmount: 1,
|
||||||
|
cpuCount: 1,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
expected = []podDesc{
|
||||||
|
{
|
||||||
|
podName: "pod-00",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-01",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-02",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-03",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 1,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
tpd.createPodsForTest(f, expected)
|
||||||
|
expectPodResources(1, cli, expected)
|
||||||
tpd.deletePodsForTest(f)
|
tpd.deletePodsForTest(f)
|
||||||
|
|
||||||
tpd = newTestPodData()
|
tpd = newTestPodData()
|
||||||
ginkgo.By("checking the output when creating pods which require resources between calls")
|
ginkgo.By("checking the output when creating pods which require resources between calls")
|
||||||
tpd.createPodsForTest(f, []podDesc{
|
if sd != nil {
|
||||||
{
|
expected = []podDesc{
|
||||||
podName: "pod-00",
|
{
|
||||||
},
|
podName: "pod-00",
|
||||||
{
|
cntName: "cnt-00",
|
||||||
podName: "pod-01",
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-01",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
resourceName: sd.resourceName,
|
||||||
|
resourceAmount: 1,
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-02",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
expected = []podDesc{
|
||||||
|
{
|
||||||
|
podName: "pod-00",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-01",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-02",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tpd.createPodsForTest(f, expected)
|
||||||
|
expectPodResources(1, cli, expected)
|
||||||
|
|
||||||
|
if sd != nil {
|
||||||
|
extra = podDesc{
|
||||||
|
podName: "pod-03",
|
||||||
|
cntName: "cnt-00",
|
||||||
resourceName: sd.resourceName,
|
resourceName: sd.resourceName,
|
||||||
resourceAmount: "1",
|
resourceAmount: 1,
|
||||||
},
|
cpuCount: 1,
|
||||||
{
|
}
|
||||||
podName: "pod-02",
|
} else {
|
||||||
},
|
extra = podDesc{
|
||||||
})
|
podName: "pod-03",
|
||||||
podResources, noResources = getPodResources(cli)
|
cntName: "cnt-00",
|
||||||
framework.ExpectEqual(len(podResources), 1)
|
cpuCount: 1,
|
||||||
framework.ExpectEqual(len(noResources), 2+1) // test pods + sriovdp
|
}
|
||||||
// TODO check for specific pods
|
|
||||||
|
}
|
||||||
|
|
||||||
tpd.createPodsForTest(f, []podDesc{
|
tpd.createPodsForTest(f, []podDesc{
|
||||||
{
|
extra,
|
||||||
podName: "pod-03",
|
|
||||||
resourceName: sd.resourceName,
|
|
||||||
resourceAmount: "1",
|
|
||||||
},
|
|
||||||
})
|
})
|
||||||
podResources, noResources = getPodResources(cli)
|
|
||||||
framework.ExpectEqual(len(podResources), 2)
|
expected = append(expected, extra)
|
||||||
framework.ExpectEqual(len(noResources), 2+1) // test pods + sriovdp
|
expectPodResources(1, cli, expected)
|
||||||
// TODO check for specific pods
|
|
||||||
tpd.deletePodsForTest(f)
|
tpd.deletePodsForTest(f)
|
||||||
|
|
||||||
tpd = newTestPodData()
|
tpd = newTestPodData()
|
||||||
ginkgo.By("checking the output when deleting pods which require resources between calls")
|
ginkgo.By("checking the output when deleting pods which require resources between calls")
|
||||||
tpd.createPodsForTest(f, []podDesc{
|
|
||||||
{
|
if sd != nil {
|
||||||
podName: "pod-00",
|
expected = []podDesc{
|
||||||
},
|
{
|
||||||
{
|
podName: "pod-00",
|
||||||
podName: "pod-01",
|
cntName: "cnt-00",
|
||||||
resourceName: sd.resourceName,
|
cpuCount: 1,
|
||||||
resourceAmount: "1",
|
},
|
||||||
},
|
{
|
||||||
{
|
podName: "pod-01",
|
||||||
podName: "pod-02",
|
cntName: "cnt-00",
|
||||||
},
|
resourceName: sd.resourceName,
|
||||||
{
|
resourceAmount: 1,
|
||||||
podName: "pod-03",
|
cpuCount: 2,
|
||||||
resourceName: sd.resourceName,
|
},
|
||||||
resourceAmount: "1",
|
{
|
||||||
},
|
podName: "pod-02",
|
||||||
})
|
cntName: "cnt-00",
|
||||||
podResources, noResources = getPodResources(cli)
|
},
|
||||||
framework.ExpectEqual(len(podResources), 2)
|
{
|
||||||
framework.ExpectEqual(len(noResources), 2+1) // test pods + sriovdp
|
podName: "pod-03",
|
||||||
// TODO check for specific pods
|
cntName: "cnt-00",
|
||||||
|
resourceName: sd.resourceName,
|
||||||
|
resourceAmount: 1,
|
||||||
|
cpuCount: 1,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
expected = []podDesc{
|
||||||
|
{
|
||||||
|
podName: "pod-00",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 1,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-01",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-02",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
podName: "pod-03",
|
||||||
|
cntName: "cnt-00",
|
||||||
|
cpuCount: 1,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tpd.createPodsForTest(f, expected)
|
||||||
|
expectPodResources(1, cli, expected)
|
||||||
|
|
||||||
tpd.deletePod(f, "pod-01")
|
tpd.deletePod(f, "pod-01")
|
||||||
podResources, noResources = getPodResources(cli)
|
expectedPostDelete := filterOutDesc(expected, "pod-01")
|
||||||
framework.ExpectEqual(len(podResources), 1)
|
expectPodResources(1, cli, expectedPostDelete)
|
||||||
framework.ExpectEqual(len(noResources), 2+1) // test pods + sriovdp
|
|
||||||
// TODO check for specific pods
|
|
||||||
tpd.deletePodsForTest(f)
|
tpd.deletePodsForTest(f)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func podresourcesGetAllocatableResourcesTests(f *framework.Framework, cli kubeletpodresourcesv1.PodResourcesListerClient, sd *sriovData, onlineCPUs, reservedSystemCPUs cpuset.CPUSet) {
|
||||||
|
ginkgo.By("checking the devices known to the kubelet")
|
||||||
|
resp, err := cli.GetAllocatableResources(context.TODO(), &kubeletpodresourcesv1.AllocatableResourcesRequest{})
|
||||||
|
framework.ExpectNoErrorWithOffset(1, err)
|
||||||
|
devs := resp.GetDevices()
|
||||||
|
allocatableCPUs := cpuset.NewCPUSetInt64(resp.GetCpuIds()...)
|
||||||
|
|
||||||
|
if onlineCPUs.Size() == 0 {
|
||||||
|
ginkgo.By("expecting no CPUs reported")
|
||||||
|
gomega.ExpectWithOffset(1, onlineCPUs.Size()).To(gomega.Equal(reservedSystemCPUs.Size()), "with no online CPUs, no CPUs should be reserved")
|
||||||
|
} else {
|
||||||
|
ginkgo.By(fmt.Sprintf("expecting online CPUs reported - online=%v (%d) reserved=%v (%d)", onlineCPUs, onlineCPUs.Size(), reservedSystemCPUs, reservedSystemCPUs.Size()))
|
||||||
|
if reservedSystemCPUs.Size() > onlineCPUs.Size() {
|
||||||
|
ginkgo.Fail("more reserved CPUs than online")
|
||||||
|
}
|
||||||
|
expectedCPUs := onlineCPUs.Difference(reservedSystemCPUs)
|
||||||
|
|
||||||
|
ginkgo.By(fmt.Sprintf("expecting CPUs '%v'='%v'", allocatableCPUs, expectedCPUs))
|
||||||
|
gomega.ExpectWithOffset(1, allocatableCPUs.Equals(expectedCPUs)).To(gomega.BeTrue(), "mismatch expecting CPUs")
|
||||||
|
}
|
||||||
|
|
||||||
|
if sd == nil { // no devices in the environment, so expect no devices
|
||||||
|
ginkgo.By("expecting no devices reported")
|
||||||
|
gomega.ExpectWithOffset(1, devs).To(gomega.BeEmpty(), fmt.Sprintf("got unexpected devices %#v", devs))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ginkgo.By(fmt.Sprintf("expecting some %q devices reported", sd.resourceName))
|
||||||
|
gomega.ExpectWithOffset(1, devs).ToNot(gomega.BeEmpty())
|
||||||
|
for _, dev := range devs {
|
||||||
|
framework.ExpectEqual(dev.ResourceName, sd.resourceName)
|
||||||
|
gomega.ExpectWithOffset(1, dev.DeviceIds).ToNot(gomega.BeEmpty())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Serial because the test updates kubelet configuration.
|
// Serial because the test updates kubelet configuration.
|
||||||
var _ = SIGDescribe("POD Resources [Serial] [Feature:PODResources][NodeFeature:PODResources]", func() {
|
var _ = SIGDescribe("POD Resources [Serial] [Feature:PODResources][NodeFeature:PODResources]", func() {
|
||||||
f := framework.NewDefaultFramework("podresources-test")
|
f := framework.NewDefaultFramework("podresources-test")
|
||||||
|
|
||||||
|
reservedSystemCPUs := cpuset.MustParse("1")
|
||||||
|
|
||||||
ginkgo.Context("With SRIOV devices in the system", func() {
|
ginkgo.Context("With SRIOV devices in the system", func() {
|
||||||
ginkgo.It("should return the expected responses from List()", func() {
|
ginkgo.It("should return the expected responses with cpumanager static policy enabled", func() {
|
||||||
// this is a very rough check. We just want to rule out system that does NOT have any SRIOV device.
|
// this is a very rough check. We just want to rule out system that does NOT have enough resources
|
||||||
|
_, cpuAlloc, _ := getLocalNodeCPUDetails(f)
|
||||||
|
|
||||||
|
if cpuAlloc < minCoreCount {
|
||||||
|
e2eskipper.Skipf("Skipping CPU Manager tests since the CPU allocatable < %d", minCoreCount)
|
||||||
|
}
|
||||||
|
if sriovdevCount, err := countSRIOVDevices(); err != nil || sriovdevCount == 0 {
|
||||||
|
e2eskipper.Skipf("this test is meant to run on a system with at least one configured VF from SRIOV device")
|
||||||
|
}
|
||||||
|
|
||||||
|
onlineCPUs, err := getOnlineCPUs()
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
|
// Enable CPU Manager in the kubelet.
|
||||||
|
oldCfg := configureCPUManagerInKubelet(f, true, reservedSystemCPUs)
|
||||||
|
defer func() {
|
||||||
|
// restore kubelet config
|
||||||
|
setOldKubeletConfig(f, oldCfg)
|
||||||
|
|
||||||
|
// Delete state file to allow repeated runs
|
||||||
|
deleteStateFile()
|
||||||
|
}()
|
||||||
|
|
||||||
|
configMap := getSRIOVDevicePluginConfigMap(framework.TestContext.SriovdpConfigMapFile)
|
||||||
|
sd := setupSRIOVConfigOrFail(f, configMap)
|
||||||
|
defer teardownSRIOVConfigOrFail(f, sd)
|
||||||
|
|
||||||
|
waitForSRIOVResources(f, sd)
|
||||||
|
|
||||||
|
endpoint, err := util.LocalEndpoint(defaultPodResourcesPath, podresources.Socket)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
|
cli, conn, err := podresources.GetV1Client(endpoint, defaultPodResourcesTimeout, defaultPodResourcesMaxSize)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
ginkgo.By("checking List()")
|
||||||
|
podresourcesListTests(f, cli, sd)
|
||||||
|
ginkgo.By("checking GetAllocatableResources()")
|
||||||
|
podresourcesGetAllocatableResourcesTests(f, cli, sd, onlineCPUs, reservedSystemCPUs)
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
ginkgo.It("should return the expected responses with cpumanager none policy", func() {
|
||||||
|
// current default is "none" policy - no need to restart the kubelet
|
||||||
|
|
||||||
if sriovdevCount, err := countSRIOVDevices(); err != nil || sriovdevCount == 0 {
|
if sriovdevCount, err := countSRIOVDevices(); err != nil || sriovdevCount == 0 {
|
||||||
e2eskipper.Skipf("this test is meant to run on a system with at least one configured VF from SRIOV device")
|
e2eskipper.Skipf("this test is meant to run on a system with at least one configured VF from SRIOV device")
|
||||||
}
|
}
|
||||||
@ -281,9 +539,77 @@ var _ = SIGDescribe("POD Resources [Serial] [Feature:PODResources][NodeFeature:P
|
|||||||
framework.ExpectNoError(err)
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
cli, conn, err := podresources.GetV1Client(endpoint, defaultPodResourcesTimeout, defaultPodResourcesMaxSize)
|
cli, conn, err := podresources.GetV1Client(endpoint, defaultPodResourcesTimeout, defaultPodResourcesMaxSize)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
||||||
podresourcesListTests(f, cli, sd)
|
// intentionally passing empty cpuset instead of onlineCPUs because with none policy
|
||||||
|
// we should get no allocatable cpus - no exclusively allocatable CPUs, depends on policy static
|
||||||
|
podresourcesGetAllocatableResourcesTests(f, cli, sd, cpuset.CPUSet{}, cpuset.CPUSet{})
|
||||||
|
})
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
ginkgo.Context("Without SRIOV devices in the system", func() {
|
||||||
|
ginkgo.It("should return the expected responses with cpumanager static policy enabled", func() {
|
||||||
|
// this is a very rough check. We just want to rule out system that does NOT have enough resources
|
||||||
|
_, cpuAlloc, _ := getLocalNodeCPUDetails(f)
|
||||||
|
|
||||||
|
if cpuAlloc < minCoreCount {
|
||||||
|
e2eskipper.Skipf("Skipping CPU Manager tests since the CPU allocatable < %d", minCoreCount)
|
||||||
|
}
|
||||||
|
if sriovdevCount, err := countSRIOVDevices(); err != nil || sriovdevCount > 0 {
|
||||||
|
e2eskipper.Skipf("this test is meant to run on a system with no configured VF from SRIOV device")
|
||||||
|
}
|
||||||
|
|
||||||
|
onlineCPUs, err := getOnlineCPUs()
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
|
// Enable CPU Manager in the kubelet.
|
||||||
|
oldCfg := configureCPUManagerInKubelet(f, true, reservedSystemCPUs)
|
||||||
|
defer func() {
|
||||||
|
// restore kubelet config
|
||||||
|
setOldKubeletConfig(f, oldCfg)
|
||||||
|
|
||||||
|
// Delete state file to allow repeated runs
|
||||||
|
deleteStateFile()
|
||||||
|
}()
|
||||||
|
|
||||||
|
endpoint, err := util.LocalEndpoint(defaultPodResourcesPath, podresources.Socket)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
|
cli, conn, err := podresources.GetV1Client(endpoint, defaultPodResourcesTimeout, defaultPodResourcesMaxSize)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
podresourcesListTests(f, cli, nil)
|
||||||
|
podresourcesGetAllocatableResourcesTests(f, cli, nil, onlineCPUs, reservedSystemCPUs)
|
||||||
|
})
|
||||||
|
|
||||||
|
ginkgo.It("should return the expected responses with cpumanager none policy", func() {
|
||||||
|
// current default is "none" policy - no need to restart the kubelet
|
||||||
|
|
||||||
|
if sriovdevCount, err := countSRIOVDevices(); err != nil || sriovdevCount > 0 {
|
||||||
|
e2eskipper.Skipf("this test is meant to run on a system with no configured VF from SRIOV device")
|
||||||
|
}
|
||||||
|
|
||||||
|
endpoint, err := util.LocalEndpoint(defaultPodResourcesPath, podresources.Socket)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
|
||||||
|
cli, conn, err := podresources.GetV1Client(endpoint, defaultPodResourcesTimeout, defaultPodResourcesMaxSize)
|
||||||
|
framework.ExpectNoError(err)
|
||||||
|
defer conn.Close()
|
||||||
|
|
||||||
|
// intentionally passing empty cpuset instead of onlineCPUs because with none policy
|
||||||
|
// we should get no allocatable cpus - no exclusively allocatable CPUs, depends on policy static
|
||||||
|
podresourcesGetAllocatableResourcesTests(f, cli, nil, cpuset.CPUSet{}, cpuset.CPUSet{})
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
func getOnlineCPUs() (cpuset.CPUSet, error) {
|
||||||
|
onlineCPUList, err := ioutil.ReadFile("/sys/devices/system/cpu/online")
|
||||||
|
if err != nil {
|
||||||
|
return cpuset.CPUSet{}, err
|
||||||
|
}
|
||||||
|
return cpuset.Parse(strings.TrimSpace(string(onlineCPUList)))
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user