pr comments

babylonlabs-io · Dec 3, 2024 · c8c3dbf · c8c3dbf
1 parent b55f6a0
commit c8c3dbf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/x/incentive/keeper/msg_server_test.go b/x/incentive/keeper/msg_server_test.go
@@ -97,6 +97,10 @@ func FuzzSetWithdrawAddr(f *testing.F) {
 		})
 		require.NoError(t, err)
 
+		rgauge := ik.GetRewardGauge(ctx, sType, sAddr)
+		require.NotNil(t, rgauge)
+		require.False(t, rgauge.IsFullyWithdrawn())
+
 		// invoke withdraw and assert consistency
 		resp, err := ms.WithdrawReward(ctx, &types.MsgWithdrawReward{
 			Type:    sType.String(),

diff --git a/x/incentive/keeper/query_delegator.go b/x/incentive/keeper/query_delegator.go
@@ -13,10 +13,6 @@ func (k Keeper) DelegatorWithdrawAddress(goCtx context.Context, req *types.Query
 		return nil, status.Error(codes.InvalidArgument, "invalid request")
 	}
 
-	if req.DelegatorAddress == "" {
-		return nil, status.Error(codes.InvalidArgument, "empty delegator address")
-	}
-
 	ctx := sdk.UnwrapSDKContext(goCtx)
 	delegatorAddress, err := sdk.AccAddressFromBech32(req.DelegatorAddress)
 	if err != nil {