Add genesis logic

proto/osmosis/tokenfactory/v1beta1/genesis.proto

@@ -29,4 +29,5 @@ message GenesisDenom {
     (gogoproto.moretags) = "yaml:\"authority_metadata\"",
     (gogoproto.nullable) = false
   ];
+  bool whitelisted = 3 [(gogoproto.moretags) = "yaml:\"whitelisted\""];
 }

x/tokenfactory/keeper/before_send_test.go

@@ -83,7 +83,7 @@ func (suite *KeeperTestSuite) TestTrackBeforeSendWasm() {
 			queryResp, err = suite.GetNeutronZoneApp(suite.ChainA).WasmKeeper.QuerySmart(suite.ChainA.GetContext(), cosmwasmAddress, []byte(`{"total_supply_at":{}}`))
 			suite.Require().NoError(err)
 
-			//Whitelisted contract has now been called
+			// Whitelisted contract has now been called
 			suite.Require().Equal("\"100\"", string(queryResp))
 		})
 	}

x/tokenfactory/keeper/genesis.go

@@ -32,6 +32,8 @@ func (k Keeper) InitGenesis(ctx sdk.Context, genState types.GenesisState) {
 		if err != nil {
 			panic(err)
 		}
+
+		k.setIsWhitelistedDenom(ctx, genDenom.GetDenom(), genDenom.GetWhitelisted())
 	}
 }
 
@@ -47,10 +49,12 @@ func (k Keeper) ExportGenesis(ctx sdk.Context) *types.GenesisState {
 		if err != nil {
 			panic(err)
 		}
+		whitelisted := k.getIsWhitelistedDenom(ctx, denom)
 
 		genDenoms = append(genDenoms, types.GenesisDenom{
 			Denom:             denom,
 			AuthorityMetadata: authorityMetadata,
+			Whitelisted:       whitelisted,
 		})
 	}
 

x/tokenfactory/keeper/genesis_test.go

@@ -26,6 +26,7 @@ func (suite *KeeperTestSuite) TestGenesis() {
 				AuthorityMetadata: types.DenomAuthorityMetadata{
 					Admin: "neutron1m9l358xunhhwds0568za49mzhvuxx9ux8xafx2",
 				},
+				Whitelisted: true,
 			},
 		},
 	}
@@ -45,4 +46,5 @@ func (suite *KeeperTestSuite) TestGenesis() {
 	exportedGenesis := app.TokenFactoryKeeper.ExportGenesis(context)
 	suite.Require().NotNil(exportedGenesis)
 	suite.Require().Equal(genesisState, *exportedGenesis)
+
 }

x/tokenfactory/keeper/whitelist.go

@@ -2,6 +2,7 @@ package keeper
 
 import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
+
 	"github.com/neutron-org/neutron/v4/x/tokenfactory/types"
 )
 
@@ -10,7 +11,7 @@ var isWhitelistedBz = []byte{1}
 func (k Keeper) setIsWhitelistedDenom(ctx sdk.Context, denom string, isWhitelisted bool) {
 	store := k.GetDenomPrefixStore(ctx, denom)
 	if isWhitelisted {
-		store.Set([]byte(types.WhitelistPrefixKey), []byte{1})
+		store.Set([]byte(types.WhitelistPrefixKey), isWhitelistedBz)
 	} else {
 		store.Delete([]byte(types.WhitelistPrefixKey))
 	}