func TestNewFileCreation(t *testing.T) {
Convey("Given a file path", t, func() {
fname := filepath.Join(os.TempDir(), "__test_filepath")
Convey("It should create an empty file", func() {
err := file.New(fname)
So(err, ShouldBeNil)
defer os.Remove(fname)
_, err = os.Stat(fname)
So(err, ShouldBeNil)
})
Convey("With optional argument Contents", func() {
c := "This is a file with text in it.\n"
Convey("It should create a new file with the given contents", func() {
err := file.New(fname, file.Contents(c))
So(err, ShouldBeNil)
defer os.Remove(fname)
out, err := ioutil.ReadFile(fname)
So(err, ShouldBeNil)
So(string(out), ShouldEqual, c)
})
})
Convey("With optional argument Permissions", func() {
p := os.FileMode(0700)
Convey("It should create a new file with the same permissions", func() {
err := file.New(fname, file.Permissions(p))
So(err, ShouldBeNil)
defer os.Remove(fname)
fi, err := os.Lstat(fname)
So(err, ShouldBeNil)
So(fi.Mode(), ShouldEqual, p)
})
})
Convey("With optional argument User ID and/or Group ID", func() {
userID, groupID := os.Getuid(), os.Getgid()
Convey("It should create a new file belonging to given IDs", func() {
err := file.New(fname, file.UID(userID), file.GID(groupID))
So(err, ShouldBeNil)
defer os.Remove(fname)
fi, err := os.Lstat(fname)
So(err, ShouldBeNil)
uid := fi.Sys().(*syscall.Stat_t).Uid
So(uid, ShouldEqual, userID)
gid := fi.Sys().(*syscall.Stat_t).Gid
So(gid, ShouldEqual, groupID)
})
})
})
}
// writeFiles persists the given files to the system.
func (imp *Implementation) writeFiles(files []cloudconfig.WriteFile) {
for _, f := range files {
p, err := strconv.Atoi(f.Permissions)
if err != nil {
flog.Error("Failed to convert permissions",
flog.Fields{
Event: "strconv.Atoi",
Error: err,
},
flog.Details{
"file": f.Path,
"permissions": f.Permissions,
},
)
continue
}
perms := os.FileMode(p)
err = file.New(f.Path, file.Permissions(perms), file.Contents(f.Content))
if err != nil {
flog.Error("Failed to create file",
flog.Fields{
Event: "file.New",
Error: err,
},
flog.Details{
"file": f.Path,
},
)
}
}
}
func TestRetrievesDataFromConfigDrive(t *testing.T) {
Convey("Given a mounted config drive", t, func() {
tmpdir, err := ioutil.TempDir("", "flamingotest")
So(err, ShouldBeNil)
defer os.RemoveAll(tmpdir)
dataPath := strings.Join([]string{tmpdir, "openstack", "2012-08-10"}, "/")
err = os.MkdirAll(dataPath, 0755)
So(err, ShouldBeNil)
buf, err := ioutil.ReadFile("../openstack/test_metadata/2012-08-10.json")
err = file.New(filepath.Join(dataPath, "meta_data.json"), file.Contents(string(buf)))
So(err, ShouldBeNil)
err = file.New(filepath.Join(dataPath, "user_data"), file.Contents("#cloud-config\n"))
So(err, ShouldBeNil)
mount := &configdrive.Mount{tmpdir}
Convey("It should return a metadata digest", func() {
digest, err := mount.FetchMetadata()
So(err, ShouldBeNil)
So(digest.Hostname, ShouldEqual, "test.novalocal")
So(digest.SSHKeys["mykey"], ShouldConsistOf,
ssh.Key("ssh-rsa RSA_PUBLIC_KEY Generated by Nova"))
})
Convey("It should return user-data", func() {
userdata, err := mount.FetchUserdata()
So(err, ShouldBeNil)
So(userdata["user-data"], ShouldEqual, "#cloud-config\n")
})
})
}