talos/internal/integration/base/base.go
Andrey Smirnov 96aa9638f7
chore: rename talos-systems/talos to siderolabs/talos
There's a cyclic dependency on siderolink library which imports talos
machinery back. We will fix that after we get talos pushed under a new
name.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2022-11-03 16:50:32 +04:00

67 lines
2.0 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/.
//go:build integration
// Package base provides shared definition of base suites for tests
package base
import (
"context"
"github.com/siderolabs/talos/pkg/cluster"
"github.com/siderolabs/talos/pkg/provision"
"github.com/siderolabs/talos/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
// GoVersion is the (expected) version of Go compiler.
GoVersion 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(_ context.Context) 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 (talosSuite *TalosSuite) SetConfig(config TalosSuite) {
*talosSuite = config
}
// NamedSuite interface provides names for test suites.
type NamedSuite interface {
SuiteName() string
}