Fix unit tests

babylonlabs-io · Oct 2, 2024 · f5b2e32 · f5b2e32
1 parent 722daa8
commit f5b2e32
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/testutil/datagen/btcstaking.go b/testutil/datagen/btcstaking.go
@@ -103,6 +103,8 @@ func GenRandomBTCDelegation(
  }
  staker := GenRandomAccount()
 
+ stakingTime := uint16(endHeight - startHeight)
+
  // staking/slashing tx
  stakingSlashingInfo := GenBTCStakingSlashingInfo(
  r,
@@ -112,7 +114,7 @@ func GenRandomBTCDelegation(
  fpPKs,
  covenantPks,
  covenantQuorum,
- uint16(endHeight-startHeight),
+ stakingTime,
  int64(totalSat),
  slashingPkScript,
  slashingRate,
@@ -153,6 +155,7 @@ func GenRandomBTCDelegation(
  BtcPk: delBTCPK,
  Pop: pop,
  FpBtcPkList: fpBTCPKs,
+ StakingTime: uint32(stakingTime),
  StartHeight: startHeight,
  EndHeight: endHeight,
  TotalSat: totalSat,

diff --git a/x/btcstaking/keeper/msg_server.go b/x/btcstaking/keeper/msg_server.go
@@ -200,6 +200,7 @@ func (ms msgServer) CreateBTCDelegation(goCtx context.Context, req *types.MsgCre
  BtcPk: parsedMsg.StakerPK.BIP340PubKey,
  Pop: parsedMsg.ParsedPop,
  FpBtcPkList: parsedMsg.FinalityProviderKeys.PublicKeysBbnFormat,
+ StakingTime: uint32(parsedMsg.StakingTime),
  StartHeight: startHeight,
  EndHeight: endHeight,
  TotalSat: uint64(parsedMsg.StakingValue),

diff --git a/x/btcstaking/keeper/msg_server_test.go b/x/btcstaking/keeper/msg_server_test.go
@@ -778,8 +778,8 @@ func createNDelegationsForFinalityProvider(
  covenantPks,
  quorum,
  slashingPkScript,
- 0,
- 0+math.MaxUint16,
+ 1,
+ 1+(math.MaxUint16-1),
  uint64(stakingValue),
  slashingRate,
  math.MaxUint16,