mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-07 23:27:07 +02:00
fa3b933705
This time use `eg` from `x/tools` repo tool to do this. Signed-off-by: Dmitriy Matrenichev <dmitry.matrenichev@siderolabs.com>
116 lines
3.3 KiB
Go
116 lines
3.3 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 (
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-multierror"
|
|
|
|
"github.com/siderolabs/talos/internal/integration/base"
|
|
"github.com/siderolabs/talos/pkg/machinery/config/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.StdoutEmpty(),
|
|
base.StderrNotEmpty(),
|
|
base.StderrMatchFunc(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("%q: 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("%q: 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(),
|
|
)
|
|
}
|
|
|
|
// TestRebootEarlyFailPrintsOutput tests the action tracker used by reboot command to track reboot status
|
|
// does not suppress the stderr output if there is an error occurring at an early stage, i.e. before the
|
|
// action status reporting starts.
|
|
func (suite *RebootSuite) TestRebootEarlyFailPrintsOutput() {
|
|
controlPlaneNode := suite.RandomDiscoveredNodeInternalIP(machine.TypeControlPlane)
|
|
invalidTalosconfig := filepath.Join(suite.T().TempDir(), "talosconfig.yaml")
|
|
|
|
suite.T().Logf("attempting to reboot node %q using talosconfig %q", controlPlaneNode, invalidTalosconfig)
|
|
|
|
suite.RunCLI([]string{"--talosconfig", invalidTalosconfig, "reboot", "-n", controlPlaneNode},
|
|
base.ShouldFail(),
|
|
base.StdoutEmpty(),
|
|
base.StderrNotEmpty(),
|
|
base.StderrMatchFunc(func(stdout string) error {
|
|
if strings.Contains(stdout, "method is not supported") {
|
|
suite.T().Skip("reboot is not supported")
|
|
}
|
|
|
|
if !strings.Contains(stdout, "failed to determine endpoints") {
|
|
return errors.New("expected to find 'failed to determine endpoints' in stderr")
|
|
}
|
|
|
|
return nil
|
|
}))
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(RebootSuite))
|
|
}
|