123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452 |
- /*
- Copyright 2016 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 framework
- import (
- "flag"
- "fmt"
- "io/ioutil"
- "os"
- "sort"
- "strings"
- "time"
- "github.com/onsi/ginkgo/config"
- "github.com/pkg/errors"
- restclient "k8s.io/client-go/rest"
- "k8s.io/client-go/tools/clientcmd"
- clientcmdapi "k8s.io/client-go/tools/clientcmd/api"
- cliflag "k8s.io/component-base/cli/flag"
- "k8s.io/klog"
- kubeletconfig "k8s.io/kubernetes/pkg/kubelet/apis/config"
- )
- const (
- defaultHost = "http://127.0.0.1:8080"
- // DefaultNumNodes is the number of nodes. If not specified, then number of nodes is auto-detected
- DefaultNumNodes = -1
- )
- // TestContextType contains test settings and global state. Due to
- // historic reasons, it is a mixture of items managed by the test
- // framework itself, cloud providers and individual tests.
- // The goal is to move anything not required by the framework
- // into the code which uses the settings.
- //
- // The recommendation for those settings is:
- // - They are stored in their own context structure or local
- // variables.
- // - The standard `flag` package is used to register them.
- // The flag name should follow the pattern <part1>.<part2>....<partn>
- // where the prefix is unlikely to conflict with other tests or
- // standard packages and each part is in lower camel case. For
- // example, test/e2e/storage/csi/context.go could define
- // storage.csi.numIterations.
- // - framework/config can be used to simplify the registration of
- // multiple options with a single function call:
- // var storageCSI {
- // NumIterations `default:"1" usage:"number of iterations"`
- // }
- // _ config.AddOptions(&storageCSI, "storage.csi")
- // - The direct use Viper in tests is possible, but discouraged because
- // it only works in test suites which use Viper (which is not
- // required) and the supported options cannot be
- // discovered by a test suite user.
- //
- // Test suite authors can use framework/viper to make all command line
- // parameters also configurable via a configuration file.
- type TestContextType struct {
- KubeConfig string
- KubeContext string
- KubeAPIContentType string
- KubeVolumeDir string
- CertDir string
- Host string
- // TODO: Deprecating this over time... instead just use gobindata_util.go , see #23987.
- RepoRoot string
- DockershimCheckpointDir string
- // ListImages will list off all images that are used then quit
- ListImages bool
- // Provider identifies the infrastructure provider (gce, gke, aws)
- Provider string
- // Tooling is the tooling in use (e.g. kops, gke). Provider is the cloud provider and might not uniquely identify the tooling.
- Tooling string
- CloudConfig CloudConfig
- KubectlPath string
- OutputDir string
- ReportDir string
- ReportPrefix string
- Prefix string
- MinStartupPods int
- // Timeout for waiting for system pods to be running
- SystemPodsStartupTimeout time.Duration
- EtcdUpgradeStorage string
- EtcdUpgradeVersion string
- GCEUpgradeScript string
- ContainerRuntime string
- ContainerRuntimeEndpoint string
- ContainerRuntimeProcessName string
- ContainerRuntimePidFile string
- // SystemdServices are comma separated list of systemd services the test framework
- // will dump logs for.
- SystemdServices string
- // DumpSystemdJournal controls whether to dump the full systemd journal.
- DumpSystemdJournal bool
- ImageServiceEndpoint string
- MasterOSDistro string
- NodeOSDistro string
- VerifyServiceAccount bool
- DeleteNamespace bool
- DeleteNamespaceOnFailure bool
- AllowedNotReadyNodes int
- CleanStart bool
- // If set to 'true' or 'all' framework will start a goroutine monitoring resource usage of system add-ons.
- // It will read the data every 30 seconds from all Nodes and print summary during afterEach. If set to 'master'
- // only master Node will be monitored.
- GatherKubeSystemResourceUsageData string
- GatherLogsSizes bool
- GatherMetricsAfterTest string
- GatherSuiteMetricsAfterTest bool
- MaxNodesToGather int
- AllowGatheringProfiles bool
- // If set to 'true' framework will gather ClusterAutoscaler metrics when gathering them for other components.
- IncludeClusterAutoscalerMetrics bool
- // Currently supported values are 'hr' for human-readable and 'json'. It's a comma separated list.
- OutputPrintType string
- // NodeSchedulableTimeout is the timeout for waiting for all nodes to be schedulable.
- NodeSchedulableTimeout time.Duration
- // SystemDaemonsetStartupTimeout is the timeout for waiting for all system daemonsets to be ready.
- SystemDaemonsetStartupTimeout time.Duration
- // CreateTestingNS is responsible for creating namespace used for executing e2e tests.
- // It accepts namespace base name, which will be prepended with e2e prefix, kube client
- // and labels to be applied to a namespace.
- CreateTestingNS CreateTestingNSFn
- // If set to true test will dump data about the namespace in which test was running.
- DumpLogsOnFailure bool
- // Disables dumping cluster log from master and nodes after all tests.
- DisableLogDump bool
- // Path to the GCS artifacts directory to dump logs from nodes. Logexporter gets enabled if this is non-empty.
- LogexporterGCSPath string
- // featureGates is a map of feature names to bools that enable or disable alpha/experimental features.
- FeatureGates map[string]bool
- // Node e2e specific test context
- NodeTestContextType
- // Indicates what path the kubernetes-anywhere is installed on
- KubernetesAnywherePath string
- // The DNS Domain of the cluster.
- ClusterDNSDomain string
- // The configuration of NodeKiller.
- NodeKiller NodeKillerConfig
- // The Default IP Family of the cluster ("ipv4" or "ipv6")
- IPFamily string
- // NonblockingTaints is the comma-delimeted string given by the user to specify taints which should not stop the test framework from running tests.
- NonblockingTaints string
- // ProgressReportURL is the URL which progress updates will be posted to as tests complete. If empty, no updates are sent.
- ProgressReportURL string
- // SriovdpConfigMapFile is the path to the ConfigMap to configure the SRIOV device plugin on this host.
- SriovdpConfigMapFile string
- }
- // NodeKillerConfig describes configuration of NodeKiller -- a utility to
- // simulate node failures.
- type NodeKillerConfig struct {
- // Enabled determines whether NodeKill should do anything at all.
- // All other options below are ignored if Enabled = false.
- Enabled bool
- // FailureRatio is a percentage of all nodes that could fail simultinously.
- FailureRatio float64
- // Interval is time between node failures.
- Interval time.Duration
- // JitterFactor is factor used to jitter node failures.
- // Node will be killed between [Interval, Interval + (1.0 + JitterFactor)].
- JitterFactor float64
- // SimulatedDowntime is a duration between node is killed and recreated.
- SimulatedDowntime time.Duration
- // NodeKillerStopCh is a channel that is used to notify NodeKiller to stop killing nodes.
- NodeKillerStopCh chan struct{}
- }
- // NodeTestContextType is part of TestContextType, it is shared by all node e2e test.
- type NodeTestContextType struct {
- // NodeE2E indicates whether it is running node e2e.
- NodeE2E bool
- // Name of the node to run tests on.
- NodeName string
- // NodeConformance indicates whether the test is running in node conformance mode.
- NodeConformance bool
- // PrepullImages indicates whether node e2e framework should prepull images.
- PrepullImages bool
- // KubeletConfig is the kubelet configuration the test is running against.
- KubeletConfig kubeletconfig.KubeletConfiguration
- // ImageDescription is the description of the image on which the test is running.
- ImageDescription string
- // SystemSpecName is the name of the system spec (e.g., gke) that's used in
- // the node e2e test. If empty, the default one (system.DefaultSpec) is
- // used. The system specs are in test/e2e_node/system/specs/.
- SystemSpecName string
- // ExtraEnvs is a map of environment names to values.
- ExtraEnvs map[string]string
- }
- // CloudConfig holds the cloud configuration for e2e test suites.
- type CloudConfig struct {
- APIEndpoint string
- ProjectID string
- Zone string // for multizone tests, arbitrarily chosen zone
- Region string
- MultiZone bool
- MultiMaster bool
- Cluster string
- MasterName string
- NodeInstanceGroup string // comma-delimited list of groups' names
- NumNodes int
- ClusterIPRange string
- ClusterTag string
- Network string
- ConfigFile string // for azure and openstack
- NodeTag string
- MasterTag string
- Provider ProviderInterface
- }
- // TestContext should be used by all tests to access common context data.
- var TestContext TestContextType
- // ClusterIsIPv6 returns true if the cluster is IPv6
- func (tc TestContextType) ClusterIsIPv6() bool {
- return tc.IPFamily == "ipv6"
- }
- // RegisterCommonFlags registers flags common to all e2e test suites.
- // The flag set can be flag.CommandLine (if desired) or a custom
- // flag set that then gets passed to viperconfig.ViperizeFlags.
- //
- // The other Register*Flags methods below can be used to add more
- // test-specific flags. However, those settings then get added
- // regardless whether the test is actually in the test suite.
- //
- // For tests that have been converted to registering their
- // options themselves, copy flags from test/e2e/framework/config
- // as shown in HandleFlags.
- func RegisterCommonFlags(flags *flag.FlagSet) {
- // Turn on verbose by default to get spec names
- config.DefaultReporterConfig.Verbose = true
- // Turn on EmitSpecProgress to get spec progress (especially on interrupt)
- config.GinkgoConfig.EmitSpecProgress = true
- // Randomize specs as well as suites
- config.GinkgoConfig.RandomizeAllSpecs = true
- flags.StringVar(&TestContext.GatherKubeSystemResourceUsageData, "gather-resource-usage", "false", "If set to 'true' or 'all' framework will be monitoring resource usage of system all add-ons in (some) e2e tests, if set to 'master' framework will be monitoring master node only, if set to 'none' of 'false' monitoring will be turned off.")
- flags.BoolVar(&TestContext.GatherLogsSizes, "gather-logs-sizes", false, "If set to true framework will be monitoring logs sizes on all machines running e2e tests.")
- flags.IntVar(&TestContext.MaxNodesToGather, "max-nodes-to-gather-from", 20, "The maximum number of nodes to gather extended info from on test failure.")
- flags.StringVar(&TestContext.GatherMetricsAfterTest, "gather-metrics-at-teardown", "false", "If set to 'true' framework will gather metrics from all components after each test. If set to 'master' only master component metrics would be gathered.")
- flags.BoolVar(&TestContext.GatherSuiteMetricsAfterTest, "gather-suite-metrics-at-teardown", false, "If set to true framwork will gather metrics from all components after the whole test suite completes.")
- flags.BoolVar(&TestContext.AllowGatheringProfiles, "allow-gathering-profiles", true, "If set to true framework will allow to gather CPU/memory allocation pprof profiles from the master.")
- flags.BoolVar(&TestContext.IncludeClusterAutoscalerMetrics, "include-cluster-autoscaler", false, "If set to true, framework will include Cluster Autoscaler when gathering metrics.")
- flags.StringVar(&TestContext.OutputPrintType, "output-print-type", "json", "Format in which summaries should be printed: 'hr' for human readable, 'json' for JSON ones.")
- flags.BoolVar(&TestContext.DumpLogsOnFailure, "dump-logs-on-failure", true, "If set to true test will dump data about the namespace in which test was running.")
- flags.BoolVar(&TestContext.DisableLogDump, "disable-log-dump", false, "If set to true, logs from master and nodes won't be gathered after test run.")
- flags.StringVar(&TestContext.LogexporterGCSPath, "logexporter-gcs-path", "", "Path to the GCS artifacts directory to dump logs from nodes. Logexporter gets enabled if this is non-empty.")
- flags.BoolVar(&TestContext.DeleteNamespace, "delete-namespace", true, "If true tests will delete namespace after completion. It is only designed to make debugging easier, DO NOT turn it off by default.")
- flags.BoolVar(&TestContext.DeleteNamespaceOnFailure, "delete-namespace-on-failure", true, "If true, framework will delete test namespace on failure. Used only during test debugging.")
- flags.IntVar(&TestContext.AllowedNotReadyNodes, "allowed-not-ready-nodes", 0, "If non-zero, framework will allow for that many non-ready nodes when checking for all ready nodes.")
- flags.StringVar(&TestContext.Host, "host", "", fmt.Sprintf("The host, or apiserver, to connect to. Will default to %s if this argument and --kubeconfig are not set", defaultHost))
- flags.StringVar(&TestContext.ReportPrefix, "report-prefix", "", "Optional prefix for JUnit XML reports. Default is empty, which doesn't prepend anything to the default name.")
- flags.StringVar(&TestContext.ReportDir, "report-dir", "", "Path to the directory where the JUnit XML reports should be saved. Default is empty, which doesn't generate these reports.")
- flags.Var(cliflag.NewMapStringBool(&TestContext.FeatureGates), "feature-gates", "A set of key=value pairs that describe feature gates for alpha/experimental features.")
- flags.StringVar(&TestContext.ContainerRuntime, "container-runtime", "docker", "The container runtime of cluster VM instances (docker/remote).")
- flags.StringVar(&TestContext.ContainerRuntimeEndpoint, "container-runtime-endpoint", "unix:///var/run/dockershim.sock", "The container runtime endpoint of cluster VM instances.")
- flags.StringVar(&TestContext.ContainerRuntimeProcessName, "container-runtime-process-name", "dockerd", "The name of the container runtime process.")
- flags.StringVar(&TestContext.ContainerRuntimePidFile, "container-runtime-pid-file", "/var/run/docker.pid", "The pid file of the container runtime.")
- flags.StringVar(&TestContext.SystemdServices, "systemd-services", "docker", "The comma separated list of systemd services the framework will dump logs for.")
- flags.BoolVar(&TestContext.DumpSystemdJournal, "dump-systemd-journal", false, "Whether to dump the full systemd journal.")
- flags.StringVar(&TestContext.ImageServiceEndpoint, "image-service-endpoint", "", "The image service endpoint of cluster VM instances.")
- flags.StringVar(&TestContext.DockershimCheckpointDir, "dockershim-checkpoint-dir", "/var/lib/dockershim/sandbox", "The directory for dockershim to store sandbox checkpoints.")
- flags.StringVar(&TestContext.KubernetesAnywherePath, "kubernetes-anywhere-path", "/workspace/k8s.io/kubernetes-anywhere", "Which directory kubernetes-anywhere is installed to.")
- flags.StringVar(&TestContext.NonblockingTaints, "non-blocking-taints", `node-role.kubernetes.io/master`, "Nodes with taints in this comma-delimited list will not block the test framework from starting tests.")
- flags.BoolVar(&TestContext.ListImages, "list-images", false, "If true, will show list of images used for runnning tests.")
- flags.StringVar(&TestContext.KubectlPath, "kubectl-path", "kubectl", "The kubectl binary to use. For development, you might use 'cluster/kubectl.sh' here.")
- flags.StringVar(&TestContext.ProgressReportURL, "progress-report-url", "", "The URL to POST progress updates to as the suite runs to assist in aiding integrations. If empty, no messages sent.")
- }
- // RegisterClusterFlags registers flags specific to the cluster e2e test suite.
- func RegisterClusterFlags(flags *flag.FlagSet) {
- flags.BoolVar(&TestContext.VerifyServiceAccount, "e2e-verify-service-account", true, "If true tests will verify the service account before running.")
- flags.StringVar(&TestContext.KubeConfig, clientcmd.RecommendedConfigPathFlag, os.Getenv(clientcmd.RecommendedConfigPathEnvVar), "Path to kubeconfig containing embedded authinfo.")
- flags.StringVar(&TestContext.KubeContext, clientcmd.FlagContext, "", "kubeconfig context to use/override. If unset, will use value from 'current-context'")
- flags.StringVar(&TestContext.KubeAPIContentType, "kube-api-content-type", "application/vnd.kubernetes.protobuf", "ContentType used to communicate with apiserver")
- flags.StringVar(&TestContext.KubeVolumeDir, "volume-dir", "/var/lib/kubelet", "Path to the directory containing the kubelet volumes.")
- flags.StringVar(&TestContext.CertDir, "cert-dir", "", "Path to the directory containing the certs. Default is empty, which doesn't use certs.")
- flags.StringVar(&TestContext.RepoRoot, "repo-root", "../../", "Root directory of kubernetes repository, for finding test files.")
- flags.StringVar(&TestContext.Provider, "provider", "", "The name of the Kubernetes provider (gce, gke, local, skeleton (the fallback if not set), etc.)")
- flags.StringVar(&TestContext.Tooling, "tooling", "", "The tooling in use (kops, gke, etc.)")
- flags.StringVar(&TestContext.OutputDir, "e2e-output-dir", "/tmp", "Output directory for interesting/useful test data, like performance data, benchmarks, and other metrics.")
- flags.StringVar(&TestContext.Prefix, "prefix", "e2e", "A prefix to be added to cloud resources created during testing.")
- flags.StringVar(&TestContext.MasterOSDistro, "master-os-distro", "debian", "The OS distribution of cluster master (debian, ubuntu, gci, coreos, or custom).")
- flags.StringVar(&TestContext.NodeOSDistro, "node-os-distro", "debian", "The OS distribution of cluster VM instances (debian, ubuntu, gci, coreos, or custom).")
- flags.StringVar(&TestContext.ClusterDNSDomain, "dns-domain", "cluster.local", "The DNS Domain of the cluster.")
- // TODO: Flags per provider? Rename gce-project/gce-zone?
- cloudConfig := &TestContext.CloudConfig
- flags.StringVar(&cloudConfig.MasterName, "kube-master", "", "Name of the kubernetes master. Only required if provider is gce or gke")
- flags.StringVar(&cloudConfig.APIEndpoint, "gce-api-endpoint", "", "The GCE APIEndpoint being used, if applicable")
- flags.StringVar(&cloudConfig.ProjectID, "gce-project", "", "The GCE project being used, if applicable")
- flags.StringVar(&cloudConfig.Zone, "gce-zone", "", "GCE zone being used, if applicable")
- flags.StringVar(&cloudConfig.Region, "gce-region", "", "GCE region being used, if applicable")
- flags.BoolVar(&cloudConfig.MultiZone, "gce-multizone", false, "If true, start GCE cloud provider with multizone support.")
- flags.BoolVar(&cloudConfig.MultiMaster, "gce-multimaster", false, "If true, the underlying GCE/GKE cluster is assumed to be multi-master.")
- flags.StringVar(&cloudConfig.Cluster, "gke-cluster", "", "GKE name of cluster being used, if applicable")
- flags.StringVar(&cloudConfig.NodeInstanceGroup, "node-instance-group", "", "Name of the managed instance group for nodes. Valid only for gce, gke or aws. If there is more than one group: comma separated list of groups.")
- flags.StringVar(&cloudConfig.Network, "network", "e2e", "The cloud provider network for this e2e cluster.")
- flags.IntVar(&cloudConfig.NumNodes, "num-nodes", DefaultNumNodes, fmt.Sprintf("Number of nodes in the cluster. If the default value of '%q' is used the number of schedulable nodes is auto-detected.", DefaultNumNodes))
- flags.StringVar(&cloudConfig.ClusterIPRange, "cluster-ip-range", "10.64.0.0/14", "A CIDR notation IP range from which to assign IPs in the cluster.")
- flags.StringVar(&cloudConfig.NodeTag, "node-tag", "", "Network tags used on node instances. Valid only for gce, gke")
- flags.StringVar(&cloudConfig.MasterTag, "master-tag", "", "Network tags used on master instances. Valid only for gce, gke")
- flags.StringVar(&cloudConfig.ClusterTag, "cluster-tag", "", "Tag used to identify resources. Only required if provider is aws.")
- flags.StringVar(&cloudConfig.ConfigFile, "cloud-config-file", "", "Cloud config file. Only required if provider is azure or vsphere.")
- flags.IntVar(&TestContext.MinStartupPods, "minStartupPods", 0, "The number of pods which we need to see in 'Running' state with a 'Ready' condition of true, before we try running tests. This is useful in any cluster which needs some base pod-based services running before it can be used.")
- flags.DurationVar(&TestContext.SystemPodsStartupTimeout, "system-pods-startup-timeout", 10*time.Minute, "Timeout for waiting for all system pods to be running before starting tests.")
- flags.DurationVar(&TestContext.NodeSchedulableTimeout, "node-schedulable-timeout", 30*time.Minute, "Timeout for waiting for all nodes to be schedulable.")
- flags.DurationVar(&TestContext.SystemDaemonsetStartupTimeout, "system-daemonsets-startup-timeout", 5*time.Minute, "Timeout for waiting for all system daemonsets to be ready.")
- flags.StringVar(&TestContext.EtcdUpgradeStorage, "etcd-upgrade-storage", "", "The storage version to upgrade to (either 'etcdv2' or 'etcdv3') if doing an etcd upgrade test.")
- flags.StringVar(&TestContext.EtcdUpgradeVersion, "etcd-upgrade-version", "", "The etcd binary version to upgrade to (e.g., '3.0.14', '2.3.7') if doing an etcd upgrade test.")
- flags.StringVar(&TestContext.GCEUpgradeScript, "gce-upgrade-script", "", "Script to use to upgrade a GCE cluster.")
- flags.BoolVar(&TestContext.CleanStart, "clean-start", false, "If true, purge all namespaces except default and system before running tests. This serves to Cleanup test namespaces from failed/interrupted e2e runs in a long-lived cluster.")
- nodeKiller := &TestContext.NodeKiller
- flags.BoolVar(&nodeKiller.Enabled, "node-killer", false, "Whether NodeKiller should kill any nodes.")
- flags.Float64Var(&nodeKiller.FailureRatio, "node-killer-failure-ratio", 0.01, "Percentage of nodes to be killed")
- flags.DurationVar(&nodeKiller.Interval, "node-killer-interval", 1*time.Minute, "Time between node failures.")
- flags.Float64Var(&nodeKiller.JitterFactor, "node-killer-jitter-factor", 60, "Factor used to jitter node failures.")
- flags.DurationVar(&nodeKiller.SimulatedDowntime, "node-killer-simulated-downtime", 10*time.Minute, "A delay between node death and recreation")
- }
- func createKubeConfig(clientCfg *restclient.Config) *clientcmdapi.Config {
- clusterNick := "cluster"
- userNick := "user"
- contextNick := "context"
- configCmd := clientcmdapi.NewConfig()
- credentials := clientcmdapi.NewAuthInfo()
- credentials.Token = clientCfg.BearerToken
- credentials.TokenFile = clientCfg.BearerTokenFile
- credentials.ClientCertificate = clientCfg.TLSClientConfig.CertFile
- if len(credentials.ClientCertificate) == 0 {
- credentials.ClientCertificateData = clientCfg.TLSClientConfig.CertData
- }
- credentials.ClientKey = clientCfg.TLSClientConfig.KeyFile
- if len(credentials.ClientKey) == 0 {
- credentials.ClientKeyData = clientCfg.TLSClientConfig.KeyData
- }
- configCmd.AuthInfos[userNick] = credentials
- cluster := clientcmdapi.NewCluster()
- cluster.Server = clientCfg.Host
- cluster.CertificateAuthority = clientCfg.CAFile
- if len(cluster.CertificateAuthority) == 0 {
- cluster.CertificateAuthorityData = clientCfg.CAData
- }
- cluster.InsecureSkipTLSVerify = clientCfg.Insecure
- configCmd.Clusters[clusterNick] = cluster
- context := clientcmdapi.NewContext()
- context.Cluster = clusterNick
- context.AuthInfo = userNick
- configCmd.Contexts[contextNick] = context
- configCmd.CurrentContext = contextNick
- return configCmd
- }
- // AfterReadingAllFlags makes changes to the context after all flags
- // have been read.
- func AfterReadingAllFlags(t *TestContextType) {
- // Only set a default host if one won't be supplied via kubeconfig
- if len(t.Host) == 0 && len(t.KubeConfig) == 0 {
- // Check if we can use the in-cluster config
- if clusterConfig, err := restclient.InClusterConfig(); err == nil {
- if tempFile, err := ioutil.TempFile(os.TempDir(), "kubeconfig-"); err == nil {
- kubeConfig := createKubeConfig(clusterConfig)
- clientcmd.WriteToFile(*kubeConfig, tempFile.Name())
- t.KubeConfig = tempFile.Name()
- klog.Infof("Using a temporary kubeconfig file from in-cluster config : %s", tempFile.Name())
- }
- }
- if len(t.KubeConfig) == 0 {
- klog.Warningf("Unable to find in-cluster config, using default host : %s", defaultHost)
- t.Host = defaultHost
- }
- }
- // Allow 1% of nodes to be unready (statistically) - relevant for large clusters.
- if t.AllowedNotReadyNodes == 0 {
- t.AllowedNotReadyNodes = t.CloudConfig.NumNodes / 100
- }
- klog.Infof("Tolerating taints %q when considering if nodes are ready", TestContext.NonblockingTaints)
- // Make sure that all test runs have a valid TestContext.CloudConfig.Provider.
- // TODO: whether and how long this code is needed is getting discussed
- // in https://github.com/kubernetes/kubernetes/issues/70194.
- if TestContext.Provider == "" {
- // Some users of the e2e.test binary pass --provider=.
- // We need to support that, changing it would break those usages.
- Logf("The --provider flag is not set. Continuing as if --provider=skeleton had been used.")
- TestContext.Provider = "skeleton"
- }
- var err error
- TestContext.CloudConfig.Provider, err = SetupProviderConfig(TestContext.Provider)
- if err != nil {
- if os.IsNotExist(errors.Cause(err)) {
- // Provide a more helpful error message when the provider is unknown.
- var providers []string
- for _, name := range GetProviders() {
- // The empty string is accepted, but looks odd in the output below unless we quote it.
- if name == "" {
- name = `""`
- }
- providers = append(providers, name)
- }
- sort.Strings(providers)
- klog.Errorf("Unknown provider %q. The following providers are known: %v", TestContext.Provider, strings.Join(providers, " "))
- } else {
- klog.Errorf("Failed to setup provider config for %q: %v", TestContext.Provider, err)
- }
- os.Exit(1)
- }
- }
|