test: fixgenesis test

x/ccv/provider/keeper/genesis_test.go

@@ -35,6 +35,8 @@ func TestInitAndExportGenesis(t *testing.T) {
 	consumerTmPubKey := consumerCryptoId.TMProtoCryptoPublicKey()
 	consumerConsAddr := consumerCryptoId.ConsumerConsAddress()
 
+	initTimeoutTime := uint64(time.Now().UTC().UnixNano()) + 10
+	vscSendTime := time.Now().Add(-30 * time.Minute)
 	// create genesis struct
 	provGenesis := providertypes.NewGenesisState(vscID,
 		[]providertypes.ValsetUpdateIdToHeight{{ValsetUpdateId: vscID, Height: initHeight}},
@@ -97,6 +99,19 @@ func TestInitAndExportGenesis(t *testing.T) {
 				ConsumerAddrs: &providertypes.AddressList{Addresses: [][]byte{consumerConsAddr.ToSdkConsAddr()}},
 			},
 		},
+		[]providertypes.InitTimeoutTimestamp{
+			{
+				ChainId:   cChainIDs[0],
+				Timestamp: initTimeoutTime,
+			},
+		},
+		[]providertypes.VscSendTimestamp{
+			{
+				ChainId:   cChainIDs[0],
+				VscId:     vscID,
+				Timestamp: vscSendTime,
+			},
+		},
 	)
 
 	// Instantiate in-mem provider keeper with mocks
@@ -163,6 +178,14 @@ func TestInitAndExportGenesis(t *testing.T) {
 
 	// check the exported genesis
 	require.Equal(t, provGenesis, pk.ExportGenesis(ctx))
+
+	initTimeoutTimestamp, found := pk.GetInitTimeoutTimestamp(ctx, cChainIDs[0])
+	require.Equal(t, initTimeoutTimestamp, initTimeoutTime)
+	require.True(t, found)
+
+	vscSendTimestamp, found := pk.GetVscSendTimestamp(ctx, cChainIDs[0], vscID)
+	require.Equal(t, vscSendTimestamp, vscSendTime)
+	require.True(t, found)
 }
 
 func assertConsumerChainStates(t *testing.T, ctx sdk.Context, pk keeper.Keeper, consumerStates ...providertypes.ConsumerState) {

x/ccv/provider/module_test.go

@@ -109,6 +109,8 @@ func TestInitGenesis(t *testing.T) {
 			nil,
 			nil,
 			nil,
+			nil,
+			nil,
 		)
 
 		cdc := keeperParams.Cdc