vendor files

vendor/k8s.io/kubernetes/cmd/kubeadm/test/util.go

@@ -0,0 +1,137 @@
+/*
+Copyright 2017 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 test
+
+import (
+	"html/template"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/renstrom/dedent"
+
+	kubeadmapi "k8s.io/kubernetes/cmd/kubeadm/app/apis/kubeadm"
+	kubeadmconstants "k8s.io/kubernetes/cmd/kubeadm/app/constants"
+	"k8s.io/kubernetes/cmd/kubeadm/app/phases/certs/pkiutil"
+	certtestutil "k8s.io/kubernetes/cmd/kubeadm/test/certs"
+)
+
+// SetupTempDir is a utility function for kubeadm testing, that creates a temporary directory
+// NB. it is up to the caller to cleanup the folder at the end of the test with defer os.RemoveAll(tmpdir)
+func SetupTempDir(t *testing.T) string {
+	tmpdir, err := ioutil.TempDir("", "")
+	if err != nil {
+		t.Fatalf("Couldn't create tmpdir")
+	}
+
+	return tmpdir
+}
+
+// SetupMasterConfigurationFile is a utility function for kubeadm testing that writes a master configuration file
+// into /config subfolder of a given temporary directory.
+// The funtion returns the path of the created master configuration file.
+func SetupMasterConfigurationFile(t *testing.T, tmpdir string, cfg *kubeadmapi.MasterConfiguration) string {
+
+	cfgPath := filepath.Join(tmpdir, "config/masterconfig.yaml")
+	if err := os.MkdirAll(filepath.Dir(cfgPath), os.FileMode(0755)); err != nil {
+		t.Fatalf("Couldn't create cfgDir")
+	}
+
+	cfgTemplate := template.Must(template.New("init").Parse(dedent.Dedent(`
+		apiVersion: kubeadm.k8s.io/v1alpha1
+		kind: MasterConfiguration
+		certificatesDir: {{.CertificatesDir}}
+		api:
+		   advertiseAddress: {{.API.AdvertiseAddress}}
+		   bindPort: {{.API.BindPort}}
+		nodeName: {{.NodeName}}
+		`)))
+
+	f, err := os.Create(cfgPath)
+	if err != nil {
+		t.Fatalf("error creating masterconfig file %s: %v", cfgPath, err)
+	}
+
+	err = cfgTemplate.Execute(f, cfg)
+	if err != nil {
+		t.Fatalf("error generating masterconfig file %s: %v", cfgPath, err)
+	}
+	f.Close()
+
+	return cfgPath
+}
+
+// SetupEmptyFiles is a utility function for kubeadm testing that creates one or more empty files (touch)
+func SetupEmptyFiles(t *testing.T, tmpdir string, fileNames ...string) {
+	for _, fileName := range fileNames {
+		newFile, err := os.Create(filepath.Join(tmpdir, fileName))
+		if err != nil {
+			t.Fatalf("Error creating file %s in %s: %v", fileName, tmpdir, err)
+		}
+		newFile.Close()
+	}
+}
+
+// SetupPkiDirWithCertificateAuthorithy is a utility function for kubeadm testing that creates a
+// CertificateAuthorithy cert/key pair into /pki subfolder of a given temporary directory.
+// The funtion returns the path of the created pki.
+func SetupPkiDirWithCertificateAuthorithy(t *testing.T, tmpdir string) string {
+	caCert, caKey := certtestutil.SetupCertificateAuthorithy(t)
+
+	certDir := filepath.Join(tmpdir, "pki")
+	if err := pkiutil.WriteCertAndKey(certDir, kubeadmconstants.CACertAndKeyBaseName, caCert, caKey); err != nil {
+		t.Fatalf("failure while saving CA certificate and key: %v", err)
+	}
+
+	return certDir
+}
+
+// AssertFilesCount is a utility function for kubeadm testing that asserts if the given folder contains
+// count files.
+func AssertFilesCount(t *testing.T, dirName string, count int) {
+	files, err := ioutil.ReadDir(dirName)
+	if err != nil {
+		t.Fatalf("Couldn't read files from tmpdir: %s", err)
+	}
+
+	countFiles := 0
+	for _, f := range files {
+		if !f.IsDir() {
+			countFiles++
+		}
+	}
+
+	if countFiles != count {
+		t.Errorf("dir does contains %d, %d expected", len(files), count)
+		for _, f := range files {
+			t.Error(f.Name())
+		}
+	}
+}
+
+// AssertFileExists is a utility function for kubeadm testing that asserts if the given folder contains
+// the given files.
+func AssertFileExists(t *testing.T, dirName string, fileNames ...string) {
+	for _, fileName := range fileNames {
+		path := filepath.Join(dirName, fileName)
+
+		if _, err := os.Stat(path); os.IsNotExist(err) {
+			t.Errorf("file %s does not exist", fileName)
+		}
+	}
+}