mirror of
https://github.com/siderolabs/talos.git
synced 2025-11-04 02:11:12 +01:00
773912833e
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>
86 lines
2.0 KiB
Go
86 lines
2.0 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_api
|
|
|
|
package api
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/talos-systems/go-retry/retry"
|
|
|
|
"github.com/talos-systems/talos/internal/integration/base"
|
|
"github.com/talos-systems/talos/pkg/machinery/api/machine"
|
|
"github.com/talos-systems/talos/pkg/machinery/client"
|
|
)
|
|
|
|
// VersionSuite verifies version API.
|
|
type VersionSuite struct {
|
|
base.APISuite
|
|
|
|
ctx context.Context
|
|
ctxCancel context.CancelFunc
|
|
}
|
|
|
|
// SuiteName ...
|
|
func (suite *VersionSuite) SuiteName() string {
|
|
return "api.VersionSuite"
|
|
}
|
|
|
|
// SetupTest ...
|
|
func (suite *VersionSuite) SetupTest() {
|
|
// make sure API calls have timeout
|
|
suite.ctx, suite.ctxCancel = context.WithTimeout(context.Background(), 2*time.Minute)
|
|
}
|
|
|
|
// TearDownTest ...
|
|
func (suite *VersionSuite) TearDownTest() {
|
|
if suite.ctxCancel != nil {
|
|
suite.ctxCancel()
|
|
}
|
|
}
|
|
|
|
// TestExpectedVersionMaster verifies master node version matches expected.
|
|
func (suite *VersionSuite) TestExpectedVersionMaster() {
|
|
v, err := suite.Client.Version(suite.ctx)
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Assert().Equal(suite.Version, v.Messages[0].Version.Tag)
|
|
}
|
|
|
|
// TestSameVersionCluster verifies that all the nodes are on the same version.
|
|
func (suite *VersionSuite) TestSameVersionCluster() {
|
|
nodes := suite.DiscoverNodes().Nodes()
|
|
suite.Require().NotEmpty(nodes)
|
|
|
|
ctx := client.WithNodes(suite.ctx, nodes...)
|
|
|
|
var v *machine.VersionResponse
|
|
|
|
err := retry.Constant(
|
|
time.Minute,
|
|
).Retry(func() error {
|
|
var e error
|
|
v, e = suite.Client.Version(ctx)
|
|
|
|
return retry.ExpectedError(e)
|
|
})
|
|
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Require().Len(v.Messages, len(nodes))
|
|
|
|
expectedVersion := v.Messages[0].Version.Tag
|
|
|
|
for _, version := range v.Messages {
|
|
suite.Assert().Equal(expectedVersion, version.Version.Tag)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(VersionSuite))
|
|
}
|