Mirrors/restic
1
0
Fork 0
mirror of https://github.com/restic/restic.git synced 2025-03-16 00:00:05 +01:00
Code Issues Releases Wiki Activity

Add test.TempDir

Browse source
This commit is contained in:
Alexander Neumann 2016-09-04 16:28:47 +02:00
parent e443454c4b
commit 17ea5b1be7
1 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/restic/test/helpers.go
View file

@ -170,6 +170,25 @@ func SetupTarTestFixture(t testing.TB, outputDir, tarFile string) {
OK(t, cmd.Run())
}
func cleanupDir(t testing.TB, dir string) {
if !TestCleanupTempDirs {
t.Logf("leaving temporary directory %v used for test", dir)
return
}
RemoveAll(t, dir)
}
// TempDir returns a temporary directory.
func TempDir(t testing.TB) (string, func()) {
tempdir, err := ioutil.TempDir(TestTempDir, "restic-test-")
OK(t, err)
return tempdir, func() {
cleanupDir(t, tempdir)
}
}
// Env creates a test environment and extracts the repository fixture.
// Returned is the repo path and a cleanup function.
func Env(t testing.TB, repoFixture string) (repodir string, cleanup func()) {
@ -185,12 +204,7 @@ func Env(t testing.TB, repoFixture string) (repodir string, cleanup func()) {
SetupTarTestFixture(t, tempdir, repoFixture)
return filepath.Join(tempdir, "repo"), func() {
if !TestCleanupTempDirs {
t.Logf("leaving temporary directory %v used for test", tempdir)
return
}
RemoveAll(t, tempdir)
cleanupDir(t, tempdir)
}
}