test: Use testing.T.Cleanup to remove tempdirs

cmd/restic/cmd_backup_test.go

@@ -14,8 +14,7 @@ import (
 )
 
 func TestCollectTargets(t *testing.T) {
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
 
 	fooSpace := "foo "
 	barStar := "bar*"              // Must sort before the others, below.

cmd/restic/exclude_test.go

@@ -84,8 +84,7 @@ func TestIsExcludedByFile(t *testing.T) {
 	}
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
-			tempDir, cleanup := test.TempDir(t)
+			tempDir := test.TempDir(t)
-			defer cleanup()
 
 			foo := filepath.Join(tempDir, "foo")
 			err := os.WriteFile(foo, []byte("foo"), 0666)
@@ -115,8 +114,7 @@ func TestIsExcludedByFile(t *testing.T) {
 // cancel each other out. It was initially written to demonstrate a bug in
 // rejectIfPresent.
 func TestMultipleIsExcludedByFile(t *testing.T) {
-	tempDir, cleanup := test.TempDir(t)
+	tempDir := test.TempDir(t)
-	defer cleanup()
 
 	// Create some files in a temporary directory.
 	// Files in UPPERCASE will be used as exclusion triggers later on.
@@ -240,8 +238,7 @@ func TestParseInvalidSizeStr(t *testing.T) {
 // TestIsExcludedByFileSize is for testing the instance of
 // --exclude-larger-than parameters
 func TestIsExcludedByFileSize(t *testing.T) {
-	tempDir, cleanup := test.TempDir(t)
+	tempDir := test.TempDir(t)
-	defer cleanup()
 
 	// Max size of file is set to be 1k
 	maxSizeStr := "1k"

cmd/restic/global_test.go

@@ -31,8 +31,7 @@ func Test_PrintFunctionsRespectsGlobalStdout(t *testing.T) {
 }
 
 func TestReadRepo(t *testing.T) {
-	tempDir, cleanup := test.TempDir(t)
+	tempDir := test.TempDir(t)
-	defer cleanup()
 
 	// test --repo option
 	var opts GlobalOptions

cmd/restic/secondary_repo_test.go

@@ -160,10 +160,8 @@ func TestFillSecondaryGlobalOpts(t *testing.T) {
 	}
 
 	//Create temp dir to create password file.
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
+	cleanup := rtest.Chdir(t, dir)
-
-	cleanup = rtest.Chdir(t, dir)
 	defer cleanup()
 
 	//Create temporary password file

internal/archiver/archiver_test.go

@@ -26,17 +26,12 @@ import (
 )
 
 func prepareTempdirRepoSrc(t testing.TB, src TestDir) (tempdir string, repo restic.Repository, cleanup func()) {
-	tempdir, removeTempdir := restictest.TempDir(t)
+	tempdir = restictest.TempDir(t)
 	repo, removeRepository := repository.TestRepository(t)
 
 	TestCreateFiles(t, tempdir, src)
 
-	cleanup = func() {
+	return tempdir, repo, removeRepository
-		removeRepository()
-		removeTempdir()
-	}
-
-	return tempdir, repo, cleanup
 }
 
 func saveFile(t testing.TB, repo restic.Repository, filename string, filesystem fs.FS) (*restic.Node, ItemStats) {
@@ -470,8 +465,7 @@ func appendToFile(t testing.TB, filename string, data []byte) {
 }
 
 func TestArchiverSaveFileIncremental(t *testing.T) {
-	tempdir, removeTempdir := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
-	defer removeTempdir()
 
 	testRepo, removeRepository := repository.TestRepository(t)
 	defer removeRepository()
@@ -688,8 +682,7 @@ func TestFileChanged(t *testing.T) {
 				t.Skip("don't run test on Windows")
 			}
 
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
 
 			filename := filepath.Join(tempdir, "file")
 			content := defaultContent
@@ -725,8 +718,7 @@ func TestFileChanged(t *testing.T) {
 }
 
 func TestFilChangedSpecialCases(t *testing.T) {
-	tempdir, cleanup := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
-	defer cleanup()
 
 	filename := filepath.Join(tempdir, "file")
 	content := []byte("foobar")
@@ -913,8 +905,7 @@ func TestArchiverSaveDir(t *testing.T) {
 }
 
 func TestArchiverSaveDirIncremental(t *testing.T) {
-	tempdir, removeTempdir := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
-	defer removeTempdir()
 
 	testRepo, removeRepository := repository.TestRepository(t)
 	defer removeRepository()
@@ -1901,11 +1892,10 @@ func TestArchiverContextCanceled(t *testing.T) {
 	ctx, cancel := context.WithCancel(context.Background())
 	cancel()
 
-	tempdir, removeTempdir := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
 	TestCreateFiles(t, tempdir, TestDir{
 		"targetfile": TestFile{Content: "foobar"},
 	})
-	defer removeTempdir()
 
 	// Ensure that the archiver itself reports the canceled context and not just the backend
 	repo, _ := repository.TestRepositoryWithBackend(t, &noCancelBackend{mem.New()}, 0)

internal/archiver/file_saver_test.go

@@ -15,8 +15,8 @@ import (
 	"golang.org/x/sync/errgroup"
 )
 
-func createTestFiles(t testing.TB, num int) (files []string, cleanup func()) {
+func createTestFiles(t testing.TB, num int) (files []string) {
-	tempdir, cleanup := test.TempDir(t)
+	tempdir := test.TempDir(t)
 
 	for i := 0; i < 15; i++ {
 		filename := fmt.Sprintf("testfile-%d", i)
@@ -27,7 +27,7 @@ func createTestFiles(t testing.TB, num int) (files []string, cleanup func()) {
 		files = append(files, filepath.Join(tempdir, filename))
 	}
 
-	return files, cleanup
+	return files
 }
 
 func startFileSaver(ctx context.Context, t testing.TB) (*FileSaver, context.Context, *errgroup.Group) {
@@ -60,8 +60,7 @@ func TestFileSaver(t *testing.T) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
-	files, cleanup := createTestFiles(t, 15)
+	files := createTestFiles(t, 15)
-	defer cleanup()
 
 	startFn := func() {}
 	completeReadingFn := func() {}

internal/archiver/scanner_test.go

@@ -81,9 +81,7 @@ func TestScanner(t *testing.T) {
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
-
 			TestCreateFiles(t, tempdir, test.src)
 
 			back := restictest.Chdir(t, tempdir)
@@ -218,9 +216,7 @@ func TestScannerError(t *testing.T) {
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
-
 			TestCreateFiles(t, tempdir, test.src)
 
 			back := restictest.Chdir(t, tempdir)
@@ -292,9 +288,7 @@ func TestScannerCancel(t *testing.T) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
-	tempdir, cleanup := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
-	defer cleanup()
-
 	TestCreateFiles(t, tempdir, src)
 
 	back := restictest.Chdir(t, tempdir)

internal/archiver/testing_test.go

@@ -101,8 +101,7 @@ func TestTestCreateFiles(t *testing.T) {
 	}
 
 	for i, test := range tests {
-		tempdir, cleanup := restictest.TempDir(t)
+		tempdir := restictest.TempDir(t)
-		defer cleanup()
 
 		t.Run("", func(t *testing.T) {
 			tempdir := filepath.Join(tempdir, fmt.Sprintf("test-%d", i))
@@ -192,8 +191,7 @@ func TestTestWalkFiles(t *testing.T) {
 
 	for _, test := range tests {
 		t.Run("", func(t *testing.T) {
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
 
 			got := make(map[string]string)
 
@@ -323,9 +321,7 @@ func TestTestEnsureFiles(t *testing.T) {
 
 	for _, test := range tests {
 		t.Run("", func(t *testing.T) {
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
-
 			createFilesAt(t, tempdir, test.files)
 
 			subtestT := testing.TB(t)
@@ -456,8 +452,7 @@ func TestTestEnsureSnapshot(t *testing.T) {
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
 
 			targetDir := filepath.Join(tempdir, "target")
 			err := fs.Mkdir(targetDir, 0700)

internal/archiver/tree_test.go

@@ -439,9 +439,7 @@ func TestTree(t *testing.T) {
 				t.Skip("skip test on unix")
 			}
 
-			tempdir, cleanup := restictest.TempDir(t)
+			tempdir := restictest.TempDir(t)
-			defer cleanup()
-
 			TestCreateFiles(t, tempdir, test.src)
 
 			back := restictest.Chdir(t, tempdir)

internal/backend/layout/layout_test.go

@@ -14,8 +14,7 @@ import (
 )
 
 func TestDefaultLayout(t *testing.T) {
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		path string
@@ -141,8 +140,7 @@ func TestDefaultLayout(t *testing.T) {
 }
 
 func TestRESTLayout(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		restic.Handle
@@ -287,8 +285,7 @@ func TestRESTLayoutURLs(t *testing.T) {
 }
 
 func TestS3LegacyLayout(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		restic.Handle
@@ -355,8 +352,7 @@ func TestS3LegacyLayout(t *testing.T) {
 }
 
 func TestDetectLayout(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		filename string
@@ -393,8 +389,7 @@ func TestDetectLayout(t *testing.T) {
 }
 
 func TestParseLayout(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		layoutName        string
@@ -433,8 +428,7 @@ func TestParseLayout(t *testing.T) {
 }
 
 func TestParseLayoutInvalid(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var invalidNames = []string{
 		"foo", "bar", "local",

internal/backend/local/layout_test.go

@@ -10,8 +10,7 @@ import (
 )
 
 func TestLayout(t *testing.T) {
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		filename        string

internal/backend/local/local_internal_test.go

@@ -24,8 +24,7 @@ func TestNoSpacePermanent(t *testing.T) {
 		return nil, fmt.Errorf("not creating tempfile, %w", syscall.ENOSPC)
 	}
 
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
 
 	be, err := Open(context.Background(), Config{Path: dir, Connections: 2})
 	rtest.OK(t, err)

internal/backend/local/local_test.go

@@ -120,8 +120,7 @@ func removeAll(t testing.TB, dir string) {
 }
 
 func TestOpenNotExistingDirectory(t *testing.T) {
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
 
 	// local.Open must not create any files dirs in the repo
 	openclose(t, filepath.Join(dir, "repo"))

internal/backend/rclone/backend_test.go

@@ -13,7 +13,7 @@ import (
 )
 
 func newTestSuite(t testing.TB) *test.Suite {
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
 
 	return &test.Suite{
 		// NewConfig returns a config for a new temporary backend that will be used in tests.
@@ -43,13 +43,6 @@ func newTestSuite(t testing.TB) *test.Suite {
 			cfg := config.(rclone.Config)
 			return rclone.Open(cfg, nil)
 		},
-
-		// CleanupFn removes data created during the tests.
-		Cleanup: func(config interface{}) error {
-			t.Logf("cleanup dir %v", dir)
-			cleanup()
-			return nil
-		},
 	}
 }
 

internal/backend/rclone/internal_test.go

@@ -12,9 +12,7 @@ import (
 
 // restic should detect rclone exiting.
 func TestRcloneExit(t *testing.T) {
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
-
 	cfg := NewConfig()
 	cfg.Remote = dir
 	be, err := Open(cfg, nil)

internal/backend/rest/rest_test.go

@@ -112,9 +112,7 @@ func TestBackendREST(t *testing.T) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
-
 	serverURL, cleanup := runRESTServer(ctx, t, dir)
 	defer cleanup()
 
@@ -144,9 +142,7 @@ func BenchmarkBackendREST(t *testing.B) {
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
-
 	serverURL, cleanup := runRESTServer(ctx, t, dir)
 	defer cleanup()
 

internal/backend/s3/s3_test.go

@@ -102,7 +102,6 @@ type MinioTestConfig struct {
 	s3.Config
 
 	tempdir    string
-	removeTempdir func()
 	stopServer func()
 }
 
@@ -132,7 +131,7 @@ func newMinioTestSuite(ctx context.Context, t testing.TB) *test.Suite {
 		NewConfig: func() (interface{}, error) {
 			cfg := MinioTestConfig{}
 
-			cfg.tempdir, cfg.removeTempdir = rtest.TempDir(t)
+			cfg.tempdir = rtest.TempDir(t)
 			key, secret := newRandomCredentials(t)
 			cfg.stopServer = runMinio(ctx, t, cfg.tempdir, key, secret)
 
@@ -179,9 +178,6 @@ func newMinioTestSuite(ctx context.Context, t testing.TB) *test.Suite {
 			if cfg.stopServer != nil {
 				cfg.stopServer()
 			}
-			if cfg.removeTempdir != nil {
-				cfg.removeTempdir()
-			}
 			return nil
 		},
 	}

internal/backend/sftp/layout_test.go

@@ -16,8 +16,7 @@ func TestLayout(t *testing.T) {
 		t.Skip("sftp server binary not available")
 	}
 
-	path, cleanup := rtest.TempDir(t)
+	path := rtest.TempDir(t)
-	defer cleanup()
 
 	var tests = []struct {
 		filename        string

internal/backend/test/suite.go

@@ -60,9 +60,11 @@ func (s *Suite) RunTests(t *testing.T) {
 		return
 	}
 
+	if s.Cleanup != nil {
 		if err = s.Cleanup(s.Config); err != nil {
 			t.Fatal(err)
 		}
+	}
 }
 
 type testFunction struct {

internal/cache/backend_test.go

@@ -57,10 +57,7 @@ func randomData(n int) (restic.Handle, []byte) {
 
 func TestBackend(t *testing.T) {
 	be := mem.New()
-
+	c := TestNewCache(t)
-	c, cleanup := TestNewCache(t)
-	defer cleanup()
-
 	wbe := c.Wrap(be)
 
 	h, data := randomData(5234142)
@@ -128,10 +125,7 @@ func (be loadErrorBackend) Load(ctx context.Context, h restic.Handle, length int
 
 func TestErrorBackend(t *testing.T) {
 	be := mem.New()
-
+	c := TestNewCache(t)
-	c, cleanup := TestNewCache(t)
-	defer cleanup()
-
 	h, data := randomData(5234142)
 
 	// save directly in backend
@@ -174,9 +168,7 @@ func TestErrorBackend(t *testing.T) {
 
 func TestBackendRemoveBroken(t *testing.T) {
 	be := mem.New()
-
+	c := TestNewCache(t)
-	c, cleanup := TestNewCache(t)
-	defer cleanup()
 
 	h, data := randomData(5234142)
 	// save directly in backend

internal/cache/file_test.go

@@ -87,8 +87,7 @@ func TestFiles(t *testing.T) {
 	t.Logf("seed is %v", seed)
 	rand.Seed(seed)
 
-	c, cleanup := TestNewCache(t)
+	c := TestNewCache(t)
-	defer cleanup()
 
 	var tests = []restic.FileType{
 		restic.SnapshotFile,
@@ -140,8 +139,7 @@ func TestFileLoad(t *testing.T) {
 	t.Logf("seed is %v", seed)
 	rand.Seed(seed)
 
-	c, cleanup := TestNewCache(t)
+	c := TestNewCache(t)
-	defer cleanup()
 
 	// save about 5 MiB of data in the cache
 	data := test.Random(rand.Int(), 5234142)
@@ -223,10 +221,8 @@ func TestFileSaveConcurrent(t *testing.T) {
 
 	const nproc = 40
 
-	c, cleanup := TestNewCache(t)
-	defer cleanup()
-
 	var (
+		c    = TestNewCache(t)
 		data = test.Random(1, 10000)
 		g    errgroup.Group
 		id   restic.ID

internal/cache/testing.go

@@ -9,12 +9,12 @@ import (
 
 // TestNewCache returns a cache in a temporary directory which is removed when
 // cleanup is called.
-func TestNewCache(t testing.TB) (*Cache, func()) {
+func TestNewCache(t testing.TB) *Cache {
-	dir, cleanup := test.TempDir(t)
+	dir := test.TempDir(t)
 	t.Logf("created new cache at %v", dir)
 	cache, err := New(restic.NewRandomID().String(), dir)
 	if err != nil {
 		t.Fatal(err)
 	}
-	return cache, cleanup
+	return cache
 }

internal/dump/common_test.go

@@ -13,17 +13,12 @@ import (
 )
 
 func prepareTempdirRepoSrc(t testing.TB, src archiver.TestDir) (tempdir string, repo restic.Repository, cleanup func()) {
-	tempdir, removeTempdir := rtest.TempDir(t)
+	tempdir = rtest.TempDir(t)
 	repo, removeRepository := repository.TestRepository(t)
 
 	archiver.TestCreateFiles(t, tempdir, src)
 
-	cleanup = func() {
+	return tempdir, repo, removeRepository
-		removeRepository()
-		removeTempdir()
-	}
-
-	return tempdir, repo, cleanup
 }
 
 type CheckDump func(t *testing.T, testDir string, testDump *bytes.Buffer) error

internal/fs/stat_test.go

@@ -9,9 +9,7 @@ import (
 )
 
 func TestExtendedStat(t *testing.T) {
-	tempdir, cleanup := restictest.TempDir(t)
+	tempdir := restictest.TempDir(t)
-	defer cleanup()
-
 	filename := filepath.Join(tempdir, "file")
 	err := os.WriteFile(filename, []byte("foobar"), 0640)
 	if err != nil {

internal/restic/rewind_reader_test.go

@@ -27,9 +27,7 @@ func TestByteReader(t *testing.T) {
 func TestFileReader(t *testing.T) {
 	buf := []byte("foobar")
 
-	d, cleanup := test.TempDir(t)
+	d := test.TempDir(t)
-	defer cleanup()
-
 	filename := filepath.Join(d, "file-reader-test")
 	err := os.WriteFile(filename, buf, 0600)
 	if err != nil {

internal/restorer/filerestorer_test.go

@@ -185,8 +185,7 @@ func verifyRestore(t *testing.T, r *fileRestorer, repo *TestRepo) {
 }
 
 func TestFileRestorerBasic(t *testing.T) {
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
 
 	for _, sparse := range []bool{false, true} {
 		restoreAndVerify(t, tempdir, []TestFile{
@@ -217,8 +216,7 @@ func TestFileRestorerBasic(t *testing.T) {
 }
 
 func TestFileRestorerPackSkip(t *testing.T) {
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
 
 	files := make(map[string]bool)
 	files["file2"] = true
@@ -250,8 +248,7 @@ func TestFileRestorerPackSkip(t *testing.T) {
 }
 
 func TestErrorRestoreFiles(t *testing.T) {
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
 	content := []TestFile{
 		{
 			name: "file1",
@@ -282,8 +279,7 @@ func TestDownloadError(t *testing.T) {
 }
 
 func testPartialDownloadError(t *testing.T, part int) {
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
 	content := []TestFile{
 		{
 			name: "file1",

internal/restorer/fileswriter_test.go

@@ -8,9 +8,7 @@ import (
 )
 
 func TestFilesWriterBasic(t *testing.T) {
-	dir, cleanup := rtest.TempDir(t)
+	dir := rtest.TempDir(t)
-	defer cleanup()
-
 	w := newFilesWriter(1)
 
 	f1 := dir + "/f1"

internal/restorer/preallocate_test.go

@@ -14,8 +14,7 @@ import (
 func TestPreallocate(t *testing.T) {
 	for _, i := range []int64{0, 1, 4096, 1024 * 1024} {
 		t.Run(strconv.FormatInt(i, 10), func(t *testing.T) {
-			dirpath, cleanup := test.TempDir(t)
+			dirpath := test.TempDir(t)
-			defer cleanup()
 
 			flags := os.O_CREATE | os.O_TRUNC | os.O_WRONLY
 			wr, err := os.OpenFile(path.Join(dirpath, "test"), flags, 0600)

internal/restorer/restorer_test.go

@@ -328,9 +328,7 @@ func TestRestorer(t *testing.T) {
 
 			res := NewRestorer(context.TODO(), repo, sn, false)
 
-			tempdir, cleanup := rtest.TempDir(t)
+			tempdir := rtest.TempDir(t)
-			defer cleanup()
-
 			// make sure we're creating a new subdir of the tempdir
 			tempdir = filepath.Join(tempdir, "target")
 
@@ -448,9 +446,7 @@ func TestRestorerRelative(t *testing.T) {
 
 			res := NewRestorer(context.TODO(), repo, sn, false)
 
-			tempdir, cleanup := rtest.TempDir(t)
+			tempdir := rtest.TempDir(t)
-			defer cleanup()
-
 			cleanup = rtest.Chdir(t, tempdir)
 			defer cleanup()
 
@@ -682,9 +678,7 @@ func TestRestorerTraverseTree(t *testing.T) {
 
 			res.SelectFilter = test.Select
 
-			tempdir, cleanup := rtest.TempDir(t)
+			tempdir := rtest.TempDir(t)
-			defer cleanup()
-
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
@@ -776,9 +770,7 @@ func TestRestorerConsistentTimestampsAndPermissions(t *testing.T) {
 		return selectedForRestore, childMayBeSelected
 	}
 
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
-
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
@@ -818,9 +810,7 @@ func TestVerifyCancel(t *testing.T) {
 
 	res := NewRestorer(context.TODO(), repo, sn, false)
 
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
-
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 
@@ -863,9 +853,7 @@ func TestRestorerSparseFiles(t *testing.T) {
 
 	res := NewRestorer(context.TODO(), repo, sn, true)
 
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
-
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 

internal/restorer/restorer_unix_test.go

@@ -36,9 +36,7 @@ func TestRestorerRestoreEmptyHardlinkedFileds(t *testing.T) {
 		return true, true
 	}
 
-	tempdir, cleanup := rtest.TempDir(t)
+	tempdir := rtest.TempDir(t)
-	defer cleanup()
-
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
 

internal/test/helpers.go

@@ -191,22 +191,23 @@ func RemoveAll(t testing.TB, path string) {
 	OK(t, err)
 }
 
-// TempDir returns a temporary directory that is removed when cleanup is
+// TempDir returns a temporary directory that is removed by t.Cleanup,
-// called, except if TestCleanupTempDirs is set to false.
+// except if TestCleanupTempDirs is set to false.
-func TempDir(t testing.TB) (path string, cleanup func()) {
+func TempDir(t testing.TB) string {
 	tempdir, err := os.MkdirTemp(TestTempDir, "restic-test-")
 	if err != nil {
 		t.Fatal(err)
 	}
 
-	return tempdir, func() {
+	t.Cleanup(func() {
 		if !TestCleanupTempDirs {
 			t.Logf("leaving temporary directory %v used for test", tempdir)
 			return
 		}
 
 		RemoveAll(t, tempdir)
-	}
+	})
+	return tempdir
 }
 
 // Chdir changes the current directory to dest.