chore: fix stakers flow

harness/app.go

@@ -74,14 +74,19 @@ func startRemoteHarness(cmdCtx context.Context, cfg config.Config) error {
 		return fmt.Errorf("error collecting the finality providers %w", err)
 	}
 
+	paramsResp, err := bbnClient.BTCCheckpointParams()
+	if err != nil {
+		return fmt.Errorf("failed to get staking params: %w", err)
+	}
+
 	var stakers []*BTCStaker
 	for i := 0; i < cfg.TotalStakers; i++ {
 		stakerSender, err := NewSenderWithBabylonClient(cmdCtx, fmt.Sprintf("staker-%d", i), cfg.BabylonRPC, cfg.BabylonGRPC)
 		if err != nil {
 			return fmt.Errorf("failed to create staker sender: %w", err)
 		}
 
-		staker := NewBTCStaker(btcClient.client, stakerSender, fpPks, nil, nil)
+		staker := NewBTCStaker(btcClient.client, stakerSender, fpPks, &paramsResp.Params, nil, nil)
 		stakers = append(stakers, staker)
 	}
 
@@ -94,6 +99,8 @@ func startRemoteHarness(cmdCtx context.Context, cfg config.Config) error {
 		return fmt.Errorf("failed to start stakers: %w", err)
 	}
 
+	<-cmdCtx.Done()
+
 	return nil
 }
 
@@ -159,6 +166,11 @@ func startHarness(cmdCtx context.Context, cfg config.Config) error {
 		return err
 	}
 
+	paramsResp, err := cpSender.BTCCheckpointParams()
+	if err != nil {
+		return fmt.Errorf("failed to get staking params: %w", err)
+	}
+
 	var stakers []*BTCStaker
 	for i := 0; i < numStakers; i++ {
 		stakerSender, err := NewSenderWithBabylonClient(ctx, fmt.Sprintf("staker-%d", i), tm.Config.Babylon1.RPCAddr, tm.Config.Babylon1.GRPCAddr)
@@ -167,7 +179,7 @@ func startHarness(cmdCtx context.Context, cfg config.Config) error {
 		}
 
 		rndFpChunk := fpMgr.getRandomChunk(3)
-		stakers = append(stakers, NewBTCStaker(tm.TestRpcClient, stakerSender, rndFpChunk, tm.fundingRequests, tm.fundingResponse))
+		stakers = append(stakers, NewBTCStaker(tm.TestRpcClient, stakerSender, rndFpChunk, &paramsResp.Params, tm.fundingRequests, tm.fundingResponse))
 	}
 
 	// periodically check if we need to fund the staker

harness/btcclient.go

@@ -51,11 +51,14 @@ func (c *BTCClient) checkBalance() error {
 func (c *BTCClient) Setup(cfg config.Config) error {
 	c.config = c.ConvertToBTCConfig(cfg)
 	rpcClient, err := rpcclient.New(&rpcclient.ConnConfig{
-		Host:         rpcHostURL(c.config.Endpoint, c.config.WalletName),
-		User:         c.config.Username,
-		Pass:         c.config.Password,
-		DisableTLS:   true,
-		HTTPPostMode: true,
+		Host:                 rpcHostURL(c.config.Endpoint, c.config.WalletName),
+		Endpoint:             c.config.Endpoint,
+		User:                 c.config.Username,
+		Pass:                 c.config.Password,
+		DisableTLS:           true,
+		DisableConnectOnNew:  true,
+		DisableAutoReconnect: false,
+		HTTPPostMode:         true,
 	}, nil)
 	if err != nil {
 		return err

harness/btcstaker.go

@@ -35,6 +35,7 @@ type BTCStaker struct {
 	client          *SenderWithBabylonClient
 	fpPK            *btcec.PublicKey
 	fpPKChunk       []*btcec.PublicKey
+	btcCkpParams    *btckpttypes.Params
 	fundingRequest  chan sdk.AccAddress
 	fundingResponse chan sdk.AccAddress
 }
@@ -43,13 +44,15 @@ func NewBTCStaker(
 	btcClient *rpcclient.Client,
 	client *SenderWithBabylonClient,
 	finalityProvidersPublicKey []*btcec.PublicKey,
+	btcCkpParams *btckpttypes.Params,
 	fundingRequest chan sdk.AccAddress,
 	fundingResponse chan sdk.AccAddress,
 ) *BTCStaker {
 	return &BTCStaker{
 		btcClient:       btcClient,
 		client:          client,
 		fpPKChunk:       finalityProvidersPublicKey,
+		btcCkpParams:    btcCkpParams,
 		fundingRequest:  fundingRequest,
 		fundingResponse: fundingResponse,
 	}
@@ -429,7 +432,7 @@ func (s *BTCStaker) buildAndSendStakingTransaction(
 	}
 
 	// TODO: hardcoded two in tests
-	inclusionProof := s.waitForTransactionConfirmation(ctx, hash, 2)
+	inclusionProof := s.waitForTransactionConfirmation(ctx, hash, s.btcCkpParams.BtcConfirmationDepth)
 
 	if inclusionProof == nil {
 		// we are quiting

harness/manager.go

@@ -306,7 +306,8 @@ func AtomicFundSignSendStakingTx(rpcClient *rpcclient.Client, stakingOutput *wir
 	feeRate := float64(0.00002)
 	pos := 1
 
-	err := rpcClient.WalletPassphrase("pass", 60)
+	//Add password in here for remote/dynamically
+	err := rpcClient.WalletPassphrase("pass", 50)
 	if err != nil {
 		return nil, nil, err
 	}