mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-11 17:17:05 +02:00
a167a54021
When integration tests run without data from Talos provisioner (e.g. against AWS/GCP), it should work only with `talosconfig` as an input. This specific flow was missing filling out `infoWrapper` properly. Clean up things a bit by reducing code duplication. Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
47 lines
1.3 KiB
Go
47 lines
1.3 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
|
|
// +build integration
|
|
|
|
package base
|
|
|
|
import (
|
|
"github.com/talos-systems/talos/pkg/cluster"
|
|
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
|
|
)
|
|
|
|
type infoWrapper struct {
|
|
nodeInfos []cluster.NodeInfo
|
|
nodeInfosByType map[machine.Type][]cluster.NodeInfo
|
|
}
|
|
|
|
func newNodeInfo(masterNodes, workerNodes []string) (*infoWrapper, error) {
|
|
controlPlaneNodeInfos, err := cluster.IPsToNodeInfos(masterNodes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
workerNodeInfos, err := cluster.IPsToNodeInfos(workerNodes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &infoWrapper{
|
|
nodeInfos: append(append([]cluster.NodeInfo(nil), controlPlaneNodeInfos...), workerNodeInfos...),
|
|
nodeInfosByType: map[machine.Type][]cluster.NodeInfo{
|
|
machine.TypeControlPlane: controlPlaneNodeInfos,
|
|
machine.TypeWorker: workerNodeInfos,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (wrapper *infoWrapper) Nodes() []cluster.NodeInfo {
|
|
return wrapper.nodeInfos
|
|
}
|
|
|
|
func (wrapper *infoWrapper) NodesByType(t machine.Type) []cluster.NodeInfo {
|
|
return wrapper.nodeInfosByType[t]
|
|
}
|