1aabacc769
This attempts to retrieve cloudconfigs from two sources: the meta-data service, and the user-data service. If only one cloudconfig is found, that is applied to the system. If both services return a cloudconfig, the two are merged into a single cloudconfig which is then applied to the system. Only a subset of parameters are merged (because the meta-data service currently only partially populates a cloudconfig). In the event of any conflicts, parameters in the user-data cloudconfig take precedence over those in the meta-data cloudconfig.
111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/coreos/coreos-cloudinit/initialize"
|
|
)
|
|
|
|
func TestMergeCloudConfig(t *testing.T) {
|
|
simplecc := initialize.CloudConfig{
|
|
SSHAuthorizedKeys: []string{"abc", "def"},
|
|
Hostname: "foobar",
|
|
NetworkConfigPath: "/path/somewhere",
|
|
}
|
|
for i, tt := range []struct {
|
|
udcc initialize.CloudConfig
|
|
mdcc initialize.CloudConfig
|
|
want initialize.CloudConfig
|
|
}{
|
|
{
|
|
// If mdcc is empty, udcc should be returned unchanged
|
|
simplecc,
|
|
initialize.CloudConfig{},
|
|
simplecc,
|
|
},
|
|
{
|
|
// If udcc is empty, mdcc should be returned unchanged(overridden)
|
|
initialize.CloudConfig{},
|
|
simplecc,
|
|
simplecc,
|
|
},
|
|
{
|
|
// user-data should override completely in the case of conflicts
|
|
simplecc,
|
|
initialize.CloudConfig{
|
|
Hostname: "meta-hostname",
|
|
NetworkConfigPath: "/path/meta",
|
|
},
|
|
simplecc,
|
|
},
|
|
{
|
|
// Mixed merge should succeed
|
|
initialize.CloudConfig{
|
|
SSHAuthorizedKeys: []string{"abc", "def"},
|
|
Hostname: "user-hostname",
|
|
NetworkConfigPath: "/path/somewhere",
|
|
},
|
|
initialize.CloudConfig{
|
|
SSHAuthorizedKeys: []string{"woof", "qux"},
|
|
Hostname: "meta-hostname",
|
|
},
|
|
initialize.CloudConfig{
|
|
SSHAuthorizedKeys: []string{"abc", "def", "woof", "qux"},
|
|
Hostname: "user-hostname",
|
|
NetworkConfigPath: "/path/somewhere",
|
|
},
|
|
},
|
|
{
|
|
// Completely non-conflicting merge should be fine
|
|
initialize.CloudConfig{
|
|
Hostname: "supercool",
|
|
},
|
|
initialize.CloudConfig{
|
|
SSHAuthorizedKeys: []string{"zaphod", "beeblebrox"},
|
|
NetworkConfigPath: "/dev/fun",
|
|
},
|
|
initialize.CloudConfig{
|
|
Hostname: "supercool",
|
|
SSHAuthorizedKeys: []string{"zaphod", "beeblebrox"},
|
|
NetworkConfigPath: "/dev/fun",
|
|
},
|
|
},
|
|
{
|
|
// Non-mergeable settings in user-data should not be affected
|
|
initialize.CloudConfig{
|
|
Hostname: "mememe",
|
|
ManageEtcHosts: initialize.EtcHosts("lolz"),
|
|
},
|
|
initialize.CloudConfig{
|
|
Hostname: "youyouyou",
|
|
NetworkConfigPath: "meta-meta-yo",
|
|
},
|
|
initialize.CloudConfig{
|
|
Hostname: "mememe",
|
|
ManageEtcHosts: initialize.EtcHosts("lolz"),
|
|
NetworkConfigPath: "meta-meta-yo",
|
|
},
|
|
},
|
|
{
|
|
// Non-mergeable (unexpected) settings in meta-data are ignored
|
|
initialize.CloudConfig{
|
|
Hostname: "mememe",
|
|
},
|
|
initialize.CloudConfig{
|
|
ManageEtcHosts: initialize.EtcHosts("lolz"),
|
|
NetworkConfigPath: "meta-meta-yo",
|
|
},
|
|
initialize.CloudConfig{
|
|
Hostname: "mememe",
|
|
NetworkConfigPath: "meta-meta-yo",
|
|
},
|
|
},
|
|
} {
|
|
got := mergeCloudConfig(tt.mdcc, tt.udcc)
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("case #%d: mergeCloudConfig mutated CloudConfig unexpectedly:\ngot:\n%s\nwant:\n%s", i, got, tt.want)
|
|
}
|
|
}
|
|
}
|