From f862bad2d527da74908c3516be0164ede8340536 Mon Sep 17 00:00:00 2001
From: Arijit Das <arijit@chainsafe.io>
Date: Tue, 26 Oct 2021 19:23:53 +0530
Subject: [PATCH] Addresss comments.

---
 lib/trie/trie_test.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/lib/trie/trie_test.go b/lib/trie/trie_test.go
index f4ca9fc22b..4a2213cc83 100644
--- a/lib/trie/trie_test.go
+++ b/lib/trie/trie_test.go
@@ -1349,9 +1349,12 @@ func TestTrie_ClearPrefixLimit(t *testing.T) {
 						isAllDeleted = false
 						require.Equal(t, test.value, val)
 					}
+				} else {
+					require.NotNil(t, val)
 				}
 			}
 			require.Equal(t, num, deleteCount)
+			require.LessOrEqual(t, deleteCount, uint32(lim))
 			if lim > 0 {
 				require.Equal(t, allDeleted, isAllDeleted)
 			}
@@ -1476,8 +1479,11 @@ func TestTrie_ClearPrefixLimitSnapshot(t *testing.T) {
 							isAllDeleted = false
 							require.Equal(t, test.value, val)
 						}
+					} else {
+						require.NotNil(t, val)
 					}
 				}
+				require.LessOrEqual(t, deleteCount, uint32(lim))
 				require.Equal(t, num, deleteCount)
 				if lim > 0 {
 					require.Equal(t, allDeleted, isAllDeleted)