mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-31 11:31:10 +02:00
5ee554128e
The gofumports does everything that gofumpt does with the addition of formatting imports. This change proposes the use of the `-local` flag so that we can have imports separated in the following order: - standard library - third party - Talos specific Signed-off-by: Andrew Rynhard <andrew@andrewrynhard.com>
64 lines
1.8 KiB
Go
64 lines
1.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/. */
|
|
|
|
package generate_test
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
"gopkg.in/yaml.v2"
|
|
|
|
v1alpha1 "github.com/talos-systems/talos/pkg/userdata/v1alpha1"
|
|
udgenv1alpha1 "github.com/talos-systems/talos/pkg/userdata/v1alpha1/generate"
|
|
)
|
|
|
|
var input *udgenv1alpha1.Input
|
|
|
|
type GenerateSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func TestGenerateSuite(t *testing.T) {
|
|
suite.Run(t, new(GenerateSuite))
|
|
}
|
|
|
|
func (suite *GenerateSuite) SetupSuite() {
|
|
var err error
|
|
input, err = udgenv1alpha1.NewInput("test", []string{"10.0.1.5", "10.0.1.6", "10.0.1.7"})
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *GenerateSuite) TestGenerateInitSuccess() {
|
|
input.IP = net.ParseIP("10.0.1.5")
|
|
dataString, err := udgenv1alpha1.Userdata(udgenv1alpha1.TypeInit, input)
|
|
suite.Require().NoError(err)
|
|
data := &v1alpha1.NodeConfig{}
|
|
err = yaml.Unmarshal([]byte(dataString), data)
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *GenerateSuite) TestGenerateControlPlaneSuccess() {
|
|
input.IP = net.ParseIP("10.0.1.6")
|
|
dataString, err := udgenv1alpha1.Userdata(udgenv1alpha1.TypeControlPlane, input)
|
|
suite.Require().NoError(err)
|
|
data := &v1alpha1.NodeConfig{}
|
|
err = yaml.Unmarshal([]byte(dataString), data)
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *GenerateSuite) TestGenerateWorkerSuccess() {
|
|
dataString, err := udgenv1alpha1.Userdata(udgenv1alpha1.TypeJoin, input)
|
|
suite.Require().NoError(err)
|
|
data := &v1alpha1.NodeConfig{}
|
|
err = yaml.Unmarshal([]byte(dataString), data)
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *GenerateSuite) TestGenerateTalosconfigSuccess() {
|
|
_, err := udgenv1alpha1.Talosconfig(input)
|
|
suite.Require().NoError(err)
|
|
}
|