mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 08:21:25 +01:00
a059454045
`initramfs` size for amd64 shrinks by 1.3 MiB. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
98 lines
2.6 KiB
Go
98 lines
2.6 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_cli
|
|
// +build integration_cli
|
|
|
|
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/talos-systems/talos/internal/integration/base"
|
|
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
|
|
)
|
|
|
|
// HealthSuite verifies health command.
|
|
type HealthSuite struct {
|
|
base.CLISuite
|
|
}
|
|
|
|
// SuiteName ...
|
|
func (suite *HealthSuite) SuiteName() string {
|
|
return "cli.HealthSuite"
|
|
}
|
|
|
|
// TestClientSide does successful health check run from client-side.
|
|
//
|
|
//nolint:gocyclo
|
|
func (suite *HealthSuite) TestClientSide() {
|
|
if suite.Cluster == nil {
|
|
suite.T().Skip("Cluster is not available, skipping test")
|
|
}
|
|
|
|
args := []string{"--server=false"}
|
|
|
|
bootstrapAPIIsUsed := true
|
|
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
if node.Type == machine.TypeInit {
|
|
bootstrapAPIIsUsed = false
|
|
}
|
|
}
|
|
|
|
if bootstrapAPIIsUsed {
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
switch node.Type {
|
|
case machine.TypeControlPlane:
|
|
args = append(args, "--control-plane-nodes", node.IPs[0].String())
|
|
case machine.TypeWorker:
|
|
args = append(args, "--worker-nodes", node.IPs[0].String())
|
|
case machine.TypeInit, machine.TypeUnknown:
|
|
fallthrough
|
|
default:
|
|
panic(fmt.Sprintf("unexpected machine type %v", node.Type))
|
|
}
|
|
}
|
|
} else {
|
|
for _, node := range suite.Cluster.Info().Nodes {
|
|
switch node.Type {
|
|
case machine.TypeInit:
|
|
args = append(args, "--init-node", node.IPs[0].String())
|
|
case machine.TypeControlPlane:
|
|
args = append(args, "--control-plane-nodes", node.IPs[0].String())
|
|
case machine.TypeWorker:
|
|
args = append(args, "--worker-nodes", node.IPs[0].String())
|
|
case machine.TypeUnknown:
|
|
fallthrough
|
|
default:
|
|
panic(fmt.Sprintf("unexpected machine type %v", node.Type))
|
|
}
|
|
}
|
|
}
|
|
|
|
if suite.K8sEndpoint != "" {
|
|
args = append(args, "--k8s-endpoint", strings.Split(suite.K8sEndpoint, ":")[0])
|
|
}
|
|
|
|
suite.RunCLI(append([]string{"health"}, args...),
|
|
base.StdoutEmpty(),
|
|
base.StderrShouldMatch(regexp.MustCompile(`waiting for all k8s nodes to report ready`)),
|
|
)
|
|
}
|
|
|
|
// TestServerSide does successful health check run from server-side.
|
|
func (suite *HealthSuite) TestServerSide() {
|
|
suite.RunCLI([]string{"health", "--nodes", suite.RandomDiscoveredNode(machine.TypeControlPlane)},
|
|
base.StdoutEmpty(),
|
|
base.StderrShouldMatch(regexp.MustCompile(`waiting for all k8s nodes to report ready`)),
|
|
)
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(HealthSuite))
|
|
}
|