restic/internal/backend/sftp/layout_test.go

88 lines
2 KiB
Go
Raw Normal View History

2017-04-10 20:35:03 +00:00
package sftp_test
import (
2017-06-03 15:39:57 +00:00
"context"
2017-04-10 20:35:03 +00:00
"fmt"
"path/filepath"
"restic"
2017-04-10 20:35:03 +00:00
"restic/backend/sftp"
. "restic/test"
"testing"
)
func TestLayout(t *testing.T) {
2017-05-01 20:55:09 +00:00
if sftpServer == "" {
t.Skip("sftp server binary not available")
}
2017-04-10 20:35:03 +00:00
path, cleanup := TempDir(t)
defer cleanup()
var tests = []struct {
filename string
layout string
failureExpected bool
datafiles map[string]bool
2017-04-10 20:35:03 +00:00
}{
2017-05-15 20:11:09 +00:00
{"repo-layout-default.tar.gz", "", false, map[string]bool{
"aa464e9fd598fe4202492ee317ffa728e82fa83a1de1a61996e5bd2d6651646c": false,
"fc919a3b421850f6fa66ad22ebcf91e433e79ffef25becf8aef7c7b1eca91683": false,
"c089d62788da14f8b7cbf77188305c0874906f0b73d3fce5a8869050e8d0c0e1": false,
}},
2017-05-15 20:11:09 +00:00
{"repo-layout-s3legacy.tar.gz", "", false, map[string]bool{
"fc919a3b421850f6fa66ad22ebcf91e433e79ffef25becf8aef7c7b1eca91683": false,
"c089d62788da14f8b7cbf77188305c0874906f0b73d3fce5a8869050e8d0c0e1": false,
"aa464e9fd598fe4202492ee317ffa728e82fa83a1de1a61996e5bd2d6651646c": false,
}},
2017-04-10 20:35:03 +00:00
}
for _, test := range tests {
t.Run(test.filename, func(t *testing.T) {
SetupTarTestFixture(t, path, filepath.Join("..", "testdata", test.filename))
repo := filepath.Join(path, "repo")
be, err := sftp.Open(sftp.Config{
2017-05-01 20:55:09 +00:00
Command: fmt.Sprintf("%q -e", sftpServer),
2017-04-10 20:35:03 +00:00
Path: repo,
Layout: test.layout,
})
if err != nil {
t.Fatal(err)
}
if be == nil {
t.Fatalf("Open() returned nil but no error")
}
datafiles := make(map[string]bool)
2017-06-03 15:39:57 +00:00
for id := range be.List(context.TODO(), restic.DataFile) {
datafiles[id] = false
}
if len(datafiles) == 0 {
t.Errorf("List() returned zero data files")
}
for id := range test.datafiles {
if _, ok := datafiles[id]; !ok {
t.Errorf("datafile with id %v not found", id)
}
datafiles[id] = true
}
for id, v := range datafiles {
if !v {
t.Errorf("unexpected id %v found", id)
}
}
if err = be.Close(); err != nil {
t.Errorf("Close() returned error %v", err)
}
2017-04-10 20:35:03 +00:00
RemoveAll(t, filepath.Join(path, "repo"))
})
}
}