mirror of
https://github.com/arnaucube/kesto.git
synced 2026-02-07 19:46:40 +01:00
arbotree: Add tree.Size, update arbo version
This commit is contained in:
@@ -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")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user