mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-05 12:41:31 +02:00
d0d2ac3c74
This moves our test scripts to using the bootstrap API. Some automation around invoking the bootstrap API was also added to give the same ease of use when creating clusters with the CLI. Signed-off-by: Andrew Rynhard <andrew@andrewrynhard.com>
92 lines
2.2 KiB
Go
92 lines
2.2 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_cli
|
|
|
|
package cli
|
|
|
|
import (
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/talos-systems/talos/internal/app/machined/pkg/runtime"
|
|
"github.com/talos-systems/talos/internal/integration/base"
|
|
)
|
|
|
|
// HealthSuite verifies health command
|
|
type HealthSuite struct {
|
|
base.CLISuite
|
|
}
|
|
|
|
// SuiteName ...
|
|
func (suite *HealthSuite) SuiteName() string {
|
|
return "cli.HealthSuite"
|
|
}
|
|
|
|
// TestRun does successful health check run.
|
|
func (suite *HealthSuite) TestRun() {
|
|
if suite.Cluster == nil {
|
|
suite.T().Skip("Cluster is not available, skipping test")
|
|
}
|
|
|
|
args := []string{}
|
|
|
|
bootstrapAPIIsUsed := true
|
|
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
if node.Type == runtime.MachineTypeInit {
|
|
bootstrapAPIIsUsed = false
|
|
}
|
|
}
|
|
|
|
if bootstrapAPIIsUsed {
|
|
nodes := []string{}
|
|
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
switch node.Type {
|
|
case runtime.MachineTypeControlPlane:
|
|
nodes = append(nodes, node.PrivateIP.String())
|
|
case runtime.MachineTypeJoin:
|
|
args = append(args, "--worker-nodes", node.PrivateIP.String())
|
|
}
|
|
}
|
|
|
|
sort.Strings(nodes)
|
|
|
|
if len(nodes) > 0 {
|
|
args = append(args, "--init-node", nodes[0])
|
|
}
|
|
|
|
if len(nodes) > 1 {
|
|
args = append(args, "--control-plane-nodes", strings.Join(nodes[1:], ","))
|
|
}
|
|
} else {
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
switch node.Type {
|
|
case runtime.MachineTypeInit:
|
|
args = append(args, "--init-node", node.PrivateIP.String())
|
|
case runtime.MachineTypeControlPlane:
|
|
args = append(args, "--control-plane-nodes", node.PrivateIP.String())
|
|
case runtime.MachineTypeJoin:
|
|
args = append(args, "--worker-nodes", node.PrivateIP.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
if suite.K8sEndpoint != "" {
|
|
args = append(args, "--k8s-endpoint", strings.Split(suite.K8sEndpoint, ":")[0])
|
|
}
|
|
|
|
suite.RunCLI(append([]string{"health"}, args...),
|
|
base.StderrNotEmpty(),
|
|
base.StdoutEmpty(),
|
|
base.StderrShouldMatch(regexp.MustCompile(`waiting for all k8s nodes to report ready`)),
|
|
)
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(HealthSuite))
|
|
}
|