github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-17 23:57:49 +00:00
Code Issues Projects Releases Wiki Activity

test: testcases for pod container manager GetPodContainerName

Browse Source 
Signed-off-by: STRRL <im@strrl.dev>
This commit is contained in:
STRRL 2022-05-02 23:51:27 +08:00
parent e7a2ce75e5
commit 28b9216b42
No known key found for this signature in database
GPG Key ID: C85379B5FF6BF1F0
1 changed files with 160 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

160
pkg/kubelet/cm/pod_container_manager_linux_test.go
View File

@ -20,6 +20,9 @@ limitations under the License.
package cm package cm
import ( import (
"github.com/stretchr/testify/assert"
"k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"strings" "strings"
"testing" "testing"
@ -126,3 +129,160 @@ func TestIsCgroupPod(t *testing.T) {
} }
} }
} }
func TestGetPodContainerName(t *testing.T) {
newGuaranteedPodWithUID := func(uid types.UID) *v1.Pod {
return &v1.Pod{
ObjectMeta: metav1.ObjectMeta{
UID: uid,
},
Spec: v1.PodSpec{
Containers: []v1.Container{
{
Name: "container",
Resources: v1.ResourceRequirements{
Requests: v1.ResourceList{
v1.ResourceCPU: resource.MustParse("1000m"),
v1.ResourceMemory: resource.MustParse("1G"),
},
Limits: v1.ResourceList{
v1.ResourceCPU: resource.MustParse("1000m"),
v1.ResourceMemory: resource.MustParse("1G"),
},
},
},
},
},
}
}
newBurstablePodWithUID := func(uid types.UID) *v1.Pod {
return &v1.Pod{
ObjectMeta: metav1.ObjectMeta{
UID: uid,
},
Spec: v1.PodSpec{
Containers: []v1.Container{
{
Name: "container",
Resources: v1.ResourceRequirements{
Requests: v1.ResourceList{
v1.ResourceCPU: resource.MustParse("1000m"),
v1.ResourceMemory: resource.MustParse("1G"),
},
},
},
},
},
}
}
newBestEffortPodWithUID := func(uid types.UID) *v1.Pod {
return &v1.Pod{
ObjectMeta: metav1.ObjectMeta{
UID: uid,
},
Spec: v1.PodSpec{
Containers: []v1.Container{
{
Name: "container",
},
},
},
}
}
qosContainersInfo := QOSContainersInfo{
Guaranteed: RootCgroupName,
Burstable: NewCgroupName(RootCgroupName, strings.ToLower(string(v1.PodQOSBurstable))),
BestEffort: NewCgroupName(RootCgroupName, strings.ToLower(string(v1.PodQOSBestEffort))),
}
type fields struct {
cgroupManager CgroupManager
}
type args struct {
pod *v1.Pod
}
tests := []struct {
name string
fields fields
args args
wantCgroupName CgroupName
wantLiteralCgroupfs string
}{
{
name: "pod with qos guaranteed and cgroupfs",
fields: fields{
cgroupManager: NewCgroupManager(nil, "cgroupfs"),
},
args: args{
pod: newGuaranteedPodWithUID("fake-uid-1"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.Guaranteed, "podfake-uid-1"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.Guaranteed, "podfake-uid-1").ToCgroupfs(),
}, {
name: "pod with qos guaranteed and systemd",
fields: fields{
cgroupManager: NewCgroupManager(nil, "systemd"),
},
args: args{
pod: newGuaranteedPodWithUID("fake-uid-2"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.Guaranteed, "podfake-uid-2"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.Guaranteed, "podfake-uid-2").ToSystemd(),
}, {
name: "pod with qos burstable and cgroupfs",
fields: fields{
cgroupManager: NewCgroupManager(nil, "cgroupfs"),
},
args: args{
pod: newBurstablePodWithUID("fake-uid-3"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.Burstable, "podfake-uid-3"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.Burstable, "podfake-uid-3").ToCgroupfs(),
}, {
name: "pod with qos burstable and systemd",
fields: fields{
cgroupManager: NewCgroupManager(nil, "systemd"),
},
args: args{
pod: newBurstablePodWithUID("fake-uid-4"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.Burstable, "podfake-uid-4"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.Burstable, "podfake-uid-4").ToSystemd(),
}, {
name: "pod with qos best-effort and cgroupfs",
fields: fields{
cgroupManager: NewCgroupManager(nil, "cgroupfs"),
},
args: args{
pod: newBestEffortPodWithUID("fake-uid-5"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.BestEffort, "podfake-uid-5"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.BestEffort, "podfake-uid-5").ToCgroupfs(),
}, {
name: "pod with qos best-effort and systemd",
fields: fields{
cgroupManager: NewCgroupManager(nil, "systemd"),
},
args: args{
pod: newBestEffortPodWithUID("fake-uid-6"),
},
wantCgroupName: NewCgroupName(qosContainersInfo.BestEffort, "podfake-uid-6"),
wantLiteralCgroupfs: NewCgroupName(qosContainersInfo.BestEffort, "podfake-uid-6").ToSystemd(),
},
}
for _, tt := range tests {
pcm := &podContainerManagerImpl{
cgroupManager: tt.fields.cgroupManager,
qosContainersInfo: qosContainersInfo,
}
t.Run(tt.name, func(t *testing.T) {
actualCgroupName, actualLiteralCgroupfs := pcm.GetPodContainerName(tt.args.pod)
assert.Equalf(t, tt.wantCgroupName, actualCgroupName, "Unexpected cgroup name for pod with UID %s, container resources: %v", tt.args.pod.UID, tt.args.pod.Spec.Containers[0].Resources)
assert.Equalf(t, tt.wantLiteralCgroupfs, actualLiteralCgroupfs, "Unexpected literal cgroupfs for pod with UID %s, container resources: %v", tt.args.pod.UID, tt.args.pod.Spec.Containers[0].Resources)
})
}
}