Skip to content

Commit 6237d28

Browse files
committed
SubtreeConsitency fuzztest + rename Consistency fuzztest args
1 parent 01bf93f commit 6237d28

1 file changed

Lines changed: 15 additions & 12 deletions

File tree

testonly/tree_fuzz_test.go

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,32 +12,32 @@ import (
1212

1313
// Compute and verify consistency proofs
1414
func FuzzConsistencyProofAndVerify(f *testing.F) {
15-
for size := 1; size <= 8; size++ {
16-
for end := 1; end <= size; end++ {
17-
for begin := 1; begin <= end; begin++ {
18-
f.Add(uint64(size), uint64(begin), uint64(end))
15+
for tSize := 1; tSize <= 8; tSize++ {
16+
for first := 1; first <= tSize; first++ {
17+
for second := 1; second <= first; second++ {
18+
f.Add(uint64(tSize), uint64(second), uint64(first))
1919
}
2020
}
2121
}
22-
f.Fuzz(func(t *testing.T, size, begin, end uint64) {
22+
f.Fuzz(func(t *testing.T, tSize, first, second uint64) {
2323
// necessary to restrict size for compile_native_go_fuzzer
24-
if size >= math.MaxUint16 {
24+
if tSize >= math.MaxUint16 {
2525
return
2626
}
27-
t.Logf("size=%d, begin=%d, end=%d", size, begin, end)
28-
if begin > end || end > size {
27+
t.Logf("size=%d, begin=%d, end=%d", tSize, first, second)
28+
if first > second || second > tSize {
2929
return
3030
}
31-
if begin == 0 {
31+
if first == 0 {
3232
return
3333
}
34-
tree := newTree(genEntries(size))
35-
p, err := tree.ConsistencyProof(begin, end)
34+
tree := newTree(genEntries(tSize))
35+
p, err := tree.ConsistencyProof(first, second)
3636
t.Logf("proof=%v", p)
3737
if err != nil {
3838
t.Error(err)
3939
}
40-
err = proof.VerifyConsistency(tree.hasher, begin, end, p, tree.HashAt(begin), tree.HashAt(end))
40+
err = proof.VerifyConsistency(tree.hasher, first, second, p, tree.HashAt(first), tree.HashAt(second))
4141
if err != nil {
4242
t.Error(err)
4343
}
@@ -70,6 +70,9 @@ func FuzzSubtreeConsistencyProofAndVerify(f *testing.F) {
7070
if begin > end || end > size {
7171
return
7272
}
73+
if begin == 0 && end > 0 {
74+
return
75+
}
7376
tree := newTree(genEntries(treeSize))
7477
p, err := tree.SubtreeConsistencyProof(begin, end, size)
7578
t.Logf("proof=%v", p)

0 commit comments

Comments
 (0)