removed ethBalance check from addStake

razor-network · Jul 20, 2022 · 5b93119 · 5b93119
1 parent 8c3502d
commit 5b93119
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/cmd/addStake.go b/cmd/addStake.go
@@ -49,8 +49,6 @@ func (*UtilsStruct) ExecuteStake(flagSet *pflag.FlagSet) {
 
 	razorUtils.CheckAmountAndBalance(valueInWei, balance)
 
-	razorUtils.CheckEthBalanceIsZero(client, address)
-
 	minSafeRazor, err := utils.UtilsInterface.GetMinSafeRazor(client)
 	utils.CheckError("Error in getting minimum safe razor amount: ", err)
 

diff --git a/cmd/addStake_test.go b/cmd/addStake_test.go
@@ -297,7 +297,6 @@ func TestExecuteStake(t *testing.T) {
 			utilsMock.On("FetchBalance", mock.AnythingOfType("*ethclient.Client"), mock.AnythingOfType("string")).Return(tt.args.balance, tt.args.balanceErr)
 			cmdUtilsMock.On("AssignAmountInWei", flagSet).Return(tt.args.amount, tt.args.amountErr)
 			utilsMock.On("CheckAmountAndBalance", mock.AnythingOfType("*big.Int"), mock.AnythingOfType("*big.Int")).Return(tt.args.amount)
-			utilsMock.On("CheckEthBalanceIsZero", mock.AnythingOfType("*ethclient.Client"), mock.AnythingOfType("string")).Return()
 			utilsPkgMock.On("GetMinSafeRazor", mock.AnythingOfType("*ethclient.Client")).Return(tt.args.minSafeRazor, tt.args.minSafeRazorErr)
 			cmdUtilsMock.On("Approve", mock.Anything).Return(tt.args.approveTxn, tt.args.approveErr)
 			cmdUtilsMock.On("StakeCoins", mock.Anything).Return(tt.args.stakeTxn, tt.args.stakeErr)