mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-21 14:41:12 +02:00
7f3dca8e4c
Modify provision library to support multiple IPs, CIDRs, gateways, which can be IPv4/IPv6. Based on IP types, enable services in the cluster to run DHCPv4/DHCPv6 in the test environment. There's outstanding bug left with routes not being properly set up in the cluster so, IPs are not properly routable, but DHCPv6 works and IPs are allocated (validates DHCPv6 client). Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
94 lines
2.5 KiB
Go
94 lines
2.5 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"
|
|
"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.TypeJoin:
|
|
args = append(args, "--worker-nodes", node.IPs[0].String())
|
|
case machine.TypeInit, machine.TypeUnknown:
|
|
panic("unexpected")
|
|
}
|
|
}
|
|
} 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.TypeJoin:
|
|
args = append(args, "--worker-nodes", node.IPs[0].String())
|
|
case machine.TypeUnknown:
|
|
panic("unexpected")
|
|
}
|
|
}
|
|
}
|
|
|
|
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`)),
|
|
)
|
|
}
|
|
|
|
// TestServerSide does successful health check run from server-side.
|
|
func (suite *HealthSuite) TestServerSide() {
|
|
suite.RunCLI([]string{"health", "--nodes", suite.RandomDiscoveredNode(machine.TypeControlPlane)},
|
|
base.StderrNotEmpty(),
|
|
base.StdoutEmpty(),
|
|
base.StderrShouldMatch(regexp.MustCompile(`waiting for all k8s nodes to report ready`)),
|
|
)
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(HealthSuite))
|
|
}
|