Files
talos/internal/integration/api/generate-config.go
Artem Chernyshev 8513123d22 feat: return client config as the second value in GenerateConfiguration
To be used in interactive installer to output the node client
configuration to a file.

Signed-off-by: Artem Chernyshev <artem.0xD2@gmail.com>
2020-11-17 07:20:05 -08:00

116 lines
3.8 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"
"fmt"
"time"
"github.com/talos-systems/talos/internal/integration/base"
"github.com/talos-systems/talos/pkg/images"
machineapi "github.com/talos-systems/talos/pkg/machinery/api/machine"
clientconfig "github.com/talos-systems/talos/pkg/machinery/client/config"
"github.com/talos-systems/talos/pkg/machinery/config/configloader"
"github.com/talos-systems/talos/pkg/machinery/config/types/v1alpha1/machine"
"github.com/talos-systems/talos/pkg/machinery/constants"
)
// GenerateConfigSuite ...
type GenerateConfigSuite struct {
base.K8sSuite
ctx context.Context
ctxCancel context.CancelFunc
}
// SuiteName ...
func (suite *GenerateConfigSuite) SuiteName() string {
return "api.GenerateConfigSuite"
}
// SetupTest ...
func (suite *GenerateConfigSuite) SetupTest() {
// make sure we abort at some point in time, but give enough room for Recovers
suite.ctx, suite.ctxCancel = context.WithTimeout(context.Background(), 30*time.Minute)
}
// TearDownTest ...
func (suite *GenerateConfigSuite) TearDownTest() {
if suite.ctxCancel != nil {
suite.ctxCancel()
}
}
// TestGenerate verifies the generate config API.
func (suite *GenerateConfigSuite) TestGenerate() {
request := &machineapi.GenerateConfigurationRequest{
ConfigVersion: "v1alpha1",
MachineConfig: &machineapi.MachineConfig{
Type: machineapi.MachineConfig_MachineType(machine.TypeInit),
NetworkConfig: &machineapi.NetworkConfig{
Hostname: "testhost",
},
InstallConfig: &machineapi.InstallConfig{
InstallDisk: "/dev/sdb",
InstallImage: images.DefaultInstallerImage,
},
KubernetesVersion: constants.DefaultKubernetesVersion,
},
ClusterConfig: &machineapi.ClusterConfig{
Name: "talos-default",
ControlPlane: &machineapi.ControlPlaneConfig{
Endpoint: "http://localhost",
},
ClusterNetwork: &machineapi.ClusterNetworkConfig{
DnsDomain: "cluster.test",
},
},
}
reply, err := suite.Client.GenerateConfiguration(
suite.ctx,
request,
)
suite.Require().NoError(err)
data := reply.GetData()
config, err := configloader.NewFromBytes(data[0])
suite.Require().NoError(err)
suite.Require().EqualValues(request.MachineConfig.Type, config.Machine().Type())
suite.Require().EqualValues(request.ConfigVersion, config.Version())
suite.Require().EqualValues(request.ClusterConfig.Name, config.Cluster().Name())
suite.Require().EqualValues(request.ClusterConfig.ControlPlane.Endpoint, config.Cluster().Endpoint().String())
suite.Require().EqualValues(request.ClusterConfig.ClusterNetwork.DnsDomain, config.Cluster().Network().DNSDomain())
suite.Require().EqualValues(fmt.Sprintf("%s:v%s", constants.KubeletImage, request.MachineConfig.KubernetesVersion), config.Machine().Kubelet().Image())
suite.Require().EqualValues(request.MachineConfig.InstallConfig.InstallDisk, config.Machine().Install().Disk())
suite.Require().EqualValues(request.MachineConfig.InstallConfig.InstallImage, config.Machine().Install().Image())
suite.Require().EqualValues(request.MachineConfig.NetworkConfig.Hostname, config.Machine().Network().Hostname())
talosconfig, err := clientconfig.FromBytes(reply.Talosconfig)
suite.Require().NoError(err)
context := talosconfig.Contexts[request.ClusterConfig.Name]
suite.Require().NotNil(context)
suite.Require().NotEmpty(context.CA)
suite.Require().NotEmpty(context.Crt)
suite.Require().NotEmpty(context.Key)
suite.Require().Greater(len(context.Endpoints), 0)
suite.Require().EqualValues(context.Endpoints[0], config.Cluster().Endpoint().Hostname())
}
func init() {
allSuites = append(allSuites, new(GenerateConfigSuite))
}