package restic_test import ( "io/ioutil" "os" "path/filepath" "runtime" "testing" "time" "github.com/restic/restic" "github.com/restic/restic/backend" . "github.com/restic/restic/test" ) func BenchmarkNodeFillUser(t *testing.B) { tempfile, err := ioutil.TempFile("", "restic-test-temp-") defer tempfile.Close() if err != nil { t.Fatal(err) } fi, err := tempfile.Stat() if err != nil { t.Fatal(err) } path := tempfile.Name() t.ResetTimer() for i := 0; i < t.N; i++ { restic.NodeFromFileInfo(path, fi) } } func BenchmarkNodeFromFileInfo(t *testing.B) { tempfile, err := ioutil.TempFile("", "restic-test-temp-") defer tempfile.Close() if err != nil { t.Fatal(err) } fi, err := tempfile.Stat() if err != nil { t.Fatal(err) } path := tempfile.Name() t.ResetTimer() for i := 0; i < t.N; i++ { _, err := restic.NodeFromFileInfo(path, fi) if err != nil { t.Fatal(err) } } } func parseTime(s string) time.Time { t, err := time.Parse("2006-01-02 15:04:05.999", s) if err != nil { panic(err) } return t.Local() } var nodeTests = []restic.Node{ restic.Node{ Name: "testFile", Type: "file", Content: []backend.ID{}, UID: uint32(os.Getuid()), GID: uint32(os.Getgid()), Mode: 0604, ModTime: parseTime("2015-05-14 21:07:23.111"), AccessTime: parseTime("2015-05-14 21:07:24.222"), ChangeTime: parseTime("2015-05-14 21:07:25.333"), }, restic.Node{ Name: "testDir", Type: "dir", Subtree: nil, UID: uint32(os.Getuid()), GID: uint32(os.Getgid()), Mode: 020000000750, ModTime: parseTime("2015-05-14 21:07:23.111"), AccessTime: parseTime("2015-05-14 21:07:24.222"), ChangeTime: parseTime("2015-05-14 21:07:25.333"), }, restic.Node{ Name: "testSymlink", Type: "symlink", LinkTarget: "invalid", UID: uint32(os.Getuid()), GID: uint32(os.Getgid()), Mode: 01000000777, ModTime: parseTime("2015-05-14 21:07:23.111"), AccessTime: parseTime("2015-05-14 21:07:24.222"), ChangeTime: parseTime("2015-05-14 21:07:25.333"), }, } func TestNodeRestoreAt(t *testing.T) { tempdir, err := ioutil.TempDir(TestTempDir, "restic-test-") OK(t, err) defer func() { if TestCleanup { OK(t, os.RemoveAll(tempdir)) } else { t.Logf("leaving tempdir at %v", tempdir) } }() for _, test := range nodeTests { nodePath := filepath.Join(tempdir, test.Name) OK(t, test.CreateAt(nodePath, nil)) if test.Type == "dir" { OK(t, test.RestoreTimestamps(nodePath)) } fi, err := os.Lstat(nodePath) OK(t, err) n2, err := restic.NodeFromFileInfo(nodePath, fi) OK(t, err) Assert(t, test.Name == n2.Name, "%v: name doesn't match (%v != %v)", test.Type, test.Name, n2.Name) Assert(t, test.Type == n2.Type, "%v: type doesn't match (%v != %v)", test.Type, test.Type, n2.Type) Assert(t, test.Size == n2.Size, "%v: size doesn't match (%v != %v)", test.Size, test.Size, n2.Size) Assert(t, test.UID == n2.UID, "%v: UID doesn't match (%v != %v)", test.Type, test.UID, n2.UID) Assert(t, test.GID == n2.GID, "%v: GID doesn't match (%v != %v)", test.Type, test.GID, n2.GID) if test.Type != "symlink" { Assert(t, test.Mode == n2.Mode, "%v: mode doesn't match (%v != %v)", test.Type, test.Mode, n2.Mode) } AssertFsTimeEqual(t, "AccessTime", test.Type, test.AccessTime, n2.AccessTime) AssertFsTimeEqual(t, "ModTime", test.Type, test.ModTime, n2.ModTime) } } func AssertFsTimeEqual(t *testing.T, label string, nodeType string, t1 time.Time, t2 time.Time) { var equal bool // Go currently doesn't support setting timestamps of symbolic links on darwin and bsd if nodeType == "symlink" { switch runtime.GOOS { case "darwin", "freebsd", "openbsd": return } } switch runtime.GOOS { case "darwin": // HFS+ timestamps don't support sub-second precision, // see https://en.wikipedia.org/wiki/Comparison_of_file_systems diff := int(t1.Sub(t2).Seconds()) equal = diff == 0 default: equal = t1.Equal(t2) } Assert(t, equal, "%s: %s doesn't match (%v != %v)", label, nodeType, t1, t2) }