From 74725fbe2c025d1a2b2314cf0a4670457743ca60 Mon Sep 17 00:00:00 2001
From: colin <102356659+colinlyguo@users.noreply.github.com>
Date: Thu, 25 May 2023 16:30:50 +0800
Subject: [PATCH] test(zkTrie): add deletion test in account update unit test
 (#344)

feat(zkTrie): extend account update unit tests with account deletion
---
 trie/zk_trie_impl_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/trie/zk_trie_impl_test.go b/trie/zk_trie_impl_test.go
index c42ab8c053..9710b6b4f1 100644
--- a/trie/zk_trie_impl_test.go
+++ b/trie/zk_trie_impl_test.go
@@ -86,6 +86,10 @@ func (mt *zkTrieImplTestWrapper) TryGet(key []byte) ([]byte, error) {
 	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
 func (mt *zkTrieImplTestWrapper) TryUpdateAccount(key []byte, acc *types.StateAccount) error {
 	value, flag := acc.MarshalFields()
@@ -271,4 +275,18 @@ func TestMerkleTree_UpdateAccount(t *testing.T) {
 	bt, err = mt.TryGet(common.HexToAddress("0x8dE13967F19410A7991D63c2c0179feBFDA0c261").Bytes())
 	assert.Nil(t, err)
 	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)
 }