minSafeRazor fix (#669)

razor-network · Apr 6, 2022 · 457fede · 457fede
1 parent 9979d65
commit 457fede
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 45 deletions.
diff --git a/utils/block.go b/utils/block.go
@@ -102,26 +102,6 @@ func (*UtilsStruct) GetMinStakeAmount(client *ethclient.Client) (*big.Int, error
 	return minStake, nil
 }
 
-func (*UtilsStruct) GetMinSafeRazor(client *ethclient.Client) (*big.Int, error) {
-	var (
-		minStake *big.Int
-		err      error
-	)
-	err = retry.Do(
-		func() error {
-			minStake, err = BlockManagerInterface.MinStake(client)
-			if err != nil {
-				log.Error("Error in fetching minimum stake amount.... Retrying")
-				return err
-			}
-			return nil
-		}, RetryInterface.RetryAttempts(core.MaxRetries))
-	if err != nil {
-		return nil, err
-	}
-	return minStake, nil
-}
-
 func (*UtilsStruct) GetMaxAltBlocks(client *ethclient.Client) (uint8, error) {
 	var (
 		maxAltBlocks uint8

diff --git a/utils/interface.go b/utils/interface.go
@@ -250,7 +250,7 @@ type StakeManagerUtils interface {
 	GetStakerId(client *ethclient.Client, address common.Address) (uint32, error)
 	GetStaker(*ethclient.Client, uint32) (bindings.StructsStaker, error)
 	GetNumStakers(*ethclient.Client) (uint32, error)
-	GetMinSafeRazor(client *ethclient.Client) (*big.Int, error)
+	MinSafeRazor(client *ethclient.Client) (*big.Int, error)
 	Locks(client *ethclient.Client, address common.Address, address1 common.Address, lockType uint8) (types.Locks, error)
 	MaxCommission(*ethclient.Client) (uint8, error)
 	EpochLimitForUpdateCommission(*ethclient.Client) (uint16, error)

diff --git a/utils/mocks/stake_manager_utils.go b/utils/mocks/stake_manager_utils.go
diff --git a/utils/stake.go b/utils/stake.go
@@ -190,3 +190,23 @@ func (*UtilsStruct) GetStakerSRZRBalance(client *ethclient.Client, staker bindin
 	}
 	return sRZRBalance, nil
 }
+
+func (*UtilsStruct) GetMinSafeRazor(client *ethclient.Client) (*big.Int, error) {
+	var (
+		minSafeRazor *big.Int
+		err          error
+	)
+	err = retry.Do(
+		func() error {
+			minSafeRazor, err = StakeManagerInterface.MinSafeRazor(client)
+			if err != nil {
+				log.Error("Error in fetching minimum safe razor.... Retrying")
+				return err
+			}
+			return nil
+		}, RetryInterface.RetryAttempts(core.MaxRetries))
+	if err != nil {
+		return nil, err
+	}
+	return minSafeRazor, nil
+}
diff --git a/utils/struct-utils.go b/utils/struct-utils.go
@@ -151,7 +151,7 @@ func (s StakeManagerStruct) GetNumStakers(client *ethclient.Client) (uint32, err
 	return stakeManager.GetNumStakers(&opts)
 }
 
-func (s StakeManagerStruct) GetMinSafeRazor(client *ethclient.Client) (*big.Int, error) {
+func (s StakeManagerStruct) MinSafeRazor(client *ethclient.Client) (*big.Int, error) {
 	stakeManager, opts := UtilsInterface.GetStakeManagerWithOpts(client)
 	return stakeManager.MinSafeRazor(&opts)
 }