From dcc5afeac6f49884473c898d19ed004c705cd334 Mon Sep 17 00:00:00 2001
From: Julien Robert <julien@rbrt.fr>
Date: Mon, 26 Jun 2023 21:29:56 +0200
Subject: [PATCH] feat: add configurator config options (#16672)

---
 testutil/configurator/configurator.go | 170 ++++++++++++++------------
 1 file changed, 95 insertions(+), 75 deletions(-)

diff --git a/testutil/configurator/configurator.go b/testutil/configurator/configurator.go
index 48b2ac2dff..b63bd78c8f 100644
--- a/testutil/configurator/configurator.go
+++ b/testutil/configurator/configurator.go
@@ -25,80 +25,103 @@ import (
 	"cosmossdk.io/depinject"
 )
 
-var beginBlockOrder = []string{
-	"upgrade",
-	"mint",
-	"distribution",
-	"slashing",
-	"evidence",
-	"staking",
-	"auth",
-	"bank",
-	"gov",
-	"crisis",
-	"genutil",
-	"authz",
-	"feegrant",
-	"nft",
-	"group",
-	"params",
-	"consensus",
-	"vesting",
-	"circuit",
-}
-
-var endBlockersOrder = []string{
-	"crisis",
-	"gov",
-	"staking",
-	"auth",
-	"bank",
-	"distribution",
-	"slashing",
-	"mint",
-	"genutil",
-	"evidence",
-	"authz",
-	"feegrant",
-	"nft",
-	"group",
-	"params",
-	"consensus",
-	"upgrade",
-	"vesting",
-	"circuit",
-}
-
-var initGenesisOrder = []string{
-	"auth",
-	"bank",
-	"distribution",
-	"staking",
-	"slashing",
-	"gov",
-	"mint",
-	"crisis",
-	"genutil",
-	"evidence",
-	"authz",
-	"feegrant",
-	"nft",
-	"group",
-	"params",
-	"consensus",
-	"upgrade",
-	"vesting",
-	"circuit",
-}
-
 // Config should never need to be instantiated manually and is solely used for ModuleOption.
 type Config struct {
-	ModuleConfigs  map[string]*appv1alpha1.ModuleConfig
-	setInitGenesis bool
+	ModuleConfigs      map[string]*appv1alpha1.ModuleConfig
+	BeginBlockersOrder []string
+	EndBlockersOrder   []string
+	InitGenesisOrder   []string
+	setInitGenesis     bool
+}
+
+var defaultConfig = &Config{
+	ModuleConfigs: make(map[string]*appv1alpha1.ModuleConfig),
+	BeginBlockersOrder: []string{
+		"upgrade",
+		"mint",
+		"distribution",
+		"slashing",
+		"evidence",
+		"staking",
+		"auth",
+		"bank",
+		"gov",
+		"crisis",
+		"genutil",
+		"authz",
+		"feegrant",
+		"nft",
+		"group",
+		"params",
+		"consensus",
+		"vesting",
+		"circuit",
+	},
+	EndBlockersOrder: []string{
+		"crisis",
+		"gov",
+		"staking",
+		"auth",
+		"bank",
+		"distribution",
+		"slashing",
+		"mint",
+		"genutil",
+		"evidence",
+		"authz",
+		"feegrant",
+		"nft",
+		"group",
+		"params",
+		"consensus",
+		"upgrade",
+		"vesting",
+		"circuit",
+	},
+	InitGenesisOrder: []string{
+		"auth",
+		"bank",
+		"distribution",
+		"staking",
+		"slashing",
+		"gov",
+		"mint",
+		"crisis",
+		"genutil",
+		"evidence",
+		"authz",
+		"feegrant",
+		"nft",
+		"group",
+		"params",
+		"consensus",
+		"upgrade",
+		"vesting",
+		"circuit",
+	},
+	setInitGenesis: true,
 }
 
 type ModuleOption func(config *Config)
 
+func WithCustomBeginBlockersOrder(beginBlockOrder ...string) ModuleOption {
+	return func(config *Config) {
+		config.BeginBlockersOrder = beginBlockOrder
+	}
+}
+
+func WithCustomEndBlockersOrder(endBlockersOrder ...string) ModuleOption {
+	return func(config *Config) {
+		config.EndBlockersOrder = endBlockersOrder
+	}
+}
+
+func WithCustomInitGenesisOrder(initGenesisOrder ...string) ModuleOption {
+	return func(config *Config) {
+		config.InitGenesisOrder = initGenesisOrder
+	}
+}
+
 func BankModule() ModuleOption {
 	return func(config *Config) {
 		config.ModuleConfigs["bank"] = &appv1alpha1.ModuleConfig{
@@ -285,10 +308,7 @@ func OmitInitGenesis() ModuleOption {
 }
 
 func NewAppConfig(opts ...ModuleOption) depinject.Config {
-	cfg := &Config{
-		ModuleConfigs:  make(map[string]*appv1alpha1.ModuleConfig),
-		setInitGenesis: true,
-	}
+	cfg := defaultConfig
 	for _, opt := range opts {
 		opt(cfg)
 	}
@@ -298,19 +318,19 @@ func NewAppConfig(opts ...ModuleOption) depinject.Config {
 	initGenesis := make([]string, 0)
 	overrides := make([]*runtimev1alpha1.StoreKeyConfig, 0)
 
-	for _, s := range beginBlockOrder {
+	for _, s := range cfg.BeginBlockersOrder {
 		if _, ok := cfg.ModuleConfigs[s]; ok {
 			beginBlockers = append(beginBlockers, s)
 		}
 	}
 
-	for _, s := range endBlockersOrder {
+	for _, s := range cfg.EndBlockersOrder {
 		if _, ok := cfg.ModuleConfigs[s]; ok {
 			endBlockers = append(endBlockers, s)
 		}
 	}
 
-	for _, s := range initGenesisOrder {
+	for _, s := range cfg.InitGenesisOrder {
 		if _, ok := cfg.ModuleConfigs[s]; ok {
 			initGenesis = append(initGenesis, s)
 		}