talos/internal/integration/cli/restart.go
Andrey Smirnov 773912833e test: clean up integration test code, fix flakes
This enables golangci-lint via build tags for integration tests (this
should have been done long ago!), and fixes the linting errors.

Two tests were updated to reduce flakiness:

* apply config: wait for nodes to issue "boot done" sequence event
before proceeding
* recover: kill pods even if they appear after the initial set gets
killed (potential race condition with previous test).

Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
2020-10-19 15:44:14 -07:00

64 lines
1.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/.
// +build integration_cli
package cli
import (
"regexp"
"testing"
"time"
"github.com/talos-systems/talos/internal/integration/base"
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
)
// RestartSuite verifies dmesg command.
type RestartSuite struct {
base.CLISuite
}
// SuiteName ...
func (suite *RestartSuite) SuiteName() string {
return "cli.RestartSuite"
}
// TestSystem restarts system containerd process.
func (suite *RestartSuite) TestSystem() {
if testing.Short() {
suite.T().Skip("skipping in short mode")
}
// trustd only runs on control plane nodes
node := suite.RandomDiscoveredNode(machine.TypeControlPlane)
suite.RunCLI([]string{"restart", "-n", node, "trustd"},
base.StdoutEmpty())
time.Sleep(200 * time.Millisecond)
suite.RunAndWaitForMatch([]string{"service", "-n", node, "trustd"}, regexp.MustCompile(`EVENTS\s+\[Running\]: Health check successful`), 30*time.Second)
}
// TestK8s restarts K8s container.
func (suite *RestartSuite) TestK8s() {
if testing.Short() {
suite.T().Skip("skipping in short mode")
}
node := suite.RandomDiscoveredNode()
suite.RunCLI([]string{"restart", "-n", node, "-k", "kubelet"},
base.StdoutEmpty())
time.Sleep(200 * time.Millisecond)
suite.RunAndWaitForMatch([]string{"service", "-n", node, "kubelet"}, regexp.MustCompile(`EVENTS\s+\[Running\]: Health check successful`), 30*time.Second)
}
func init() {
allSuites = append(allSuites, new(RestartSuite))
}