109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package agent
|
|
|
|
import (
|
|
"archive/tar"
|
|
"bytes"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestParseStatLine(t *testing.T) {
|
|
// statCommand prints: type|size|mtime|mode|name
|
|
e, ok := parseStatLine("directory|4096|1760000000|drwxr-xr-x|my dir")
|
|
if !ok || !e.IsDir || e.Name != "my dir" || e.MTime != 1760000000 || e.Mode != "drwxr-xr-x" {
|
|
t.Fatalf("dir: %+v %v", e, ok)
|
|
}
|
|
e, ok = parseStatLine("regular file|12|1760000001|-rw-r--r--|a|b.txt")
|
|
if !ok || e.IsDir || e.Size != 12 || e.Name != "a|b.txt" {
|
|
t.Fatalf("file with | in name: %+v %v", e, ok)
|
|
}
|
|
if _, ok := parseStatLine("garbage"); ok {
|
|
t.Fatal("garbage parsed")
|
|
}
|
|
}
|
|
|
|
func TestStatCommandQuotes(t *testing.T) {
|
|
for _, fl := range []string{"gnu", "bsd"} {
|
|
c := statCommand(fl, "/srv/it's here")
|
|
if !strings.Contains(c, `'/srv/it'\''s here/'`) {
|
|
t.Errorf("%s: dir not quoted: %s", fl, c)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseLsLine(t *testing.T) {
|
|
e, ok := parseLsLine("-rw-r--r-- 1 root root 1234 Oct 15 10:00 my file.txt")
|
|
if !ok || e.Name != "my file.txt" || e.Size != 1234 || e.IsDir {
|
|
t.Fatalf("file: %+v %v", e, ok)
|
|
}
|
|
e, ok = parseLsLine("lrwxrwxrwx 1 root root 7 Oct 15 10:00 link -> target")
|
|
if !ok || e.Name != "link" {
|
|
t.Fatalf("link: %+v %v", e, ok)
|
|
}
|
|
if _, ok := parseLsLine("drwxr-xr-x 2 root root 4096 Oct 15 10:00 .."); ok {
|
|
t.Fatal(".. listed")
|
|
}
|
|
if _, ok := parseLsLine("total 8"); ok {
|
|
t.Fatal("total parsed")
|
|
}
|
|
}
|
|
|
|
func TestRenameTarEntries(t *testing.T) {
|
|
var src bytes.Buffer
|
|
tw := tar.NewWriter(&src)
|
|
for _, n := range []string{"old/", "old/a.txt", "old/sub/b"} {
|
|
typ := byte(tar.TypeReg)
|
|
if strings.HasSuffix(n, "/") {
|
|
typ = tar.TypeDir
|
|
}
|
|
_ = tw.WriteHeader(&tar.Header{Name: n, Mode: 0o644, Typeflag: typ})
|
|
}
|
|
_ = tw.Close()
|
|
var out bytes.Buffer
|
|
if err := renameTar(&out, &src, "old", "new"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tr := tar.NewReader(&out)
|
|
var names []string
|
|
for {
|
|
h, err := tr.Next()
|
|
if err != nil {
|
|
break
|
|
}
|
|
names = append(names, h.Name)
|
|
}
|
|
if strings.Join(names, ",") != "new/,new/a.txt,new/sub/b" {
|
|
t.Fatalf("names: %v", names)
|
|
}
|
|
}
|
|
|
|
// A macOS host's tar packs "._name" companions with the file's attributes;
|
|
// they are not the file and must not land under its name.
|
|
func TestRenameTarDropsOtherEntries(t *testing.T) {
|
|
var src bytes.Buffer
|
|
tw := tar.NewWriter(&src)
|
|
for _, e := range []struct{ n, body string }{{"./._r.txt", "attrs"}, {"./r.txt", "data"}} {
|
|
_ = tw.WriteHeader(&tar.Header{Name: e.n, Mode: 0o644, Typeflag: tar.TypeReg, Size: int64(len(e.body))})
|
|
_, _ = tw.Write([]byte(e.body))
|
|
}
|
|
_ = tw.Close()
|
|
var out bytes.Buffer
|
|
if err := renameTar(&out, &src, "r.txt", "new.txt"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tr := tar.NewReader(&out)
|
|
var got []string
|
|
for {
|
|
h, err := tr.Next()
|
|
if err != nil {
|
|
break
|
|
}
|
|
b, _ := io.ReadAll(tr)
|
|
got = append(got, h.Name+"="+string(b))
|
|
}
|
|
if strings.Join(got, ",") != "new.txt=data" {
|
|
t.Fatalf("entries: %v", got)
|
|
}
|
|
}
|