X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=cmd%2Fsafcm-remote%2Fsync%2Ffiles_test.go;h=e7864ba8b8080e489da5964663c1496308599bd1;hb=3473766d8afb07cb0685694656947883cbbd1138;hp=22daa63fae92537239b21cb3fab375f7329ab313;hpb=f2f2bc47e8729548f3c10117f7f008b547c4afc5;p=safcm%2Fsafcm.git diff --git a/cmd/safcm-remote/sync/files_test.go b/cmd/safcm-remote/sync/files_test.go index 22daa63..e7864ba 100644 --- a/cmd/safcm-remote/sync/files_test.go +++ b/cmd/safcm-remote/sync/files_test.go @@ -20,66 +20,15 @@ import ( "io/fs" "math/rand" "os" - "os/user" "path/filepath" - "reflect" "regexp" - "strconv" - "syscall" "testing" - "github.com/google/go-cmp/cmp" - "ruderich.org/simon/safcm" + ft "ruderich.org/simon/safcm/cmd/safcm-remote/sync/filetest" + "ruderich.org/simon/safcm/testutil" ) -type File struct { - Path string - Mode fs.FileMode - Data []byte -} - -func walkDir(basePath string) ([]File, error) { - var res []File - err := filepath.WalkDir(basePath, func(path string, d fs.DirEntry, err error) error { - if err != nil { - return err - } - info, err := d.Info() - if err != nil { - return err - } - rel, err := filepath.Rel(basePath, path) - if err != nil { - return err - } - - f := File{ - Path: rel, - Mode: info.Mode(), - } - if f.Mode.Type() == 0 { - x, err := os.ReadFile(path) - if err != nil { - return err - } - f.Data = x - } else if f.Mode.Type() == fs.ModeSymlink { - x, err := os.Readlink(path) - if err != nil { - return err - } - f.Data = []byte(x) - } - res = append(res, f) - return nil - }) - if err != nil { - return nil, err - } - return res, nil -} - var randFilesRegexp = regexp.MustCompile(`\d+"$`) func TestSyncFiles(t *testing.T) { @@ -98,11 +47,11 @@ func TestSyncFiles(t *testing.T) { t.Fatal(err) } - root := File{ + root := ft.File{ Path: ".", Mode: fs.ModeDir | 0700, } - user, uid, group, gid := currentUserAndGroup() + user, uid, group, gid := ft.CurrentUserAndGroup() tmpTestFilePath := "/tmp/safcm-sync-files-test-file" @@ -111,7 +60,7 @@ func TestSyncFiles(t *testing.T) { req safcm.MsgSyncReq prepare func() triggers []string - expFiles []File + expFiles []ft.File expResp safcm.MsgSyncResp expDbg []string expErr error @@ -153,7 +102,7 @@ func TestSyncFiles(t *testing.T) { }, nil, nil, - []File{ + []ft.File{ root, { Path: "dir", @@ -235,11 +184,11 @@ func TestSyncFiles(t *testing.T) { }, }, func() { - createDirectory("dir", 0755) - createFile("dir/file", "content\n", 0644) + ft.CreateDirectory("dir", 0755) + ft.CreateFile("dir/file", "content\n", 0644) }, nil, - []File{ + []ft.File{ root, { Path: "dir", @@ -276,7 +225,7 @@ func TestSyncFiles(t *testing.T) { }, nil, nil, - []File{ + []ft.File{ root, }, safcm.MsgSyncResp{}, @@ -299,7 +248,7 @@ func TestSyncFiles(t *testing.T) { }, nil, nil, - []File{ + []ft.File{ root, }, safcm.MsgSyncResp{}, @@ -355,7 +304,7 @@ func TestSyncFiles(t *testing.T) { }, nil, nil, - []File{ + []ft.File{ root, }, safcm.MsgSyncResp{}, @@ -406,11 +355,11 @@ func TestSyncFiles(t *testing.T) { }, }, func() { - createDirectory("dir", 0755) - createFile("dir/file", "content\n", 0644) + ft.CreateDirectory("dir", 0755) + ft.CreateFile("dir/file", "content\n", 0644) }, nil, - []File{ + []ft.File{ root, { Path: "dir", @@ -473,13 +422,13 @@ func TestSyncFiles(t *testing.T) { if err != nil { panic(err) } - createDirectory("dir", 0755) - createFile("dir/file", "content\n", 0644) + ft.CreateDirectory("dir", 0755) + ft.CreateFile("dir/file", "content\n", 0644) }, []string{ ".", }, - []File{ + []ft.File{ root, { Path: "dir", @@ -561,14 +510,14 @@ func TestSyncFiles(t *testing.T) { }, }, func() { - createDirectory("dir", 0750) - createFile("dir/file", "content\n", 0644) + ft.CreateDirectory("dir", 0750) + ft.CreateFile("dir/file", "content\n", 0644) }, []string{ ".", "dir", }, - []File{ + []ft.File{ root, { Path: "dir", @@ -651,14 +600,14 @@ func TestSyncFiles(t *testing.T) { }, }, func() { - createDirectory("dir", 0755) + ft.CreateDirectory("dir", 0755) }, []string{ ".", "dir", "dir/file", }, - []File{ + []ft.File{ root, { Path: "dir", @@ -742,7 +691,7 @@ func TestSyncFiles(t *testing.T) { "dir", "dir/file", }, - []File{ + []ft.File{ root, { Path: "dir", @@ -856,7 +805,7 @@ func TestSyncFiles(t *testing.T) { // Don't use variable for more robust test "/tmp/safcm-sync-files-test-file", }, - []File{ + []ft.File{ root, }, safcm.MsgSyncResp{ @@ -890,60 +839,46 @@ func TestSyncFiles(t *testing.T) { } for _, tc := range tests { - // Create separate test directory for each test case - path := filepath.Join(cwd, "testdata", "files-"+tc.name) - err = os.Mkdir(path, 0700) - if err != nil { - t.Fatal(err) - } - err = os.Chdir(path) - if err != nil { - t.Fatal(err) - } - - if tc.prepare != nil { - tc.prepare() - } + t.Run(tc.name, func(t *testing.T) { + // Create separate test directory for each test case + path := filepath.Join(cwd, "testdata", "files-"+tc.name) + err = os.Mkdir(path, 0700) + if err != nil { + t.Fatal(err) + } + err = os.Chdir(path) + if err != nil { + t.Fatal(err) + } - s, res := prepareSync(tc.req, &testRunner{ - t: t, - name: tc.name, - }) - s.setDefaults() + if tc.prepare != nil { + tc.prepare() + } - err := s.syncFiles() - // Ugly but the simplest way to compare errors (including nil) - if fmt.Sprintf("%s", err) != fmt.Sprintf("%s", tc.expErr) { - t.Errorf("%s: err = %#v, want %#v", - tc.name, err, tc.expErr) - } - dbg := res.Wait() - // Remove random file names from result - for i, x := range dbg { - dbg[i] = randFilesRegexp.ReplaceAllString(x, `RND"`) - } - if !reflect.DeepEqual(tc.expDbg, dbg) { - t.Errorf("%s: dbg: %s", tc.name, - cmp.Diff(tc.expDbg, dbg)) - } + s, res := prepareSync(tc.req, &testRunner{ + t: t, + }) + s.setDefaults() + + err := s.syncFiles() + testutil.AssertErrorEqual(t, "err", err, tc.expErr) + dbg := res.Wait() + // Remove random file names from result + for i, x := range dbg { + dbg[i] = randFilesRegexp.ReplaceAllString(x, `RND"`) + } + testutil.AssertEqual(t, "dbg", dbg, tc.expDbg) - files, err := walkDir(path) - if err != nil { - t.Fatal(err) - } - if !reflect.DeepEqual(tc.expFiles, files) { - t.Errorf("%s: files: %s", tc.name, - cmp.Diff(tc.expFiles, files)) - } + files, err := ft.WalkDir(path) + if err != nil { + t.Fatal(err) + } + testutil.AssertEqual(t, "files", files, tc.expFiles) - if !reflect.DeepEqual(tc.expResp, s.resp) { - t.Errorf("%s: resp: %s", tc.name, - cmp.Diff(tc.expResp, s.resp)) - } - if !reflect.DeepEqual(tc.triggers, s.triggers) { - t.Errorf("%s: triggers: %s", tc.name, - cmp.Diff(tc.triggers, s.triggers)) - } + testutil.AssertEqual(t, "resp", s.resp, tc.expResp) + testutil.AssertEqual(t, "triggers", + s.triggers, tc.triggers) + }) } os.Remove(tmpTestFilePath) @@ -971,11 +906,11 @@ func TestSyncFile(t *testing.T) { t.Fatal(err) } - root := File{ + root := ft.File{ Path: ".", Mode: fs.ModeDir | 0700, } - user, uid, group, gid := currentUserAndGroup() + user, uid, group, gid := ft.CurrentUserAndGroup() tests := []struct { name string @@ -983,7 +918,7 @@ func TestSyncFile(t *testing.T) { file *safcm.File prepare func() expChanged bool - expFiles []File + expFiles []ft.File expResp safcm.MsgSyncResp expDbg []string expErr error @@ -1009,7 +944,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{ + []ft.File{ root, { Path: "file", @@ -1055,7 +990,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{root}, + []ft.File{root}, safcm.MsgSyncResp{ FileChanges: []safcm.FileChange{ { @@ -1091,10 +1026,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile("file", "content\n", 0644) + ft.CreateFile("file", "content\n", 0644) }, false, - []File{ + []ft.File{ root, { Path: "file", @@ -1123,10 +1058,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile("file", "content\n", 0644) + ft.CreateFile("file", "content\n", 0644) }, false, - []File{ + []ft.File{ root, { Path: "file", @@ -1153,10 +1088,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile("file", "content\n", 0755) + ft.CreateFile("file", "content\n", 0755) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -1206,10 +1141,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile("file", "old content\n", 0644) + ft.CreateFile("file", "old content\n", 0644) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -1267,7 +1202,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{ + []ft.File{ root, { Path: "link", @@ -1310,10 +1245,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile(".link8717895732742165505", "", 0600) + ft.CreateFile(".link8717895732742165505", "", 0600) }, true, - []File{ + []ft.File{ root, { Path: ".link8717895732742165505", @@ -1365,7 +1300,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{root}, + []ft.File{root}, safcm.MsgSyncResp{ FileChanges: []safcm.FileChange{ { @@ -1401,10 +1336,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createSymlink("link", "target") + ft.CreateSymlink("link", "target") }, false, - []File{ + []ft.File{ root, { Path: "link", @@ -1431,10 +1366,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createSymlink("link", "old-target") + ft.CreateSymlink("link", "old-target") }, true, - []File{ + []ft.File{ root, { Path: "link", @@ -1490,7 +1425,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{ + []ft.File{ root, { Path: "dir", @@ -1535,7 +1470,7 @@ func TestSyncFile(t *testing.T) { }, nil, true, - []File{root}, + []ft.File{root}, safcm.MsgSyncResp{ FileChanges: []safcm.FileChange{ { @@ -1570,10 +1505,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createDirectory("dir", 0755) + ft.CreateDirectory("dir", 0755) }, false, - []File{ + []ft.File{ root, { Path: "dir", @@ -1598,10 +1533,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createDirectory("dir", 0500|fs.ModeSticky) + ft.CreateDirectory("dir", 0500|fs.ModeSticky) }, true, - []File{ + []ft.File{ root, { Path: "dir", @@ -1650,10 +1585,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFile("path", "content\n", 0644) + ft.CreateFile("path", "content\n", 0644) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1708,10 +1643,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("target"), }, func() { - createFile("path", "content\n", 0644) + ft.CreateFile("path", "content\n", 0644) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1766,10 +1701,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("content\n"), }, func() { - createSymlink("path", "target") + ft.CreateSymlink("path", "target") }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1823,10 +1758,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createSymlink("path", "target") + ft.CreateSymlink("path", "target") }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1881,10 +1816,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("content\n"), }, func() { - createDirectory("path", 0777) + ft.CreateDirectory("path", 0777) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1935,10 +1870,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("target"), }, func() { - createDirectory("path", 0777) + ft.CreateDirectory("path", 0777) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -1989,10 +1924,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("content\n"), }, func() { - createFifo("path", 0666) + ft.CreateFifo("path", 0666) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -2042,10 +1977,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("target"), }, func() { - createFifo("path", 0666) + ft.CreateFifo("path", 0666) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -2094,10 +2029,10 @@ func TestSyncFile(t *testing.T) { OrigGroup: "group", }, func() { - createFifo("path", 0666) + ft.CreateFifo("path", 0666) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -2148,10 +2083,10 @@ func TestSyncFile(t *testing.T) { Data: []byte("target"), }, func() { - createFile("path", "target", 0644) + ft.CreateFile("path", "target", 0644) }, true, - []File{ + []ft.File{ root, { Path: "path", @@ -2211,14 +2146,14 @@ file OrigGroup: "group", }, func() { - createFile("file", `this + ft.CreateFile("file", `this is file ! `, 0644) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -2283,10 +2218,10 @@ file OrigGroup: "group", }, func() { - createFile("file", "\x00\x01\x02", 0644) + ft.CreateFile("file", "\x00\x01\x02", 0644) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -2338,10 +2273,10 @@ file OrigGroup: "group", }, func() { - createFile("file", "\x00\x01\x02", 0644) + ft.CreateFile("file", "\x00\x01\x02", 0644) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -2397,10 +2332,10 @@ file OrigGroup: "group", }, func() { - createFile("file", "content\n", 0644) + ft.CreateFile("file", "content\n", 0644) }, true, - []File{ + []ft.File{ root, { Path: "file", @@ -2444,64 +2379,49 @@ file } for _, tc := range tests { - // Create separate test directory for each test case - path := filepath.Join(cwd, "testdata", "file-"+tc.name) - err = os.Mkdir(path, 0700) - if err != nil { - t.Fatal(err) - } - err = os.Chdir(path) - if err != nil { - t.Fatal(err) - } - - if tc.prepare != nil { - tc.prepare() - } - - s, res := prepareSync(tc.req, &testRunner{ - t: t, - name: tc.name, - }) - s.setDefaults() + t.Run(tc.name, func(t *testing.T) { + // Create separate test directory for each test case + path := filepath.Join(cwd, "testdata", "file-"+tc.name) + err = os.Mkdir(path, 0700) + if err != nil { + t.Fatal(err) + } + err = os.Chdir(path) + if err != nil { + t.Fatal(err) + } - // Deterministic temporary symlink names - rand.Seed(0) + if tc.prepare != nil { + tc.prepare() + } - var changed bool - err := s.syncFile(tc.file, &changed) - // Ugly but the simplest way to compare errors (including nil) - if fmt.Sprintf("%s", err) != fmt.Sprintf("%s", tc.expErr) { - t.Errorf("%s: err = %#v, want %#v", - tc.name, err, tc.expErr) - } - dbg := res.Wait() - // Remove random file names from result - for i, x := range dbg { - dbg[i] = randFilesRegexp.ReplaceAllString(x, `RND"`) - } - if !reflect.DeepEqual(tc.expDbg, dbg) { - t.Errorf("%s: dbg: %s", tc.name, - cmp.Diff(tc.expDbg, dbg)) - } + s, res := prepareSync(tc.req, &testRunner{ + t: t, + }) + s.setDefaults() + + // Deterministic temporary symlink names + rand.Seed(0) + + var changed bool + err := s.syncFile(tc.file, &changed) + testutil.AssertErrorEqual(t, "err", err, tc.expErr) + dbg := res.Wait() + // Remove random file names from result + for i, x := range dbg { + dbg[i] = randFilesRegexp.ReplaceAllString(x, `RND"`) + } + testutil.AssertEqual(t, "dbg", dbg, tc.expDbg) - files, err := walkDir(path) - if err != nil { - t.Fatal(err) - } - if !reflect.DeepEqual(tc.expFiles, files) { - t.Errorf("%s: files: %s", tc.name, - cmp.Diff(tc.expFiles, files)) - } + files, err := ft.WalkDir(path) + if err != nil { + t.Fatal(err) + } + testutil.AssertEqual(t, "files", files, tc.expFiles) - if tc.expChanged != changed { - t.Errorf("%s: changed = %#v, want %#v", - tc.name, changed, tc.expChanged) - } - if !reflect.DeepEqual(tc.expResp, s.resp) { - t.Errorf("%s: resp: %s", tc.name, - cmp.Diff(tc.expResp, s.resp)) - } + testutil.AssertEqual(t, "changed", changed, tc.expChanged) + testutil.AssertEqual(t, "resp", s.resp, tc.expResp) + }) } if !t.Failed() { @@ -2511,62 +2431,3 @@ file } } } - -// Helper functions - -func createFile(path string, data string, mode fs.FileMode) { - err := os.WriteFile(path, []byte(data), 0644) - if err != nil { - panic(err) - } - err = os.Chmod(path, mode) - if err != nil { - panic(err) - } -} -func createSymlink(path string, data string) { - err := os.Symlink(data, path) - if err != nil { - panic(err) - } -} -func createDirectory(path string, mode fs.FileMode) { - err := os.Mkdir(path, 0700) - if err != nil { - panic(err) - } - err = os.Chmod(path, mode) - if err != nil { - panic(err) - } -} -func createFifo(path string, mode fs.FileMode) { - err := syscall.Mkfifo(path, 0600) - if err != nil { - panic(err) - } - err = os.Chmod(path, mode) - if err != nil { - panic(err) - } -} - -func currentUserAndGroup() (string, int, string, int) { - u, err := user.Current() - if err != nil { - panic(err) - } - g, err := user.LookupGroupId(u.Gid) - if err != nil { - panic(err) - } - uid, err := strconv.Atoi(u.Uid) - if err != nil { - panic(err) - } - gid, err := strconv.Atoi(g.Gid) - if err != nil { - panic(err) - } - return u.Username, uid, g.Name, gid -}