talos/internal/integration/cli/reboot.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

85 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/.
//go:build integration_cli
// +build integration_cli
package cli
import (
"fmt"
"strings"
"testing"
"github.com/hashicorp/go-multierror"
"github.com/siderolabs/talos/internal/integration/base"
"github.com/siderolabs/talos/pkg/machinery/config/types/v1alpha1/machine"
)
// RebootSuite tests reboot command.
type RebootSuite struct {
base.CLISuite
}
// SuiteName ...
func (suite *RebootSuite) SuiteName() string {
return "cli.RebootSuite"
}
// TestReboot tests rebooting nodes.
func (suite *RebootSuite) TestReboot() {
if testing.Short() {
suite.T().Skip("skipping in short mode")
}
controlPlaneNode := suite.RandomDiscoveredNodeInternalIP(machine.TypeControlPlane)
workerNode := suite.RandomDiscoveredNodeInternalIP(machine.TypeWorker)
nodes := fmt.Sprintf(
"%s,%s",
controlPlaneNode,
workerNode,
)
suite.T().Logf("rebooting nodes %s via the CLI", nodes)
suite.RunCLI([]string{"reboot", "-n", nodes, "--debug"}, base.StdoutMatchFunc(func(stdout string) error {
if strings.Contains(stdout, "method is not supported") {
suite.T().Skip("reboot is not supported")
}
var err error
for _, node := range []string{controlPlaneNode, workerNode} {
if !strings.Contains(stdout, fmt.Sprintf("%v: events check condition met", node)) {
err = multierror.Append(err, fmt.Errorf("events check condition not met on %v", node))
}
if !strings.Contains(stdout, fmt.Sprintf("%v: post check passed", node)) {
err = multierror.Append(err, fmt.Errorf("post check not passed on %v", node))
}
}
return err
}))
suite.T().Logf("running the cluster health check")
// run the health check to make sure cluster is fully healthy after a node reboot
args := []string{"--server=false"}
if suite.K8sEndpoint != "" {
args = append(args, "--k8s-endpoint", strings.Split(suite.K8sEndpoint, ":")[0])
}
suite.RunCLI(append([]string{"health"}, args...),
base.StdoutEmpty(),
base.StderrNotEmpty(),
)
}
func init() {
allSuites = append(allSuites, new(RebootSuite))
}