restic/internal/repository/repack_test.go

242 lines
5.5 KiB
Go
Raw Normal View History

2016-08-01 18:04:23 +00:00
package repository_test
import (
2017-06-05 21:56:59 +00:00
"context"
2016-08-01 18:24:15 +00:00
"io"
"math/rand"
"testing"
2017-07-23 12:21:03 +00:00
"github.com/restic/restic/internal/index"
"github.com/restic/restic/internal/repository"
2017-07-24 15:42:25 +00:00
"github.com/restic/restic/internal/restic"
)
2016-08-01 18:24:15 +00:00
func randomSize(min, max int) int {
return rand.Intn(max-min) + min
}
func random(t testing.TB, length int) []byte {
2016-08-31 18:29:54 +00:00
rd := restic.NewRandReader(rand.New(rand.NewSource(int64(length))))
2016-08-01 18:24:15 +00:00
buf := make([]byte, length)
_, err := io.ReadFull(rd, buf)
if err != nil {
t.Fatalf("unable to read %d random bytes: %v", length, err)
}
return buf
}
func createRandomBlobs(t testing.TB, repo restic.Repository, blobs int, pData float32) {
2016-08-01 18:24:15 +00:00
for i := 0; i < blobs; i++ {
var (
2016-08-31 18:58:57 +00:00
tpe restic.BlobType
2016-08-01 18:24:15 +00:00
length int
)
if rand.Float32() < pData {
2016-08-31 21:07:50 +00:00
tpe = restic.DataBlob
length = randomSize(10*1024, 1024*1024) // 10KiB to 1MiB of data
2016-08-01 18:24:15 +00:00
} else {
2016-08-31 21:07:50 +00:00
tpe = restic.TreeBlob
length = randomSize(1*1024, 20*1024) // 1KiB to 20KiB
2016-08-01 18:24:15 +00:00
}
buf := random(t, length)
2016-08-31 18:29:54 +00:00
id := restic.Hash(buf)
2016-08-31 21:07:50 +00:00
if repo.Index().Has(id, restic.DataBlob) {
t.Errorf("duplicate blob %v/%v ignored", id, restic.DataBlob)
continue
}
2017-06-05 21:56:59 +00:00
_, err := repo.SaveBlob(context.TODO(), tpe, buf, id)
2016-08-01 18:24:15 +00:00
if err != nil {
t.Fatalf("SaveFrom() error %v", err)
}
if rand.Float32() < 0.2 {
if err = repo.Flush(context.Background()); err != nil {
2016-08-01 18:24:15 +00:00
t.Fatalf("repo.Flush() returned error %v", err)
}
}
}
if err := repo.Flush(context.Background()); err != nil {
2016-08-01 18:24:15 +00:00
t.Fatalf("repo.Flush() returned error %v", err)
}
}
// selectBlobs splits the list of all blobs randomly into two lists. A blob
// will be contained in the firstone ith probability p.
func selectBlobs(t *testing.T, repo restic.Repository, p float32) (list1, list2 restic.BlobSet) {
2016-08-31 21:07:50 +00:00
list1 = restic.NewBlobSet()
list2 = restic.NewBlobSet()
2016-08-01 18:24:15 +00:00
2016-08-31 21:07:50 +00:00
blobs := restic.NewBlobSet()
2017-06-05 21:56:59 +00:00
for id := range repo.List(context.TODO(), restic.DataFile) {
entries, _, err := repo.ListPack(context.TODO(), id)
2016-08-01 18:24:15 +00:00
if err != nil {
t.Fatalf("error listing pack %v: %v", id, err)
}
for _, entry := range entries {
2016-08-31 21:07:50 +00:00
h := restic.BlobHandle{ID: entry.ID, Type: entry.Type}
if blobs.Has(h) {
t.Errorf("ignoring duplicate blob %v", h)
continue
}
blobs.Insert(h)
2016-08-01 18:24:15 +00:00
if rand.Float32() <= p {
2016-08-31 21:07:50 +00:00
list1.Insert(restic.BlobHandle{ID: entry.ID, Type: entry.Type})
2016-08-01 18:24:15 +00:00
} else {
2016-08-31 21:07:50 +00:00
list2.Insert(restic.BlobHandle{ID: entry.ID, Type: entry.Type})
2016-08-01 18:24:15 +00:00
}
2016-08-01 18:24:15 +00:00
}
}
return list1, list2
}
func listPacks(t *testing.T, repo restic.Repository) restic.IDSet {
2016-08-31 18:29:54 +00:00
list := restic.NewIDSet()
2017-06-05 21:56:59 +00:00
for id := range repo.List(context.TODO(), restic.DataFile) {
2016-08-01 18:24:15 +00:00
list.Insert(id)
}
return list
}
func findPacksForBlobs(t *testing.T, repo restic.Repository, blobs restic.BlobSet) restic.IDSet {
2016-08-31 18:29:54 +00:00
packs := restic.NewIDSet()
2016-08-01 18:24:15 +00:00
idx := repo.Index()
for h := range blobs {
list, err := idx.Lookup(h.ID, h.Type)
2016-08-01 18:24:15 +00:00
if err != nil {
t.Fatal(err)
}
for _, pb := range list {
packs.Insert(pb.PackID)
}
2016-08-01 18:24:15 +00:00
}
return packs
}
func repack(t *testing.T, repo restic.Repository, packs restic.IDSet, blobs restic.BlobSet) {
repackedBlobs, err := repository.Repack(context.TODO(), repo, packs, blobs, nil)
if err != nil {
t.Fatal(err)
}
for id := range repackedBlobs {
err = repo.Backend().Remove(context.TODO(), restic.Handle{Type: restic.DataFile, Name: id.String()})
if err != nil {
t.Fatal(err)
}
}
}
func saveIndex(t *testing.T, repo restic.Repository) {
2017-06-05 21:56:59 +00:00
if err := repo.SaveIndex(context.TODO()); err != nil {
t.Fatalf("repo.SaveIndex() %v", err)
}
}
func rebuildIndex(t *testing.T, repo restic.Repository) {
idx, _, err := index.New(context.TODO(), repo, restic.NewIDSet(), nil)
if err != nil {
t.Fatal(err)
}
2017-06-05 21:56:59 +00:00
for id := range repo.List(context.TODO(), restic.IndexFile) {
h := restic.Handle{
Type: restic.IndexFile,
Name: id.String(),
2017-06-05 21:56:59 +00:00
}
err = repo.Backend().Remove(context.TODO(), h)
if err != nil {
t.Fatal(err)
}
}
2017-06-05 21:56:59 +00:00
_, err = idx.Save(context.TODO(), repo, nil)
if err != nil {
t.Fatal(err)
}
}
func reloadIndex(t *testing.T, repo restic.Repository) {
repo.SetIndex(repository.NewMasterIndex())
2017-06-05 21:56:59 +00:00
if err := repo.LoadIndex(context.TODO()); err != nil {
t.Fatalf("error loading new index: %v", err)
}
}
func TestRepack(t *testing.T) {
repo, cleanup := repository.TestRepository(t)
defer cleanup()
createRandomBlobs(t, repo, 100, 0.7)
packsBefore := listPacks(t, repo)
// Running repack on empty ID sets should not do anything at all.
repack(t, repo, nil, nil)
packsAfter := listPacks(t, repo)
if !packsAfter.Equals(packsBefore) {
t.Fatalf("packs are not equal, Repack modified something. Before:\n %v\nAfter:\n %v",
packsBefore, packsAfter)
}
saveIndex(t, repo)
removeBlobs, keepBlobs := selectBlobs(t, repo, 0.2)
removePacks := findPacksForBlobs(t, repo, removeBlobs)
repack(t, repo, removePacks, keepBlobs)
rebuildIndex(t, repo)
reloadIndex(t, repo)
packsAfter = listPacks(t, repo)
for id := range removePacks {
if packsAfter.Has(id) {
t.Errorf("pack %v still present although it should have been repacked and removed", id.Str())
}
}
idx := repo.Index()
for h := range keepBlobs {
list, err := idx.Lookup(h.ID, h.Type)
if err != nil {
t.Errorf("unable to find blob %v in repo", h.ID.Str())
continue
}
if len(list) != 1 {
t.Errorf("expected one pack in the list, got: %v", list)
continue
}
pb := list[0]
if removePacks.Has(pb.PackID) {
t.Errorf("lookup returned pack ID %v that should've been removed", pb.PackID)
}
}
for h := range removeBlobs {
if _, err := idx.Lookup(h.ID, h.Type); err == nil {
t.Errorf("blob %v still contained in the repo", h)
}
}
}