mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-30 16:01:12 +01:00
96aa9638f7
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>
87 lines
2.1 KiB
Go
87 lines
2.1 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_api
|
|
|
|
package api
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/siderolabs/go-retry/retry"
|
|
|
|
"github.com/siderolabs/talos/internal/integration/base"
|
|
"github.com/siderolabs/talos/pkg/machinery/api/machine"
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
)
|
|
|
|
// VersionSuite verifies version API.
|
|
type VersionSuite struct {
|
|
base.APISuite
|
|
|
|
ctx context.Context //nolint:containedctx
|
|
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)
|
|
suite.Assert().Equal(suite.GoVersion, v.Messages[0].Version.GoVersion)
|
|
}
|
|
|
|
// TestSameVersionCluster verifies that all the nodes are on the same version.
|
|
func (suite *VersionSuite) TestSameVersionCluster() {
|
|
nodes := suite.DiscoverNodeInternalIPs(suite.ctx)
|
|
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))
|
|
}
|