unistack-org/cloudinit
4
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #14 from bcwaldon/fix-config-names

Browse Source 
Fix config names
This commit is contained in:
Brian Waldon 2014-03-13 11:47:27 -07:00
commit e3741a0fa3
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cloudinit/cloud_config.go
View File

@ -10,14 +10,14 @@ import (
const DefaultSSHKeyName = "coreos-cloudinit"
type CloudConfig struct {
SSH_Authorized_Keys []string
Coreos struct {
SSHAuthorizedKeys []string `yaml:"ssh_authorized_keys"`
Coreos struct {
Etcd struct{ Discovery_URL string }
Fleet struct{ Autostart bool }
Units []Unit
}
Write_Files []WriteFile
Hostname string
WriteFiles []WriteFile `yaml:"write_files"`
Hostname string
}
func NewCloudConfig(contents []byte) (*CloudConfig, error) {
@ -46,8 +46,8 @@ func ApplyCloudConfig(cfg CloudConfig, sshKeyName string) error {
log.Printf("Set hostname to %s", cfg.Hostname)
}
if len(cfg.SSH_Authorized_Keys) > 0 {
err := AuthorizeSSHKeys(sshKeyName, cfg.SSH_Authorized_Keys)
if len(cfg.SSHAuthorizedKeys) > 0 {
err := AuthorizeSSHKeys(sshKeyName, cfg.SSHAuthorizedKeys)
if err == nil {
log.Printf("Authorized SSH keys for core user")
} else {
@ -55,8 +55,8 @@ func ApplyCloudConfig(cfg CloudConfig, sshKeyName string) error {
}
}
if len(cfg.Write_Files) > 0 {
for _, file := range cfg.Write_Files {
if len(cfg.WriteFiles) > 0 {
for _, file := range cfg.WriteFiles {
if err := ProcessWriteFile("/", &file); err != nil {
return err
}

14
cloudinit/cloud_config_test.go
View File

@ -12,7 +12,7 @@ func TestCloudConfigEmpty(t *testing.T) {
t.Fatalf("Encountered unexpected error :%v", err)
}
keys := cfg.SSH_Authorized_Keys
keys := cfg.SSHAuthorizedKeys
if len(keys) != 0 {
t.Error("Parsed incorrect number of SSH keys")
}
@ -25,8 +25,8 @@ func TestCloudConfigEmpty(t *testing.T) {
t.Error("Expected AutostartFleet not to be defined")
}
if len(cfg.Write_Files) != 0 {
t.Error("Expected zero Write_Files")
if len(cfg.WriteFiles) != 0 {
t.Error("Expected zero WriteFiles")
}
if cfg.Hostname != "" {
@ -72,7 +72,7 @@ hostname: trontastic
t.Fatalf("Encountered unexpected error :%v", err)
}
keys := cfg.SSH_Authorized_Keys
keys := cfg.SSHAuthorizedKeys
if len(keys) != 2 {
t.Error("Parsed incorrect number of SSH keys")
} else if keys[0] != "foobar" {
@ -89,10 +89,10 @@ hostname: trontastic
t.Error("Expected AutostartFleet to be true")
}
if len(cfg.Write_Files) != 1 {
if len(cfg.WriteFiles) != 1 {
t.Error("Failed to parse correct number of write_files")
} else {
wf := cfg.Write_Files[0]
wf := cfg.WriteFiles[0]
if wf.Content != "penny\nelroy\n" {
t.Errorf("WriteFile has incorrect contents '%s'", wf.Content)
}
@ -150,7 +150,7 @@ ssh_authorized_keys:
t.Fatalf("Encountered unexpected error :%v", err)
}
keys := cfg.SSH_Authorized_Keys
keys := cfg.SSHAuthorizedKeys
if len(keys) != 0 {
t.Error("Parsed incorrect number of SSH keys")
}