From 621e8f7621fae172d1cc1bbc9e5326a76acdd874 Mon Sep 17 00:00:00 2001
From: puneetmahajan <puneet@persistence.one>
Date: Fri, 12 Jan 2024 00:29:11 +0530
Subject: [PATCH] more tests for msg_server_test.go

---
 x/ratesync/keeper/msg_server_test.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/x/ratesync/keeper/msg_server_test.go b/x/ratesync/keeper/msg_server_test.go
index d6168b115..efa45a817 100644
--- a/x/ratesync/keeper/msg_server_test.go
+++ b/x/ratesync/keeper/msg_server_test.go
@@ -59,9 +59,9 @@ func (suite *IntegrationTestSuite) TestChainMsgServerUpdate() {
 
 	hc3.Features.LiquidStakeIBC.Instantiation = types.InstantiationState_INSTANTIATION_INITIATED
 	hc3.Features.LiquidStakeIBC.CodeID = 1
-
-	hc3.Features.LiquidStake.Instantiation = types.InstantiationState_INSTANTIATION_INITIATED
-	hc3.Features.LiquidStake.CodeID = 1
+	hc4, _ := k.GetHostChain(ctx, hc3.ID)
+	hc4.Features.LiquidStake.Instantiation = types.InstantiationState_INSTANTIATION_INITIATED
+	hc4.Features.LiquidStake.CodeID = 1
 	tests := []struct {
 		desc    string
 		request *types.MsgUpdateHostChain
@@ -93,6 +93,12 @@ func (suite *IntegrationTestSuite) TestChainMsgServerUpdate() {
 				HostChain: hc3,
 			},
 		},
+		{
+			desc: "Update feature2",
+			request: &types.MsgUpdateHostChain{Authority: GovAddress.String(),
+				HostChain: hc4,
+			},
+		},
 	}
 	for _, tc := range tests {
 		suite.T().Run(tc.desc, func(t *testing.T) {