test: tx

0glabs · Nov 4, 2024 · 605a71e · 605a71e
1 parent 1602c96
commit 605a71e
Show file tree

Hide file tree

Showing 8 changed files with 395 additions and 31 deletions.
diff --git a/precompiles/interfaces/contracts/IStaking.sol b/precompiles/interfaces/contracts/IStaking.sol
@@ -22,7 +22,6 @@ struct CommissionRates {
     uint maxChangeRate; // 18 decimals
 }
 
-
 /**
  * @dev Commission defines the commission parameters.
  */
@@ -207,7 +206,7 @@ interface IStaking {
     function delegate(
         string memory validatorAddress,
         uint amount // in bond denom
-    ) external returns (bool success);
+    ) external;
 
     /**
      * @dev BeginRedelegate defines a method for performing a redelegationA

diff --git a/precompiles/staking/IStaking.abi b/precompiles/staking/IStaking.abi
@@ -142,13 +142,7 @@
       }
     ],
     "name": "delegate",
-    "outputs": [
-      {
-        "internalType": "bool",
-        "name": "success",
-        "type": "bool"
-      }
-    ],
+    "outputs": [],
     "stateMutability": "nonpayable",
     "type": "function"
   },

diff --git a/precompiles/staking/contract.go b/precompiles/staking/contract.go
diff --git a/precompiles/staking/staking_test.go b/precompiles/staking/staking_test.go
@@ -103,6 +103,8 @@ func (suite *StakingTestSuite) setupValidator(signer *testutil.TestSigner) {
 	suite.Assert().NoError(err)
 	_, err = suite.runTx(input, signer, 10000000)
 	suite.Assert().NoError(err)
+	_, err = suite.stakingKeeper.ApplyAndReturnValidatorSetUpdates(suite.Ctx)
+	suite.Assert().NoError(err)
 }
 
 func (suite *StakingTestSuite) firstBondedValidator() (sdk.ValAddress, error) {

diff --git a/precompiles/staking/tx.go b/precompiles/staking/tx.go
@@ -2,6 +2,7 @@ package staking
 
 import (
 	"fmt"
+	"math/big"
 
 	precopmiles_common "github.com/0glabs/0g-chain/precompiles/common"
 	sdk "github.com/cosmos/cosmos-sdk/types"
@@ -103,12 +104,12 @@ func (s *StakingPrecompile) BeginRedelegate(
 		return nil, fmt.Errorf(precopmiles_common.ErrSenderNotOrigin)
 	}
 	// execute
-	_, err = stakingkeeper.NewMsgServerImpl(s.stakingKeeper).BeginRedelegate(ctx, msg)
+	response, err := stakingkeeper.NewMsgServerImpl(s.stakingKeeper).BeginRedelegate(ctx, msg)
 	if err != nil {
 		return nil, err
 	}
 	// emit events
-	return method.Outputs.Pack()
+	return method.Outputs.Pack(big.NewInt(response.CompletionTime.UTC().Unix()))
 }
 
 func (s *StakingPrecompile) Undelegate(
@@ -128,12 +129,12 @@ func (s *StakingPrecompile) Undelegate(
 		return nil, fmt.Errorf(precopmiles_common.ErrSenderNotOrigin)
 	}
 	// execute
-	_, err = stakingkeeper.NewMsgServerImpl(s.stakingKeeper).Undelegate(ctx, msg)
+	response, err := stakingkeeper.NewMsgServerImpl(s.stakingKeeper).Undelegate(ctx, msg)
 	if err != nil {
 		return nil, err
 	}
 	// emit events
-	return method.Outputs.Pack()
+	return method.Outputs.Pack(big.NewInt(response.CompletionTime.UTC().Unix()))
 }
 
 func (s *StakingPrecompile) CancelUnbondingDelegation(