arnaucube/arbo
1
1
Fork 0
mirror of https://github.com/arnaucube/arbo.git synced 2026-01-15 01:41:28 +01:00
Code Issues Projects Releases Wiki Activity

Add tree.Get

Browse Source
This commit is contained in:
arnaucube
2021-03-31 23:05:10 +02:00
parent bde87ca844
commit 4cd2ff6182
2 changed files with 36 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tree_test.go
View File

@@ -145,6 +145,26 @@ func TestAux(t *testing.T) {
assert.Nil(t, err)
}
func TestGet(t *testing.T) {
tree, err := NewTree(memory.NewMemoryStorage(), 100, HashFunctionPoseidon)
require.Nil(t, err)
defer tree.db.Close()
for i := 0; i < 10; i++ {
k := BigIntToBytes(big.NewInt(int64(i)))
v := BigIntToBytes(big.NewInt(int64(i * 2)))
if err := tree.Add(k, v); err != nil {
t.Fatal(err)
}
}
k := BigIntToBytes(big.NewInt(int64(7)))
gettedKey, gettedValue, err := tree.Get(k)
assert.Nil(t, err)
assert.Equal(t, k, gettedKey)
assert.Equal(t, BigIntToBytes(big.NewInt(int64(7*2))), gettedValue)
}
func TestGenProofAndVerify(t *testing.T) {
tree, err := NewTree(memory.NewMemoryStorage(), 100, HashFunctionPoseidon)
require.Nil(t, err)