mirror of
https://github.com/restic/restic.git
synced 2025-03-30 00:00:14 +01:00
d129baba7a
This method had a buffer argument, but that was nil at all call sites. That's removed, and instead LoadUnpacked now reuses whatever it allocates inside its retry loop.
65 lines
1.4 KiB
Go
65 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(ctx 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(ctx 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)
|
|
}
|