fix test

dydxprotocol · Dec 11, 2024 · d8048f4 · d8048f4
1 parent 39d5ce7
commit d8048f4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/protocol/app/upgrades/v8.0/migrate_accountplus_test.go b/protocol/app/upgrades/v8.0/migrate_accountplus_test.go
@@ -3,6 +3,7 @@ package v_8_0_test
 import (
 	"testing"
 
+	"github.com/cometbft/cometbft/types"
 	v_8_0 "github.com/dydxprotocol/v4-chain/protocol/app/upgrades/v8.0"
 
 	"cosmossdk.io/store/prefix"
@@ -24,7 +25,16 @@ func TestMigrateAccountplusAccountState(t *testing.T) {
 }
 
 func (s *UpgradeTestSuite) SetupTest() {
-	s.tApp = testapp.NewTestAppBuilder(s.T()).Build()
+	s.tApp = testapp.NewTestAppBuilder(s.T()).WithGenesisDocFn(func() (genesis types.GenesisDoc) {
+		genesis = testapp.DefaultGenesis()
+		testapp.UpdateGenesisDocWithAppStateForModule(
+			&genesis,
+			func(genesisState *accountplustypes.GenesisState) {
+				genesisState.Params.IsSmartAccountActive = false
+			},
+		)
+		return genesis
+	}).Build()
 	s.Ctx = s.tApp.InitChain()
 }
 

diff --git a/protocol/x/accountplus/types/genesis.go b/protocol/x/accountplus/types/genesis.go
@@ -2,7 +2,11 @@ package types
 
 // DefaultGenesis returns the default genesis state
 func DefaultGenesis() *GenesisState {
-	return &GenesisState{}
+	return &GenesisState{
+		Params: Params{
+			IsSmartAccountActive: true,
+		},
+	}
 }
 
 // Validate performs basic genesis state validation returning an error upon any