github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-29 14:37:00 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #119786 from bzsuni/ut/pkg/kubelet/util

Browse Source 
[UT] add ut for pkg/kubelet/util
This commit is contained in:
Kubernetes Prow Robot 2024-06-04 11:23:28 -07:00 committed by GitHub
commit 9c4181e234
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 57 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
pkg/kubelet/util/util_test.go
View File

@ -20,6 +20,8 @@ import (
"testing"
"github.com/stretchr/testify/assert"
v1 "k8s.io/api/core/v1"
)
func TestGetNodenameForKernel(t *testing.T) {
@ -86,3 +88,58 @@ func TestGetNodenameForKernel(t *testing.T) {
}
}
func TestGetContainerByIndex(t *testing.T) {
testCases := []struct {
title string
containers []v1.Container
statuses []v1.ContainerStatus
idx int
expectedContainer v1.Container
expectedOK bool
}{
{
title: "idx is less than zero",
containers: []v1.Container{{Name: "container-1"}},
statuses: []v1.ContainerStatus{{Name: "container-1"}},
idx: -1,
expectedContainer: v1.Container{},
expectedOK: false,
}, {
title: "idx is large then number of containers",
containers: []v1.Container{{Name: "container-1"}},
statuses: []v1.ContainerStatus{{Name: "container-1"}},
idx: 2,
expectedContainer: v1.Container{},
expectedOK: false,
}, {
title: "idx is large then number of statuses",
containers: []v1.Container{{Name: "container-1"}, {Name: "container-2"}},
statuses: []v1.ContainerStatus{{Name: "container-1"}},
idx: 2,
expectedContainer: v1.Container{},
expectedOK: false,
}, {
title: "names do not match",
containers: []v1.Container{{Name: "container-1"}},
statuses: []v1.ContainerStatus{{Name: "invalid-container"}},
idx: 0,
expectedContainer: v1.Container{},
expectedOK: false,
}, {
title: "valid container index",
containers: []v1.Container{{Name: "container-1"}, {Name: "container-2"}},
statuses: []v1.ContainerStatus{{Name: "container-1"}, {Name: "container-2"}},
idx: 1,
expectedContainer: v1.Container{Name: "container-2"},
expectedOK: true,
},
}
for _, tc := range testCases {
container, ok := GetContainerByIndex(tc.containers, tc.statuses, tc.idx)
if container.Name != tc.expectedContainer.Name || ok != tc.expectedOK {
t.Errorf("%s - Expected container: %v, got container: %v, expected ok: %v, got ok: %v", tc.title, tc.expectedContainer, container, tc.expectedOK, ok)
}
}
}