mirror of https://github.com/octoleo/restic.git
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package restic_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
rtest "github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
type saver struct {
|
|
fn func(restic.FileType, []byte) (restic.ID, error)
|
|
}
|
|
|
|
func (s saver) SaveUnpacked(_ context.Context, t restic.FileType, buf []byte) (restic.ID, error) {
|
|
return s.fn(t, buf)
|
|
}
|
|
|
|
func (s saver) Connections() uint {
|
|
return 2
|
|
}
|
|
|
|
type loader struct {
|
|
fn func(restic.FileType, restic.ID) ([]byte, error)
|
|
}
|
|
|
|
func (l loader) LoadUnpacked(_ context.Context, t restic.FileType, id restic.ID) (data []byte, err error) {
|
|
return l.fn(t, id)
|
|
}
|
|
|
|
func (l loader) Connections() uint {
|
|
return 2
|
|
}
|
|
|
|
func TestConfig(t *testing.T) {
|
|
var resultBuf []byte
|
|
save := func(tpe restic.FileType, buf []byte) (restic.ID, error) {
|
|
rtest.Assert(t, tpe == restic.ConfigFile,
|
|
"wrong backend type: got %v, wanted %v",
|
|
tpe, restic.ConfigFile)
|
|
|
|
resultBuf = buf
|
|
return restic.ID{}, nil
|
|
}
|
|
|
|
cfg1, err := restic.CreateConfig(restic.MaxRepoVersion)
|
|
rtest.OK(t, err)
|
|
|
|
err = restic.SaveConfig(context.TODO(), saver{save}, cfg1)
|
|
rtest.OK(t, err)
|
|
|
|
load := func(tpe restic.FileType, id restic.ID) ([]byte, error) {
|
|
rtest.Assert(t, tpe == restic.ConfigFile,
|
|
"wrong backend type: got %v, wanted %v",
|
|
tpe, restic.ConfigFile)
|
|
|
|
return resultBuf, nil
|
|
}
|
|
|
|
cfg2, err := restic.LoadConfig(context.TODO(), loader{load})
|
|
rtest.OK(t, err)
|
|
|
|
rtest.Assert(t, cfg1 == cfg2,
|
|
"configs aren't equal: %v != %v", cfg1, cfg2)
|
|
}
|