From 972462206496827ccea0d435493e9d0608790fb4 Mon Sep 17 00:00:00 2001
From: zakir <80246097+zakir-code@users.noreply.github.com>
Date: Thu, 8 Aug 2024 15:17:36 +0800
Subject: [PATCH] test: update DelegateMultiple params for crosschain module
 (#636)

---
 tests/crosschain_suite.go | 11 +++++++----
 tests/crosschain_test.go  | 17 ++++++++++++++++-
 2 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/tests/crosschain_suite.go b/tests/crosschain_suite.go
index a986db32..f7b71b40 100644
--- a/tests/crosschain_suite.go
+++ b/tests/crosschain_suite.go
@@ -732,13 +732,16 @@ func (suite *CrosschainTestSuite) PendingExecuteClaim() []crosschaintypes.Extern
 	return externalClaims
 }
 
-func (suite *CrosschainTestSuite) UpdateParamsEnablePending() (*sdk.TxResponse, uint64) {
-	suite.params.EnableBridgeCallPending = true
-	suite.params.EnableSendToExternalPending = true
+func (suite *CrosschainTestSuite) UpdateParams(opts ...func(params *crosschaintypes.Params)) (*sdk.TxResponse, uint64) {
+	params := suite.QueryParams()
+	for _, opt := range opts {
+		opt(&params)
+	}
+	suite.params = params
 	msg := &crosschaintypes.MsgUpdateParams{
 		ChainName: suite.chainName,
 		Authority: authtypes.NewModuleAddress(govtypes.ModuleName).String(),
-		Params:    suite.params,
+		Params:    params,
 	}
 	return suite.BroadcastProposalTx2([]sdk.Msg{msg}, "UpdateParams", "UpdateParams")
 }
diff --git a/tests/crosschain_test.go b/tests/crosschain_test.go
index 24e61094..aaf8f8d2 100644
--- a/tests/crosschain_test.go
+++ b/tests/crosschain_test.go
@@ -101,8 +101,13 @@ func (suite *IntegrationTest) CrossChainTest() {
 				uint64(pundixMetadata.DenomUnits[1].Exponent), pundixAddress, "")
 		}
 
-		chain.UpdateParamsEnablePending()
+		chain.UpdateParams(func(params *crosschaintypes.Params) {
+			params.EnableBridgeCallPending = true
+			params.EnableSendToExternalPending = true
+		})
 	}
+
+	// suite.UpdateParamsTest()
 }
 
 func (suite *IntegrationTest) OriginalCrossChainTest() {
@@ -414,3 +419,13 @@ func (suite *IntegrationTest) BridgeCallTest() {
 	ethChain.BridgeCallConfirm(ethBridgeCallNonce, true)
 	suite.True(suite.QueryBalances(erc20ModuleAddr).IsZero())
 }
+
+func (suite *IntegrationTest) UpdateParamsTest() {
+	for _, chain := range suite.crosschain {
+		chain.UpdateParams(func(params *crosschaintypes.Params) {
+			params.DelegateMultiple = 100
+		})
+		params := chain.QueryParams()
+		suite.Require().Equal(params.DelegateMultiple, int64(100))
+	}
+}