mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-08-03 09:22:44 +00:00
Merge pull request #116932 from my-git9/ut-preemption
[ut] Increase test coverage for kubelet/pleg
This commit is contained in:
commit
e823b56a0b
@ -18,12 +18,14 @@ package pleg
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
|
|
||||||
"k8s.io/apimachinery/pkg/types"
|
"k8s.io/apimachinery/pkg/types"
|
||||||
"k8s.io/component-base/metrics/testutil"
|
"k8s.io/component-base/metrics/testutil"
|
||||||
v1 "k8s.io/cri-api/pkg/apis/runtime/v1"
|
v1 "k8s.io/cri-api/pkg/apis/runtime/v1"
|
||||||
@ -131,3 +133,101 @@ func testMetric(t *testing.T, expectedMetric string, metricName string) {
|
|||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestEventedPLEG_getPodIPs(t *testing.T) {
|
||||||
|
cache := kubecontainer.NewCache()
|
||||||
|
type args struct {
|
||||||
|
pid types.UID
|
||||||
|
status *kubecontainer.PodStatus
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
args args
|
||||||
|
oldstatus *kubecontainer.PodStatus
|
||||||
|
expected []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "status ips is not empty",
|
||||||
|
args: args{
|
||||||
|
pid: "62212",
|
||||||
|
status: &kubecontainer.PodStatus{
|
||||||
|
IPs: []string{"10.0.0.10", "10.23.0.1"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
oldstatus: &kubecontainer.PodStatus{
|
||||||
|
IPs: []string{"192.168.0.10", "192.168.0.1"},
|
||||||
|
},
|
||||||
|
expected: []string{"10.0.0.10", "10.23.0.1"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "status ips is empty and SandboxStatuses has PodSandboxState_SANDBOX_READY state",
|
||||||
|
args: args{
|
||||||
|
pid: "62212",
|
||||||
|
status: &kubecontainer.PodStatus{
|
||||||
|
SandboxStatuses: []*v1.PodSandboxStatus{
|
||||||
|
{
|
||||||
|
Id: "sandboxID2",
|
||||||
|
Metadata: &v1.PodSandboxMetadata{Attempt: uint32(1)},
|
||||||
|
State: v1.PodSandboxState_SANDBOX_READY,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Id: "sandboxID1",
|
||||||
|
Metadata: &v1.PodSandboxMetadata{Attempt: uint32(0)},
|
||||||
|
State: v1.PodSandboxState_SANDBOX_NOTREADY,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
oldstatus: &kubecontainer.PodStatus{
|
||||||
|
IPs: []string{"192.168.0.10", "192.168.0.1"},
|
||||||
|
},
|
||||||
|
expected: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "status and cache ips are empty",
|
||||||
|
args: args{
|
||||||
|
pid: "62212",
|
||||||
|
status: &kubecontainer.PodStatus{},
|
||||||
|
},
|
||||||
|
oldstatus: &kubecontainer.PodStatus{
|
||||||
|
IPs: []string{},
|
||||||
|
},
|
||||||
|
expected: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "sandbox state is no PodSandboxState_SANDBOX_READY",
|
||||||
|
args: args{
|
||||||
|
pid: "62212",
|
||||||
|
status: &kubecontainer.PodStatus{
|
||||||
|
SandboxStatuses: []*v1.PodSandboxStatus{
|
||||||
|
{
|
||||||
|
Id: "sandboxID2",
|
||||||
|
Metadata: &v1.PodSandboxMetadata{Attempt: uint32(1)},
|
||||||
|
State: v1.PodSandboxState_SANDBOX_NOTREADY,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Id: "sandboxID1",
|
||||||
|
Metadata: &v1.PodSandboxMetadata{Attempt: uint32(0)},
|
||||||
|
State: v1.PodSandboxState_SANDBOX_NOTREADY,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
oldstatus: &kubecontainer.PodStatus{
|
||||||
|
IPs: []string{"192.168.0.10", "192.168.0.1"},
|
||||||
|
},
|
||||||
|
expected: []string{"192.168.0.10", "192.168.0.1"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
cache.Set(test.args.pid, test.oldstatus, nil, time.Time{})
|
||||||
|
e := &EventedPLEG{
|
||||||
|
cache: cache,
|
||||||
|
}
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
if got := e.getPodIPs(test.args.pid, test.args.status); !reflect.DeepEqual(got, test.expected) {
|
||||||
|
t.Errorf("EventedPLEG.getPodIPs() = %v, expected %v", got, test.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user