mirror of
https://github.com/vbatts/go-mtree.git
synced 2024-11-16 05:38:39 +00:00
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package mtree
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/vbatts/go-mtree/xattr"
|
|
)
|
|
|
|
func init() {
|
|
//logrus.SetLevel(logrus.DebugLevel)
|
|
}
|
|
|
|
//gocyclo:ignore
|
|
func TestXattrUpdate(t *testing.T) {
|
|
content := []byte("I know half of you half as well as I ought to")
|
|
// a bit dirty to create/destroy a directory in cwd, but often /tmp is
|
|
// mounted tmpfs and doesn't support xattrs
|
|
dir, err := ioutil.TempDir(".", "test.xattr.restore.")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(dir) // clean up
|
|
|
|
tmpfn := filepath.Join(dir, "tmpfile")
|
|
if err := ioutil.WriteFile(tmpfn, content, 0666); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := xattr.Set(dir, "user.test", []byte("directory")); err != nil {
|
|
t.Skipf("skipping: %q does not support xattrs", dir)
|
|
}
|
|
if err := xattr.Set(tmpfn, "user.test", []byte("regular file")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Walk this tempdir
|
|
dh, err := Walk(dir, nil, append(DefaultKeywords, []Keyword{"xattr", "sha1"}...), nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Now check that we're sane
|
|
res, err := Check(dir, dh, nil, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res) != 0 {
|
|
t.Errorf("expecting no failures, but got %q", res)
|
|
}
|
|
|
|
if err := xattr.Set(tmpfn, "user.test", []byte("let it fly")); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Now check that we fail the check
|
|
res, err = Check(dir, dh, nil, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res) == 0 {
|
|
t.Error("expected failures (like xattrs), but got none")
|
|
}
|
|
|
|
// restore the xattrs to original
|
|
res, err = Update(dir, dh, append(DefaultUpdateKeywords, "xattr"), nil)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if len(res) != 0 {
|
|
t.Errorf("expecting no failures, but got %q", res)
|
|
}
|
|
|
|
// Now check that we're sane again
|
|
res, err = Check(dir, dh, nil, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res) != 0 {
|
|
// pretty this shit up
|
|
buf, err := json.MarshalIndent(res, "", " ")
|
|
if err != nil {
|
|
t.Errorf("expecting no failures, but got %q", res)
|
|
} else {
|
|
t.Errorf("expecting no failures, but got %s", string(buf))
|
|
}
|
|
}
|
|
|
|
// TODO make a test for xattr here. Likely in the user space for privileges. Even still this may be prone to error for some tmpfs don't act right with xattrs. :-\
|
|
// I'd hate to have to t.Skip() a test rather than fail altogether.
|
|
}
|