mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-11-03 15:25:19 +00:00
119 lines
4.0 KiB
Go
119 lines
4.0 KiB
Go
/*
|
|
Copyright 2019 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 kubectl
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
"k8s.io/kubernetes/test/e2e/framework"
|
|
e2epod "k8s.io/kubernetes/test/e2e/framework/pod"
|
|
testutils "k8s.io/kubernetes/test/utils"
|
|
)
|
|
|
|
// TestKubeconfig is a struct containing the minimum attributes needed to run KubectlCmd.
|
|
type TestKubeconfig struct {
|
|
CertDir string
|
|
Host string
|
|
KubeConfig string
|
|
KubeContext string
|
|
KubectlPath string
|
|
}
|
|
|
|
// NewTestKubeconfig returns a new Kubeconfig struct instance.
|
|
func NewTestKubeconfig(certdir string, host string, kubeconfig string, kubecontext string, kubectlpath string) *TestKubeconfig {
|
|
return &TestKubeconfig{
|
|
CertDir: certdir,
|
|
Host: host,
|
|
KubeConfig: kubeconfig,
|
|
KubeContext: kubecontext,
|
|
KubectlPath: kubectlpath,
|
|
}
|
|
}
|
|
|
|
// KubectlCmd runs the kubectl executable through the wrapper script.
|
|
func (tk *TestKubeconfig) KubectlCmd(args ...string) *exec.Cmd {
|
|
defaultArgs := []string{}
|
|
|
|
// Reference a --server option so tests can run anywhere.
|
|
if tk.Host != "" {
|
|
defaultArgs = append(defaultArgs, "--"+clientcmd.FlagAPIServer+"="+tk.Host)
|
|
}
|
|
if tk.KubeConfig != "" {
|
|
defaultArgs = append(defaultArgs, "--"+clientcmd.RecommendedConfigPathFlag+"="+tk.KubeConfig)
|
|
|
|
// Reference the KubeContext
|
|
if tk.KubeContext != "" {
|
|
defaultArgs = append(defaultArgs, "--"+clientcmd.FlagContext+"="+tk.KubeContext)
|
|
}
|
|
|
|
} else {
|
|
if tk.CertDir != "" {
|
|
defaultArgs = append(defaultArgs,
|
|
fmt.Sprintf("--certificate-authority=%s", filepath.Join(tk.CertDir, "ca.crt")),
|
|
fmt.Sprintf("--client-certificate=%s", filepath.Join(tk.CertDir, "kubecfg.crt")),
|
|
fmt.Sprintf("--client-key=%s", filepath.Join(tk.CertDir, "kubecfg.key")))
|
|
}
|
|
}
|
|
kubectlArgs := append(defaultArgs, args...)
|
|
|
|
//We allow users to specify path to kubectl, so you can test either "kubectl" or "cluster/kubectl.sh"
|
|
//and so on.
|
|
cmd := exec.Command(tk.KubectlPath, kubectlArgs...)
|
|
|
|
//caller will invoke this and wait on it.
|
|
return cmd
|
|
}
|
|
|
|
// LogFailedContainers runs `kubectl logs` on a failed containers.
|
|
func LogFailedContainers(c clientset.Interface, ns string, logFunc func(ftm string, args ...interface{})) {
|
|
podList, err := c.CoreV1().Pods(ns).List(metav1.ListOptions{})
|
|
if err != nil {
|
|
logFunc("Error getting pods in namespace '%s': %v", ns, err)
|
|
return
|
|
}
|
|
logFunc("Running kubectl logs on non-ready containers in %v", ns)
|
|
for _, pod := range podList.Items {
|
|
if res, err := testutils.PodRunningReady(&pod); !res || err != nil {
|
|
kubectlLogPod(c, pod, "", framework.Logf)
|
|
}
|
|
}
|
|
}
|
|
|
|
func kubectlLogPod(c clientset.Interface, pod v1.Pod, containerNameSubstr string, logFunc func(ftm string, args ...interface{})) {
|
|
for _, container := range pod.Spec.Containers {
|
|
if strings.Contains(container.Name, containerNameSubstr) {
|
|
// Contains() matches all strings if substr is empty
|
|
logs, err := e2epod.GetPodLogs(c, pod.Namespace, pod.Name, container.Name)
|
|
if err != nil {
|
|
logs, err = e2epod.GetPreviousPodLogs(c, pod.Namespace, pod.Name, container.Name)
|
|
if err != nil {
|
|
logFunc("Failed to get logs of pod %v, container %v, err: %v", pod.Name, container.Name, err)
|
|
}
|
|
}
|
|
logFunc("Logs of %v/%v:%v on node %v", pod.Namespace, pod.Name, container.Name, pod.Spec.NodeName)
|
|
logFunc("%s : STARTLOG\n%s\nENDLOG for container %v:%v:%v", containerNameSubstr, logs, pod.Namespace, pod.Name, container.Name)
|
|
}
|
|
}
|
|
}
|