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

Merge pull request #126610 from liyuerich/postupgrade

Browse Source 
kubeadm: add UT for postupgrade
This commit is contained in:
Kubernetes Prow Robot 2024-09-10 15:16:13 +01:00 committed by GitHub
commit a8fb2cefbb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 142 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

142
cmd/kubeadm/app/phases/upgrade/postupgrade_test.go
View File

@ -19,13 +19,19 @@ package upgrade
import (
"os"
"path/filepath"
"reflect"
"regexp"
"strings"
"testing"
"github.com/pkg/errors"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
errorsutil "k8s.io/apimachinery/pkg/util/errors"
"k8s.io/client-go/kubernetes/fake"
kubeadmapi "k8s.io/kubernetes/cmd/kubeadm/app/apis/kubeadm"
"k8s.io/kubernetes/cmd/kubeadm/app/componentconfigs"
"k8s.io/kubernetes/cmd/kubeadm/app/constants"
@ -169,6 +175,142 @@ func TestWriteKubeletConfigFiles(t *testing.T) {
}
}
func TestUnupgradedControlPlaneInstances(t *testing.T) {
testCases := []struct {
name string
pods []corev1.Pod
currentNode string
expectedNodes []string
expectError bool
}{
{
name: "two nodes, one needs upgrade",
pods: []corev1.Pod{
{
ObjectMeta: metav1.ObjectMeta{
Name: "kube-apiserver-1",
Namespace: metav1.NamespaceSystem,
Labels: map[string]string{
"component": constants.KubeAPIServer,
},
},
Spec: corev1.PodSpec{
NodeName: "node-1",
Containers: []corev1.Container{
{Name: constants.KubeAPIServer, Image: "registry.kl8s.io/kube-apiserver:v2"},
},
},
},
{
ObjectMeta: metav1.ObjectMeta{
Name: "kube-apiserver-2",
Namespace: metav1.NamespaceSystem,
Labels: map[string]string{
"component": constants.KubeAPIServer,
},
},
Spec: corev1.PodSpec{
NodeName: "node-2",
Containers: []corev1.Container{
{Name: constants.KubeAPIServer, Image: "registry.kl8s.io/kube-apiserver:v1"},
},
},
},
},
currentNode: "node-1",
expectedNodes: []string{"node-2"},
expectError: false,
},
{
name: "one node which is already upgraded",
pods: []corev1.Pod{
{
ObjectMeta: metav1.ObjectMeta{
Name: "kube-apiserver-1",
Namespace: metav1.NamespaceSystem,
Labels: map[string]string{
"component": constants.KubeAPIServer,
},
},
Spec: corev1.PodSpec{
NodeName: "node-1",
Containers: []corev1.Container{
{Name: constants.KubeAPIServer, Image: "registry.kl8s.io/kube-apiserver:v2"},
},
},
},
},
currentNode: "node-1",
expectedNodes: nil,
expectError: false,
},
{
name: "two nodes, both already upgraded",
pods: []corev1.Pod{
{
ObjectMeta: metav1.ObjectMeta{
Name: "kube-apiserver-1",
Namespace: metav1.NamespaceSystem,
Labels: map[string]string{
"component": constants.KubeAPIServer,
},
},
Spec: corev1.PodSpec{
NodeName: "node-1",
Containers: []corev1.Container{
{Name: constants.KubeAPIServer, Image: "registry.kl8s.io/kube-apiserver:v2"},
},
},
},
{
ObjectMeta: metav1.ObjectMeta{
Name: "kube-apiserver-2",
Namespace: metav1.NamespaceSystem,
Labels: map[string]string{
"component": constants.KubeAPIServer,
},
},
Spec: corev1.PodSpec{
NodeName: "node-2",
Containers: []corev1.Container{
{Name: constants.KubeAPIServer, Image: "registry.kl8s.io/kube-apiserver:v2"},
},
},
},
},
currentNode: "node-1",
expectedNodes: nil,
expectError: false,
},
{
name: "no kube-apiserver pods",
pods: []corev1.Pod{},
currentNode: "node-1",
expectedNodes: nil,
expectError: true,
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
var runtimeObjs []runtime.Object
for _, pod := range tc.pods {
runtimeObjs = append(runtimeObjs, &pod) // Use pointer
}
client := fake.NewSimpleClientset(runtimeObjs...)
nodes, err := UnupgradedControlPlaneInstances(client, tc.currentNode)
if tc.expectError != (err != nil) {
t.Fatalf("expected error: %v, got: %v", tc.expectError, err)
}
if !reflect.DeepEqual(nodes, tc.expectedNodes) {
t.Fatalf("expected unupgraded control plane instances: %v, got: %v", tc.expectedNodes, nodes)
}
})
}
}
// Just some stub code, the code could be enriched when necessary.
type componentConfig struct {
userSupplied bool