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

Add missing emptyHash in Snapshot Tree

Browse Source
This commit is contained in:
arnaucube
2021-09-09 11:42:41 +02:00
parent de5914f453
commit 64c8f8d0bc
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tree.go
View File

@@ -860,6 +860,7 @@ func (t *Tree) Snapshot(fromRoot []byte) (*Tree, error) {
db: t.db, db: t.db,
maxLevels: t.maxLevels, maxLevels: t.maxLevels,
snapshotRoot: fromRoot, snapshotRoot: fromRoot,
emptyHash: t.emptyHash,
hashFunction: t.hashFunction, hashFunction: t.hashFunction,
dbg: t.dbg, dbg: t.dbg,
}, nil }, nil

21
tree_test.go
View File

@@ -563,6 +563,27 @@ func TestSnapshot(t *testing.T) {
"1025190963769001718196479367844646783678188389989148142691917685159698888868") "1025190963769001718196479367844646783678188389989148142691917685159698888868")
} }
func TestGetFromSnapshotExpectArboErrKeyNotFound(t *testing.T) {
c := qt.New(t)
database, err := badgerdb.New(badgerdb.Options{Path: c.TempDir()})
c.Assert(err, qt.IsNil)
tree, err := NewTree(database, 100, HashFunctionPoseidon)
c.Assert(err, qt.IsNil)
defer tree.db.Close() //nolint:errcheck
bLen := tree.HashFunction().Len()
k := BigIntToBytes(bLen, big.NewInt(int64(3)))
root, err := tree.Root()
c.Assert(err, qt.IsNil)
tree, err = tree.Snapshot(root)
c.Assert(err, qt.IsNil)
_, _, err = tree.Get(k)
c.Assert(err, qt.Equals, ErrKeyNotFound) // and not equal to db.ErrKeyNotFound
}
func BenchmarkAdd(b *testing.B) { func BenchmarkAdd(b *testing.B) {
bLen := 32 // for both Poseidon & Sha256 bLen := 32 // for both Poseidon & Sha256
// prepare inputs // prepare inputs