Update key_test.go

ava-labs · Oct 17, 2023 · c9c562f · c9c562f
1 parent b06be10
commit c9c562f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/x/merkledb/key_test.go b/x/merkledb/key_test.go
@@ -157,6 +157,10 @@ func Test_Key_Take(t *testing.T) {
 	require := require.New(t)
 
 	for _, bf := range branchFactors {
+		// take more tokens than the key has
+		require.Equal(ToKey([]byte{0}, bf).Take(1000), ToKey([]byte{0}, bf))
+
+		// take no tokens
 		require.Equal(ToKey([]byte{0}, bf).Take(0), emptyKey(bf))
 		if bf == BranchFactor256 {
 			continue