arbotree: Add tree.Size, update arbo version

This commit is contained in:
arnaucube
2021-05-18 11:00:17 +02:00
parent 88cbe4f29f
commit aff9bcc186
4 changed files with 46 additions and 3 deletions

View File

@@ -20,3 +20,36 @@ func TestInterface(t *testing.T) {
t.Fatal("censustree interface not matched by arbotree wrapper")
}
}
func TestGenProof(t *testing.T) {
storage := t.TempDir()
tr1 := &Tree{}
err := tr1.Init("test1", storage)
if err != nil {
t.Fatal(err)
}
var keys, values [][]byte
for i := 0; i < 10; i++ {
keys = append(keys, []byte{byte(i)})
values = append(values, []byte{byte(i)})
err = tr1.Add([]byte{byte(i)}, []byte{byte(i)})
if err != nil {
t.Fatal(err)
}
}
for i := 0; i < 10; i++ {
p, err := tr1.GenProof(keys[i], values[i])
if err != nil {
t.Fatal(err)
}
v, err := tr1.CheckProof(keys[i], values[i], tr1.Root(), p)
if err != nil {
t.Fatal(err)
}
if !v {
t.Fatal("CheckProof failed")
}
}
}