provider: fix init module_test.go

cosmos · Oct 13, 2023 · 0735fca · 0735fca
1 parent 2e7f946
commit 0735fca
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/x/ccv/provider/module_test.go b/x/ccv/provider/module_test.go
@@ -159,6 +159,21 @@ func TestInitGenesis(t *testing.T) {
 			continue // Nothing else to verify
 		}
 
+		appModule.InitGenesis(ctx, cdc, jsonBytes)
+
+		numStatesCounted := 0
+		for _, state := range tc.consumerStates {
+			numStatesCounted += 1
+			channelID, found := providerKeeper.GetChainToChannel(ctx, state.ChainId)
+			require.True(t, found)
+			require.Equal(t, state.ChannelId, channelID)
+
+			chainID, found := providerKeeper.GetChannelToChain(ctx, state.ChannelId)
+			require.True(t, found)
+			require.Equal(t, state.ChainId, chainID)
+		}
+		require.Equal(t, len(tc.consumerStates), numStatesCounted)
+
 		// Expect slash meter to be initialized to it's allowance value
 		// (replenish fraction * mocked value defined above)
 		slashMeter := providerKeeper.GetSlashMeter(ctx)