mirror of
https://github.com/restic/restic.git
synced 2024-12-25 17:27:25 +00:00
c0b5ec55ab
TestRepository and its variants always returned no-op cleanup functions. If they ever do need to do cleanup, using testing.T.Cleanup is easier than passing these functions around.
97 lines
2.4 KiB
Go
97 lines
2.4 KiB
Go
package dump
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/restic/restic/internal/archiver"
|
|
"github.com/restic/restic/internal/fs"
|
|
"github.com/restic/restic/internal/repository"
|
|
"github.com/restic/restic/internal/restic"
|
|
rtest "github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
func prepareTempdirRepoSrc(t testing.TB, src archiver.TestDir) (string, restic.Repository) {
|
|
tempdir := rtest.TempDir(t)
|
|
repo := repository.TestRepository(t)
|
|
|
|
archiver.TestCreateFiles(t, tempdir, src)
|
|
|
|
return tempdir, repo
|
|
}
|
|
|
|
type CheckDump func(t *testing.T, testDir string, testDump *bytes.Buffer) error
|
|
|
|
func WriteTest(t *testing.T, format string, cd CheckDump) {
|
|
tests := []struct {
|
|
name string
|
|
args archiver.TestDir
|
|
target string
|
|
}{
|
|
{
|
|
name: "single file in root",
|
|
args: archiver.TestDir{
|
|
"file": archiver.TestFile{Content: "string"},
|
|
},
|
|
target: "/",
|
|
},
|
|
{
|
|
name: "multiple files in root",
|
|
args: archiver.TestDir{
|
|
"file1": archiver.TestFile{Content: "string"},
|
|
"file2": archiver.TestFile{Content: "string"},
|
|
},
|
|
target: "/",
|
|
},
|
|
{
|
|
name: "multiple files and folders in root",
|
|
args: archiver.TestDir{
|
|
"file1": archiver.TestFile{Content: "string"},
|
|
"file2": archiver.TestFile{Content: "string"},
|
|
"firstDir": archiver.TestDir{
|
|
"another": archiver.TestFile{Content: "string"},
|
|
},
|
|
"secondDir": archiver.TestDir{
|
|
"another2": archiver.TestFile{Content: "string"},
|
|
},
|
|
},
|
|
target: "/",
|
|
},
|
|
{
|
|
name: "file and symlink in root",
|
|
args: archiver.TestDir{
|
|
"file1": archiver.TestFile{Content: "string"},
|
|
"file2": archiver.TestSymlink{Target: "file1"},
|
|
},
|
|
target: "/",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
tmpdir, repo := prepareTempdirRepoSrc(t, tt.args)
|
|
arch := archiver.New(repo, fs.Track{FS: fs.Local{}}, archiver.Options{})
|
|
|
|
back := rtest.Chdir(t, tmpdir)
|
|
defer back()
|
|
|
|
sn, _, err := arch.Snapshot(ctx, []string{"."}, archiver.SnapshotOptions{})
|
|
rtest.OK(t, err)
|
|
|
|
tree, err := restic.LoadTree(ctx, repo, *sn.Tree)
|
|
rtest.OK(t, err)
|
|
|
|
dst := &bytes.Buffer{}
|
|
d := New(format, repo, dst)
|
|
if err := d.DumpTree(ctx, tree, tt.target); err != nil {
|
|
t.Fatalf("Dumper.Run error = %v", err)
|
|
}
|
|
if err := cd(t, tmpdir, dst); err != nil {
|
|
t.Errorf("WriteDump() = does not match: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|