forked from TrueCloudLab/restic
Do not create duplicate content for tests
This commit is contained in:
parent
f1bc181c5b
commit
035d0aeb31
1 changed files with 29 additions and 3 deletions
|
@ -1,6 +1,7 @@
|
||||||
package restic
|
package restic
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
|
@ -34,10 +35,14 @@ func saveFile(t testing.TB, repo *repository.Repository, rd io.Reader) (blobs ba
|
||||||
t.Fatalf("unable to save chunk in repo: %v", err)
|
t.Fatalf("unable to save chunk in repo: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
id, err := repo.SaveAndEncrypt(pack.Data, chunk.Data, nil)
|
id := backend.Hash(chunk.Data)
|
||||||
if err != nil {
|
if !repo.Index().Has(id) {
|
||||||
t.Fatalf("error saving chunk: %v", err)
|
_, err := repo.SaveAndEncrypt(pack.Data, chunk.Data, &id)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("error saving chunk: %v", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
blobs = append(blobs, id)
|
blobs = append(blobs, id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -50,6 +55,23 @@ const (
|
||||||
maxNodes = 32
|
maxNodes = 32
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func treeIsKnown(t testing.TB, repo *repository.Repository, tree *Tree) (bool, backend.ID) {
|
||||||
|
data, err := json.Marshal(tree)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("json.Marshal(tree) returned error: %v", err)
|
||||||
|
return false, backend.ID{}
|
||||||
|
}
|
||||||
|
data = append(data, '\n')
|
||||||
|
|
||||||
|
// check if tree has been saved before
|
||||||
|
id := backend.Hash(data)
|
||||||
|
if repo.Index().Has(id) {
|
||||||
|
return true, id
|
||||||
|
}
|
||||||
|
|
||||||
|
return false, id
|
||||||
|
}
|
||||||
|
|
||||||
// saveTree saves a tree of fake files in the repo and returns the ID.
|
// saveTree saves a tree of fake files in the repo and returns the ID.
|
||||||
func saveTree(t testing.TB, repo *repository.Repository, seed int64, depth int) backend.ID {
|
func saveTree(t testing.TB, repo *repository.Repository, seed int64, depth int) backend.ID {
|
||||||
rnd := rand.NewSource(seed)
|
rnd := rand.NewSource(seed)
|
||||||
|
@ -88,6 +110,10 @@ func saveTree(t testing.TB, repo *repository.Repository, seed int64, depth int)
|
||||||
tree.Nodes = append(tree.Nodes, node)
|
tree.Nodes = append(tree.Nodes, node)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if known, id := treeIsKnown(t, repo, &tree); known {
|
||||||
|
return id
|
||||||
|
}
|
||||||
|
|
||||||
id, err := repo.SaveJSON(pack.Tree, tree)
|
id, err := repo.SaveJSON(pack.Tree, tree)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
|
|
Loading…
Reference in a new issue