talos/internal/integration/api/version.go
Andrey Smirnov 7f0300f108
feat: update dependencies, Kubernetes 1.34.0-alpha.2
Bump all dependencies, many small changes due to new golangci-lint
version.

Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
2025-06-30 19:05:22 +04:00

100 lines
2.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/.
//go:build integration_api
package api
import (
"context"
"time"
"github.com/cosi-project/runtime/pkg/safe"
"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"
configmachine "github.com/siderolabs/talos/pkg/machinery/config/machine"
"github.com/siderolabs/talos/pkg/machinery/resources/runtime"
)
// 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)
node := suite.RandomDiscoveredNodeInternalIP(configmachine.TypeControlPlane)
clientCtx := client.WithNode(suite.ctx, node)
version, err := safe.StateGetByID[*runtime.Version](clientCtx, suite.Client.COSI, "version")
suite.Require().NoError(err)
suite.Require().NotNil(version)
suite.Assert().Equal(v.Messages[0].Version.Tag, version.TypedSpec().Version)
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))
}