forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-28 07:36:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

test(zktrie): add TestZkTrieDelete (#201)

Browse source 
update zktrie for deletion fix

Co-authored-by: HAOYUatHZ <37070449+HAOYUatHZ@users.noreply.github.com>
This commit is contained in:
Ho 2023-02-09 21:14:10 +08:00 committed by GitHub
parent 81e7775aa8
commit 3a080e5cb4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
trie/zk_trie_test.go
View file

@ -223,3 +223,38 @@ func BenchmarkZkTrieUpdate(b *testing.B) {
}
b.StopTimer()
}
func TestZkTrieDelete(t *testing.T) {
key := make([]byte, 32)
value := make([]byte, 32)
trie1 := newEmptyZkTrie()
var count int = 6
var hashes []common.Hash
hashes = append(hashes, trie1.Hash())
for i := 0; i < count; i++ {
binary.LittleEndian.PutUint64(key, uint64(i))
binary.LittleEndian.PutUint64(value, uint64(i))
err := trie1.TryUpdate(key, value)
assert.NoError(t, err)
hashes = append(hashes, trie1.Hash())
}
// binary.LittleEndian.PutUint64(key, uint64(0xffffff))
// err := trie1.TryDelete(key)
// assert.Equal(t, err, zktrie.ErrKeyNotFound)
trie1.Commit(nil)
for i := count - 1; i >= 0; i-- {
binary.LittleEndian.PutUint64(key, uint64(i))
v, err := trie1.TryGet(key)
assert.NoError(t, err)
assert.NotEmpty(t, v)
err = trie1.TryDelete(key)
assert.NoError(t, err)
hash := trie1.Hash()
assert.Equal(t, hashes[i].Hex(), hash.Hex())
}
}