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

feat(serialization): Add #cloud-config header

Browse Source
This commit is contained in:
Brian Waldon 2014-03-12 16:52:32 -07:00
parent 8fa8495100
commit c33b492af1
2 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cloudinit/cloud_config.go
View File

@ -1,6 +1,7 @@
package cloudinit package cloudinit
import ( import (
"fmt"
"log" "log"
"launchpad.net/goyaml" "launchpad.net/goyaml"
@ -26,11 +27,14 @@ func NewCloudConfig(contents []byte) (*CloudConfig, error) {
func (cc CloudConfig) String() string { func (cc CloudConfig) String() string {
bytes, err := goyaml.Marshal(cc) bytes, err := goyaml.Marshal(cc)
if err == nil { if err != nil {
return string(bytes)
} else {
return "" return ""
} }
stringified := string(bytes)
stringified = fmt.Sprintf("#cloud-config\n%s", stringified)
return stringified
} }
func ApplyCloudConfig(cfg CloudConfig, sshKeyName string) error { func ApplyCloudConfig(cfg CloudConfig, sshKeyName string) error {

10
cloudinit/cloud_config_test.go
View File

@ -1,6 +1,7 @@
package cloudinit package cloudinit
import ( import (
"strings"
"testing" "testing"
) )
@ -146,3 +147,12 @@ ssh_authorized_keys:
t.Error("Parsed incorrect number of SSH keys") t.Error("Parsed incorrect number of SSH keys")
} }
} }
func TestCloudConfigSerializationHeader(t *testing.T) {
cfg, _ := NewCloudConfig([]byte{})
contents := cfg.String()
header := strings.SplitN(contents, "\n", 2)[0]
if header != "#cloud-config" {
t.Fatalf("Serialized config did not have expected header")
}
}