talos/pkg/argsbuilder/argsbuilder_test.go
Andrey Smirnov 91a858b537
fix: sort output of the argument builder
This fixes the instabilitiy on some of the internal resources, as they
get regenerated as a result of machine config changes. As map iteration
order is not stable this might cause unexpected static pod defition
regeneration with the only difference is the flag order.

Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
2021-10-08 19:31:21 +03:00

110 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/.
package argsbuilder_test
import (
"testing"
"github.com/stretchr/testify/suite"
"github.com/talos-systems/talos/pkg/argsbuilder"
)
type ArgsbuilderSuite struct {
suite.Suite
}
func (suite *ArgsbuilderSuite) TestMergeAdditive() {
args := argsbuilder.Args{
"param": "value1,value2,value3",
"param2": "",
}
suite.Require().NoError(
args.Merge(argsbuilder.Args{
"param": "value2, value10",
},
argsbuilder.WithMergePolicies(argsbuilder.MergePolicies{
"param": argsbuilder.MergeAdditive,
}),
),
)
suite.Require().Equal("value1,value2,value3,value10", args["param"])
suite.Assert().Equal([]string{"--param=value1,value2,value3,value10", "--param2="}, args.Args())
suite.Require().NoError(
args.Merge(argsbuilder.Args{
"param2": "value1, value5",
},
argsbuilder.WithMergePolicies(argsbuilder.MergePolicies{
"param2": argsbuilder.MergeAdditive,
}),
),
)
suite.Require().Equal("value1,value5", args["param2"])
suite.Assert().Equal([]string{"--param=value1,value2,value3,value10", "--param2=value1,value5"}, args.Args())
}
func (suite *ArgsbuilderSuite) TestMergeOverwrite() {
args := argsbuilder.Args{
"param": "value1,value2",
}
suite.Require().NoError(
args.Merge(argsbuilder.Args{
"param": "value10",
},
argsbuilder.WithMergePolicies(argsbuilder.MergePolicies{
"param2": argsbuilder.MergeAdditive,
}),
),
)
suite.Require().Equal("value10", args["param"])
suite.Assert().Equal([]string{"--param=value10"}, args.Args())
}
func (suite *ArgsbuilderSuite) TestMergeDenied() {
args := argsbuilder.Args{
"param": "value1,value2",
}
suite.Require().Error(
args.Merge(argsbuilder.Args{
"param": "value10",
},
argsbuilder.WithMergePolicies(argsbuilder.MergePolicies{
"param": argsbuilder.MergeDenied,
}),
),
)
}
func (suite *ArgsbuilderSuite) TestMergeDenyList() {
args := argsbuilder.Args{
"param": "value1,value2",
}
denyList := argsbuilder.Args{
"param1": "",
"param2": "",
"param3": "",
}
suite.Require().Error(
args.Merge(argsbuilder.Args{
"param2": "value10",
},
argsbuilder.WithDenyList(denyList),
),
)
}
func TestArgsbuilderSuite(t *testing.T) {
suite.Run(t, &ArgsbuilderSuite{})
}