fix(test): allow default genesis creation in testutil

axone-protocol · Jan 17, 2024 · bb35f38 · bb35f38
1 parent dca3eb2
commit bb35f38
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/app/genesis.go b/app/genesis.go
@@ -2,6 +2,12 @@ package app
 
 import (
 	"encoding/json"
+	"testing"
+
+	"cosmossdk.io/log"
+
+	dbm "github.com/cosmos/cosmos-db"
+	simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
 
 	"github.com/cosmos/cosmos-sdk/codec"
 )
@@ -15,7 +21,8 @@ import (
 // object provided to it during init.
 type GenesisState map[string]json.RawMessage
 
-// NewDefaultGenesisState generates the default state for the application.
-func NewDefaultGenesisState(cdc codec.JSONCodec) GenesisState {
-	return ModuleBasics.DefaultGenesis(cdc)
+// NewDefaultGenesisState generates the default state for the application, for testing purpose.
+func NewDefaultGenesisState(t testing.TB, cdc codec.JSONCodec) GenesisState {
+	tempApp := New(log.NewNopLogger(), dbm.NewMemDB(), nil, true, simtestutil.NewAppOptionsWithFlagHome(t.TempDir()))
+	return tempApp.BasicModuleManager.DefaultGenesis(cdc)
 }
diff --git a/testutil/network/network.go b/testutil/network/network.go
@@ -73,7 +73,7 @@ func DefaultConfig(t *testing.T) network.Config {
 				baseapp.SetChainID(chainID),
 			)
 		},
-		GenesisState:    app.NewDefaultGenesisState(encoding.Codec),
+		GenesisState:    app.NewDefaultGenesisState(t, encoding.Codec),
 		TimeoutCommit:   2 * time.Second,
 		ChainID:         chainID,
 		NumValidators:   1,