mirror of
https://github.com/octoleo/restic.git
synced 2024-12-22 10:58:55 +00:00
test helpers: add RemoveAll and ResetReadOnly
This is mainly needed in Windows, where files and dirs cannot be removed unless they are writeable.
This commit is contained in:
parent
36ed3add3a
commit
1a47ea4ab8
@ -56,7 +56,7 @@ func cmdMount(t testing.TB, global GlobalOptions, dir string, ready, done chan s
|
||||
cmd := &CmdMount{global: &global, ready: ready, done: done}
|
||||
OK(t, cmd.Execute([]string{dir}))
|
||||
if TestCleanup {
|
||||
OK(t, os.RemoveAll(dir))
|
||||
RemoveAll(t, dir)
|
||||
}
|
||||
}
|
||||
|
||||
@ -101,7 +101,7 @@ func TestMount(t *testing.T) {
|
||||
OK(t, err)
|
||||
|
||||
// We remove the mountpoint now to check that cmdMount creates it
|
||||
OK(t, os.RemoveAll(mountpoint))
|
||||
RemoveAll(t, mountpoint)
|
||||
|
||||
ready := make(chan struct{}, 1)
|
||||
done := make(chan struct{})
|
||||
|
@ -216,7 +216,7 @@ func cleanupTempdir(t testing.TB, tempdir string) {
|
||||
return
|
||||
}
|
||||
|
||||
OK(t, os.RemoveAll(tempdir))
|
||||
RemoveAll(t, tempdir)
|
||||
}
|
||||
|
||||
// withTestEnvironment creates a test environment and calls f with it. After f has
|
||||
@ -245,5 +245,5 @@ func withTestEnvironment(t testing.TB, f func(*testEnvironment, GlobalOptions))
|
||||
return
|
||||
}
|
||||
|
||||
OK(t, os.RemoveAll(tempdir))
|
||||
RemoveAll(t, tempdir)
|
||||
}
|
||||
|
@ -109,7 +109,7 @@ func TestNodeRestoreAt(t *testing.T) {
|
||||
|
||||
defer func() {
|
||||
if TestCleanup {
|
||||
OK(t, os.RemoveAll(tempdir))
|
||||
RemoveAll(t, tempdir)
|
||||
} else {
|
||||
t.Logf("leaving tempdir at %v", tempdir)
|
||||
}
|
||||
|
@ -59,6 +59,7 @@ func Equals(tb testing.TB, exp, act interface{}) {
|
||||
}
|
||||
}
|
||||
|
||||
// ParseID parses s as a backend.ID and panics if that fails.
|
||||
func ParseID(s string) backend.ID {
|
||||
id, err := backend.ParseID(s)
|
||||
if err != nil {
|
||||
@ -123,7 +124,7 @@ func WithTestEnvironment(t testing.TB, repoFixture string, f func(repodir string
|
||||
return
|
||||
}
|
||||
|
||||
OK(t, os.RemoveAll(tempdir))
|
||||
RemoveAll(t, tempdir)
|
||||
}
|
||||
|
||||
// OpenLocalRepo opens the local repository located at dir.
|
||||
@ -137,3 +138,31 @@ func OpenLocalRepo(t testing.TB, dir string) *repository.Repository {
|
||||
|
||||
return repo
|
||||
}
|
||||
|
||||
func isFile(fi os.FileInfo) bool {
|
||||
return fi.Mode()&(os.ModeType|os.ModeCharDevice) == 0
|
||||
}
|
||||
|
||||
// ResetReadOnly recursively resets the read-only flag recursively for dir.
|
||||
// This is mainly used for tests on Windows, which is unable to delete a file
|
||||
// set read-only.
|
||||
func ResetReadOnly(t testing.TB, dir string) {
|
||||
OK(t, filepath.Walk(dir, func(path string, fi os.FileInfo, err error) error {
|
||||
if fi.IsDir() {
|
||||
return os.Chmod(path, 0777)
|
||||
}
|
||||
|
||||
if isFile(fi) {
|
||||
return os.Chmod(path, 0666)
|
||||
}
|
||||
|
||||
return nil
|
||||
}))
|
||||
}
|
||||
|
||||
// RemoveAll recursively resets the read-only flag of all files and dirs and
|
||||
// afterwards uses os.RemoveAll() to remove the path.
|
||||
func RemoveAll(t testing.TB, path string) {
|
||||
ResetReadOnly(t, path)
|
||||
OK(t, os.RemoveAll(path))
|
||||
}
|
||||
|
@ -50,7 +50,7 @@ func TestTree(t *testing.T) {
|
||||
dir := createTempDir(t)
|
||||
defer func() {
|
||||
if TestCleanup {
|
||||
OK(t, os.RemoveAll(dir))
|
||||
RemoveAll(t, dir)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user