Addresss comments.

ChainSafe · Oct 26, 2021 · f862bad · f862bad
1 parent 8faf315
commit f862bad
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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)