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

test(zkTrie): add deletion test in account update unit test (#344)

Browse source 
feat(zkTrie): extend account update unit tests with account deletion
This commit is contained in:
colin 2023-05-25 16:30:50 +08:00 committed by GitHub
parent ad43d0ea3f
commit 74725fbe2c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
trie/zk_trie_impl_test.go
View file

@ -86,6 +86,10 @@ func (mt *zkTrieImplTestWrapper) TryGet(key []byte) ([]byte, error) {
return mt.ZkTrieImpl.TryGet(zkt.NewHashFromBytes(key)) return mt.ZkTrieImpl.TryGet(zkt.NewHashFromBytes(key))
} }
func (mt *zkTrieImplTestWrapper) TryDelete(key []byte) error {
return mt.ZkTrieImpl.TryDelete(zkt.NewHashFromBytes(key))
}
// TryUpdateAccount will abstract the write of an account to the trie // TryUpdateAccount will abstract the write of an account to the trie
func (mt *zkTrieImplTestWrapper) TryUpdateAccount(key []byte, acc *types.StateAccount) error { func (mt *zkTrieImplTestWrapper) TryUpdateAccount(key []byte, acc *types.StateAccount) error {
value, flag := acc.MarshalFields() value, flag := acc.MarshalFields()
@ -271,4 +275,18 @@ func TestMerkleTree_UpdateAccount(t *testing.T) {
bt, err = mt.TryGet(common.HexToAddress("0x8dE13967F19410A7991D63c2c0179feBFDA0c261").Bytes()) bt, err = mt.TryGet(common.HexToAddress("0x8dE13967F19410A7991D63c2c0179feBFDA0c261").Bytes())
assert.Nil(t, err) assert.Nil(t, err)
assert.Nil(t, bt) assert.Nil(t, bt)
err = mt.TryDelete(common.HexToHash("0x05fDbDfaE180345C6Cff5316c286727CF1a43327").Bytes())
assert.Nil(t, err)
bt, err = mt.TryGet(common.HexToAddress("0x05fDbDfaE180345C6Cff5316c286727CF1a43327").Bytes())
assert.Nil(t, err)
assert.Nil(t, bt)
err = mt.TryDelete(common.HexToAddress("0x4cb1aB63aF5D8931Ce09673EbD8ae2ce16fD6571").Bytes())
assert.Nil(t, err)
bt, err = mt.TryGet(common.HexToAddress("0x4cb1aB63aF5D8931Ce09673EbD8ae2ce16fD6571").Bytes())
assert.Nil(t, err)
assert.Nil(t, bt)
} }