mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-23 19:56:01 +00:00
[node][e2e test]: Make sure reconcile labels is working fine
This commit is contained in:
parent
21c5c2ec5c
commit
3af5d37be7
101
test/e2e_node/os_label_rename_test.go
Normal file
101
test/e2e_node/os_label_rename_test.go
Normal file
@ -0,0 +1,101 @@
|
||||
//go:build cgo && linux
|
||||
// +build cgo,linux
|
||||
|
||||
/*
|
||||
Copyright 2021 The Kubernetes Authors.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
package e2enode
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"runtime"
|
||||
"time"
|
||||
|
||||
"github.com/onsi/ginkgo"
|
||||
|
||||
v1 "k8s.io/api/core/v1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/types"
|
||||
"k8s.io/apimachinery/pkg/util/wait"
|
||||
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
||||
node2 "k8s.io/kubernetes/pkg/util/node"
|
||||
"k8s.io/kubernetes/test/e2e/framework"
|
||||
)
|
||||
|
||||
var _ = SIGDescribe("OSArchLabelReconciliation [Serial] [Slow] [Disruptive]", func() {
|
||||
f := framework.NewDefaultFramework("node-label-reconciliation")
|
||||
ginkgo.Context("Kubelet", func() {
|
||||
ginkgo.It("should reconcile the OS and Arch labels when restarted", func() {
|
||||
node := getLocalNode(f)
|
||||
framework.ExpectNodeHasLabel(f.ClientSet, node.Name, v1.LabelOSStable, runtime.GOOS)
|
||||
framework.ExpectNodeHasLabel(f.ClientSet, node.Name, v1.LabelArchStable, runtime.GOARCH)
|
||||
|
||||
ginkgo.By("killing and restarting kubelet")
|
||||
// Let's kill the kubelet
|
||||
startKubelet := stopKubelet()
|
||||
// Update labels
|
||||
newNode := node.DeepCopy()
|
||||
newNode.Labels[v1.LabelOSStable] = "dummyOS"
|
||||
newNode.Labels[v1.LabelArchStable] = "dummyArch"
|
||||
_, _, err := node2.PatchNodeStatus(f.ClientSet.CoreV1(), types.NodeName(node.Name), node, newNode)
|
||||
framework.ExpectNoError(err)
|
||||
// Restart kubelet
|
||||
startKubelet()
|
||||
framework.ExpectNoError(framework.WaitForAllNodesSchedulable(f.ClientSet, framework.RestartNodeReadyAgainTimeout))
|
||||
// If this happens right, node should have all the labels reset properly
|
||||
err = waitForNodeLabels(f.ClientSet.CoreV1(), node.Name, 5*time.Minute)
|
||||
framework.ExpectNoError(err)
|
||||
})
|
||||
ginkgo.It("should reconcile the OS and Arch labels when running", func() {
|
||||
|
||||
node := getLocalNode(f)
|
||||
framework.ExpectNodeHasLabel(f.ClientSet, node.Name, v1.LabelOSStable, runtime.GOOS)
|
||||
framework.ExpectNodeHasLabel(f.ClientSet, node.Name, v1.LabelArchStable, runtime.GOARCH)
|
||||
|
||||
// Update labels
|
||||
newNode := node.DeepCopy()
|
||||
newNode.Labels[v1.LabelOSStable] = "dummyOS"
|
||||
newNode.Labels[v1.LabelArchStable] = "dummyArch"
|
||||
_, _, err := node2.PatchNodeStatus(f.ClientSet.CoreV1(), types.NodeName(node.Name), node, newNode)
|
||||
framework.ExpectNoError(err)
|
||||
err = waitForNodeLabels(f.ClientSet.CoreV1(), node.Name, 5*time.Minute)
|
||||
framework.ExpectNoError(err)
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
// waitForNodeLabels waits for the nodes to be have appropriate labels.
|
||||
func waitForNodeLabels(c v1core.CoreV1Interface, nodeName string, timeout time.Duration) error {
|
||||
ginkgo.By(fmt.Sprintf("Waiting for node %v to have appropriate labels", nodeName))
|
||||
// Poll until the node has desired labels
|
||||
return wait.Poll(framework.Poll, timeout,
|
||||
func() (bool, error) {
|
||||
node, err := c.Nodes().Get(context.TODO(), nodeName, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
osLabel, ok := node.Labels[v1.LabelOSStable]
|
||||
if !ok || osLabel != runtime.GOOS {
|
||||
return false, nil
|
||||
}
|
||||
archLabel, ok := node.Labels[v1.LabelArchStable]
|
||||
if !ok || archLabel != runtime.GOARCH {
|
||||
return false, nil
|
||||
}
|
||||
return true, nil
|
||||
})
|
||||
}
|
Loading…
Reference in New Issue
Block a user