kubeadm: add better test coverage to reset.go

added tests in:
- TestNewReset()
- TestNewCmdReset()
- TestConfigDirCleaner()
pull/8/head
Lubomir I. Ivanov 2018-01-19 23:07:46 +02:00
parent 891b471064
commit 29f8ed23ec
2 changed files with 44 additions and 1 deletions

View File

@ -86,6 +86,7 @@ go_test(
embed = [":go_default_library"],
deps = [
"//cmd/kubeadm/app/apis/kubeadm/v1alpha1:go_default_library",
"//cmd/kubeadm/app/apis/kubeadm/validation:go_default_library",
"//cmd/kubeadm/app/constants:go_default_library",
"//cmd/kubeadm/app/preflight:go_default_library",
"//vendor/k8s.io/api/core/v1:go_default_library",

View File

@ -18,12 +18,15 @@ package cmd
import (
"errors"
"io"
"io/ioutil"
"os"
"path/filepath"
"strings"
"testing"
kubeadmapiext "k8s.io/kubernetes/cmd/kubeadm/app/apis/kubeadm/v1alpha1"
"k8s.io/kubernetes/cmd/kubeadm/app/apis/kubeadm/validation"
kubeadmconstants "k8s.io/kubernetes/cmd/kubeadm/app/constants"
"k8s.io/kubernetes/cmd/kubeadm/app/preflight"
"k8s.io/utils/exec"
@ -52,8 +55,38 @@ func assertDirEmpty(t *testing.T, path string) {
}
}
func TestNewReset(t *testing.T) {
certsDir := kubeadmapiext.DefaultCertificatesDir
criSocketPath := "/var/run/dockershim.sock"
skipPreFlight := false
ignorePreflightErrors := []string{"all"}
ignorePreflightErrorsSet, _ := validation.ValidateIgnorePreflightErrors(ignorePreflightErrors, skipPreFlight)
NewReset(ignorePreflightErrorsSet, certsDir, criSocketPath)
ignorePreflightErrors = []string{}
ignorePreflightErrorsSet, _ = validation.ValidateIgnorePreflightErrors(ignorePreflightErrors, skipPreFlight)
NewReset(ignorePreflightErrorsSet, certsDir, criSocketPath)
}
func TestNewCmdReset(t *testing.T) {
var out io.Writer
cmd := NewCmdReset(out)
tmpDir, err := ioutil.TempDir("", "kubeadm-reset-test")
if err != nil {
t.Errorf("Unable to create temporary directory: %v", err)
}
args := []string{"--ignore-preflight-errors=all", "--cert-dir=" + tmpDir}
cmd.SetArgs(args)
if err := cmd.Execute(); err != nil {
t.Errorf("Cannot execute reset command: %v", err)
}
}
func TestConfigDirCleaner(t *testing.T) {
tests := map[string]struct {
resetDir string
setupDirs []string
setupFiles []string
verifyExists []string
@ -139,6 +172,12 @@ func TestConfigDirCleaner(t *testing.T) {
"manifests",
},
},
"not a directory": {
resetDir: "test-path",
setupFiles: []string{
"test-path",
},
},
}
for name, test := range tests {
@ -167,7 +206,10 @@ func TestConfigDirCleaner(t *testing.T) {
defer f.Close()
}
resetConfigDir(tmpDir, filepath.Join(tmpDir, "pki"))
if test.resetDir == "" {
test.resetDir = "pki"
}
resetConfigDir(tmpDir, filepath.Join(tmpDir, test.resetDir))
// Verify the files we cleanup implicitly in every test:
assertExists(t, tmpDir)