mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-20 10:20:51 +00:00
kubeadm: add testcases for kubeadm reset
For parity with `init` and `join`, the test file only covers some basic scenarios and is expected to be expanded when the config file for `kubeadm reset` is ready. Signed-off-by: Dave Chen <dave.chen@arm.com>
This commit is contained in:
parent
78c704d4f6
commit
3ddedf00e5
115
cmd/kubeadm/app/cmd/reset_test.go
Normal file
115
cmd/kubeadm/app/cmd/reset_test.go
Normal file
@ -0,0 +1,115 @@
|
|||||||
|
/*
|
||||||
|
Copyright 2022 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 cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"github.com/google/go-cmp/cmp/cmpopts"
|
||||||
|
"k8s.io/apimachinery/pkg/util/sets"
|
||||||
|
|
||||||
|
"k8s.io/kubernetes/cmd/kubeadm/app/cmd/options"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewResetData(t *testing.T) {
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
args []string
|
||||||
|
flags map[string]string
|
||||||
|
validate func(*testing.T, *resetData)
|
||||||
|
expectError string
|
||||||
|
data *resetData
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "flags parsed correctly",
|
||||||
|
flags: map[string]string{
|
||||||
|
options.CertificatesDir: "/tmp",
|
||||||
|
options.NodeCRISocket: "unix:///var/run/crio/crio.sock",
|
||||||
|
options.IgnorePreflightErrors: "all",
|
||||||
|
options.ForceReset: "true",
|
||||||
|
options.DryRun: "true",
|
||||||
|
options.CleanupTmpDir: "true",
|
||||||
|
},
|
||||||
|
data: &resetData{
|
||||||
|
certificatesDir: "/tmp",
|
||||||
|
criSocketPath: "unix:///var/run/crio/crio.sock",
|
||||||
|
ignorePreflightErrors: sets.NewString("all"),
|
||||||
|
forceReset: true,
|
||||||
|
dryRun: true,
|
||||||
|
cleanupTmpDir: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "fails if preflight ignores all but has individual check",
|
||||||
|
flags: map[string]string{
|
||||||
|
options.IgnorePreflightErrors: "all,something-else",
|
||||||
|
},
|
||||||
|
expectError: "don't specify individual checks if 'all' is used",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "pre-flights errors from CLI args",
|
||||||
|
flags: map[string]string{
|
||||||
|
options.IgnorePreflightErrors: "a,b",
|
||||||
|
},
|
||||||
|
validate: expectedResetIgnorePreflightErrors(sets.NewString("a", "b")),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
// initialize an external reset option and inject it to the reset cmd
|
||||||
|
resetOptions := newResetOptions()
|
||||||
|
cmd := newCmdReset(nil, nil, resetOptions)
|
||||||
|
|
||||||
|
// sets cmd flags (that will be reflected on the reset options)
|
||||||
|
for f, v := range tc.flags {
|
||||||
|
cmd.Flags().Set(f, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
// test newResetData method
|
||||||
|
data, err := newResetData(cmd, resetOptions, nil, nil)
|
||||||
|
if err != nil && !strings.Contains(err.Error(), tc.expectError) {
|
||||||
|
t.Fatalf("newResetData returned unexpected error, expected: %s, got %v", tc.expectError, err)
|
||||||
|
}
|
||||||
|
if err == nil && len(tc.expectError) != 0 {
|
||||||
|
t.Fatalf("newResetData didn't return error expected %s", tc.expectError)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tc.data != nil {
|
||||||
|
if diff := cmp.Diff(tc.data, data, cmp.AllowUnexported(resetData{}), cmpopts.IgnoreFields(resetData{}, "client", "cfg")); diff != "" {
|
||||||
|
t.Fatalf("newResetData returned data (-want,+got):\n%s", diff)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// exec additional validation on the returned value
|
||||||
|
if tc.validate != nil {
|
||||||
|
tc.validate(t, data)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectedResetIgnorePreflightErrors(expected sets.String) func(t *testing.T, data *resetData) {
|
||||||
|
return func(t *testing.T, data *resetData) {
|
||||||
|
if !expected.Equal(data.ignorePreflightErrors) {
|
||||||
|
t.Errorf("Invalid ignore preflight errors. Expected: %v. Actual: %v", expected.List(), data.ignorePreflightErrors.List())
|
||||||
|
}
|
||||||
|
if data.cfg != nil && !expected.HasAll(data.cfg.NodeRegistration.IgnorePreflightErrors...) {
|
||||||
|
t.Errorf("Invalid ignore preflight errors in InitConfiguration. Expected: %v. Actual: %v", expected.List(), data.cfg.NodeRegistration.IgnorePreflightErrors)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user