mirror of
https://github.com/restic/restic.git
synced 2025-12-08 06:09:56 +00:00
ac4642b479
The new method combines both step into a single wrapper function. Thus it ensures that both are always called in pairs. As an additional benefit this slightly reduces the boilerplate to upload blobs.
36 lines
902 B
Go
36 lines
902 B
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
rtest "github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
// Test saving a blob and loading it again, with varying buffer sizes.
|
|
// Also a regression test for #3783.
|
|
func FuzzSaveLoadBlob(f *testing.F) {
|
|
f.Fuzz(func(t *testing.T, blob []byte, buflen uint) {
|
|
if buflen > 64<<20 {
|
|
// Don't allocate enormous buffers. We're not testing the allocator.
|
|
t.Skip()
|
|
}
|
|
|
|
id := restic.Hash(blob)
|
|
repo, _, _ := TestRepositoryWithVersion(t, 2)
|
|
|
|
rtest.OK(t, repo.WithBlobUploader(context.TODO(), func(ctx context.Context) error {
|
|
_, _, _, err := repo.SaveBlob(ctx, restic.DataBlob, blob, id, false)
|
|
return err
|
|
}))
|
|
|
|
buf, err := repo.LoadBlob(context.TODO(), restic.DataBlob, id, make([]byte, buflen))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if restic.Hash(buf) != id {
|
|
t.Fatal("mismatch")
|
|
}
|
|
})
|
|
}
|