Andrey Smirnov 6a81f30941 test: provide node discovery for cli tests via kubectl
Fixes #2330

CLI tests require node discovery as `--nodes` flag is enforced for most
of the `talosctl commands`.

For clusters created via `talosctl cluster create`, cluster provisioner
state provides all the necessary information, but clusters created via
CAPI don't have the state attached.

API tests rely on Talos and Kubernetes APIs to fetch kubeconfig and
access Nodes K8s API.

CLI tests should rely only on CLI tools, so we use `kubectl get nodes` +
`talosctl kubeconfig` to fetch list of master and worker nodes.

This discovery method relies on "bootstrap" node being set in
`talosconfig` (to fetch `kubeconfig`).

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-07-28 11:35:47 -07:00

63 lines
1.9 KiB
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
// +build integration
// Package base provides shared definition of base suites for tests
package base
import (
"github.com/talos-systems/talos/internal/pkg/cluster"
"github.com/talos-systems/talos/internal/pkg/provision"
"github.com/talos-systems/talos/internal/pkg/provision/access"
)
// TalosSuite defines most common settings for integration test suites.
type TalosSuite struct {
// Endpoint to use to connect, if not set config is used
Endpoint string
// K8sEndpoint is API server endpoint, if set overrides kubeconfig
K8sEndpoint string
// Cluster describes provisioned cluster, used for discovery purposes
Cluster provision.Cluster
// TalosConfig is a path to talosconfig
TalosConfig string
// Version is the (expected) version of Talos tests are running against
Version string
// TalosctlPath is a path to talosctl binary
TalosctlPath string
// KubectlPath is a path to kubectl binary
KubectlPath string
discoveredNodes cluster.Info
}
// DiscoverNodes provides basic functionality to discover cluster nodes via test settings.
//
// This method is overridden in specific suites to allow for specific discovery.
func (talosSuite *TalosSuite) DiscoverNodes() cluster.Info {
if talosSuite.discoveredNodes == nil {
if talosSuite.Cluster != nil {
talosSuite.discoveredNodes = access.NewAdapter(talosSuite.Cluster).Info
}
}
return talosSuite.discoveredNodes
}
// ConfiguredSuite expects config to be set before running.
type ConfiguredSuite interface {
SetConfig(config TalosSuite)
}
// SetConfig implements ConfiguredSuite.
func (suite *TalosSuite) SetConfig(config TalosSuite) {
*suite = config
}
// NamedSuite interface provides names for test suites.
type NamedSuite interface {
SuiteName() string
}