mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-08 07:37:06 +02:00
30860210cc
On e.g. Azure VMs there are non reported. Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
87 lines
2.4 KiB
Go
87 lines
2.4 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/resource"
|
|
"github.com/cosi-project/runtime/pkg/safe"
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/siderolabs/talos/internal/integration/base"
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
"github.com/siderolabs/talos/pkg/machinery/resources/hardware"
|
|
)
|
|
|
|
// HardwareSuite ...
|
|
type HardwareSuite struct {
|
|
base.APISuite
|
|
|
|
ctx context.Context //nolint:containedctx
|
|
ctxCancel context.CancelFunc
|
|
}
|
|
|
|
// SuiteName ...
|
|
func (suite *HardwareSuite) SuiteName() string {
|
|
return "api.HardwareSuite"
|
|
}
|
|
|
|
// SetupTest ...
|
|
func (suite *HardwareSuite) SetupTest() {
|
|
if !suite.Capabilities().RunsTalosKernel {
|
|
suite.T().Skipf("doesn't run Talos kernel, skipping")
|
|
}
|
|
|
|
suite.ctx, suite.ctxCancel = context.WithTimeout(context.Background(), 15*time.Second)
|
|
}
|
|
|
|
// TearDownTest ...
|
|
func (suite *HardwareSuite) TearDownTest() {
|
|
if suite.ctxCancel != nil {
|
|
suite.ctxCancel()
|
|
}
|
|
}
|
|
|
|
// TestSystemInformation tests that SystemInformation is populated.
|
|
func (suite *HardwareSuite) TestSystemInformation() {
|
|
node := suite.RandomDiscoveredNodeInternalIP()
|
|
|
|
sysInfo, err := safe.StateGetByID[*hardware.SystemInformation](client.WithNode(suite.ctx, node), suite.Client.COSI, hardware.SystemInformationID)
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Assert().NotEmpty(sysInfo.TypedSpec().UUID)
|
|
suite.Assert().NotEqual((uuid.UUID{}).String(), sysInfo.TypedSpec().UUID)
|
|
}
|
|
|
|
// TestHardwareInfo tests that hardware info is populated.
|
|
func (suite *HardwareSuite) TestHardwareInfo() {
|
|
node := suite.RandomDiscoveredNodeInternalIP()
|
|
|
|
resourceList := []resource.Type{
|
|
hardware.MemoryModuleType,
|
|
hardware.ProcessorType,
|
|
}
|
|
|
|
if suite.Cluster != nil {
|
|
// cloud VMs might not publish PCI devices
|
|
resourceList = append(resourceList, hardware.PCIDeviceType)
|
|
}
|
|
|
|
for _, resourceType := range resourceList {
|
|
items, err := suite.Client.COSI.List(client.WithNode(suite.ctx, node), resource.NewMetadata(hardware.NamespaceName, resourceType, "", resource.VersionUndefined))
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Assert().NotEmpty(items.Items, "resource type %s is not populated", resourceType)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
allSuites = append(allSuites, new(HardwareSuite))
|
|
}
|