forked from celestiaorg/celestia-app
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
chore!: refactor testnode to expose appcreator (celestiaorg#1991)
## Overview This refactor is mainly targeting celestiaorg#1990, but doing so in a way that uses common patters to avoid continuing to blow up the api closes celestiaorg#1990 ## Checklist - [x] New and updated code has appropriate documentation - [x] New and updated code has new and/or updated testing - [x] Required CI checks are passing - [x] Visual proof for any user facing features like CLI or documentation updates - [x] Linked issues closed with keywords
- Loading branch information
1 parent
4725fd8
commit ce0be95
Showing
13 changed files
with
290 additions
and
248 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,157 @@ | ||
| package testnode | ||
|
|
||
| import ( | ||
| "time" | ||
|
|
||
| "github.com/celestiaorg/celestia-app/cmd/celestia-appd/cmd" | ||
| "github.com/celestiaorg/celestia-app/pkg/appconsts" | ||
| pruningtypes "github.com/cosmos/cosmos-sdk/pruning/types" | ||
| "github.com/cosmos/cosmos-sdk/server" | ||
| srvconfig "github.com/cosmos/cosmos-sdk/server/config" | ||
| srvtypes "github.com/cosmos/cosmos-sdk/server/types" | ||
| "github.com/tendermint/tendermint/config" | ||
| tmconfig "github.com/tendermint/tendermint/config" | ||
| tmrand "github.com/tendermint/tendermint/libs/rand" | ||
| tmproto "github.com/tendermint/tendermint/proto/tendermint/types" | ||
| "github.com/tendermint/tendermint/types" | ||
| ) | ||
|
|
||
| // Config is the configuration of a test node. | ||
| type Config struct { | ||
| // ChainID is the chain ID of the network. | ||
| ChainID string | ||
| // TmConfig is the Tendermint configuration used for the network. | ||
| TmConfig *tmconfig.Config | ||
| // AppConfig is the application configuration of the test node. | ||
| AppConfig *srvconfig.Config | ||
| // ConsensusParams are the consensus parameters of the test node. | ||
| ConsensusParams *tmproto.ConsensusParams | ||
| // AppOptions are the application options of the test node. | ||
| AppOptions *KVAppOptions | ||
| // GenesisOptions are the genesis options of the test node. | ||
| GenesisOptions []GenesisOption | ||
| // Accounts are the accounts of the test node. | ||
| Accounts []string | ||
| // AppCreator is used to create the application for the testnode. | ||
| AppCreator srvtypes.AppCreator | ||
| // SupressLogs | ||
| SupressLogs bool | ||
| } | ||
|
|
||
| // WithChainID sets the ChainID and returns the Config. | ||
| func (c *Config) WithChainID(s string) *Config { | ||
| c.ChainID = s | ||
| return c | ||
| } | ||
|
|
||
| // WithTendermintConfig sets the TmConfig and returns the *Config. | ||
| func (c *Config) WithTendermintConfig(conf *tmconfig.Config) *Config { | ||
| c.TmConfig = conf | ||
| return c | ||
| } | ||
|
|
||
| // WithAppConfig sets the AppConfig and returns the Config. | ||
| func (c *Config) WithAppConfig(conf *srvconfig.Config) *Config { | ||
| c.AppConfig = conf | ||
| return c | ||
| } | ||
|
|
||
| // WithConsensusParams sets the ConsensusParams and returns the Config. | ||
| func (c *Config) WithConsensusParams(params *tmproto.ConsensusParams) *Config { | ||
| c.ConsensusParams = params | ||
| return c | ||
| } | ||
|
|
||
| // WithAppOptions sets the AppOptions and returns the Config. | ||
| func (c *Config) WithAppOptions(opts *KVAppOptions) *Config { | ||
| c.AppOptions = opts | ||
| return c | ||
| } | ||
|
|
||
| // WithGenesisOptions sets the GenesisOptions and returns the Config. | ||
| func (c *Config) WithGenesisOptions(opts ...GenesisOption) *Config { | ||
| c.GenesisOptions = opts | ||
| return c | ||
| } | ||
|
|
||
| // WithAccounts sets the Accounts and returns the Config. | ||
| func (c *Config) WithAccounts(accs []string) *Config { | ||
| c.Accounts = accs | ||
| return c | ||
| } | ||
|
|
||
| // WithAppCreator sets the AppCreator and returns the Config. | ||
| func (c *Config) WithAppCreator(creator srvtypes.AppCreator) *Config { | ||
| c.AppCreator = creator | ||
| return c | ||
| } | ||
|
|
||
| // WithSupressLogs sets the SupressLogs and returns the Config. | ||
| func (c *Config) WithSupressLogs(sl bool) *Config { | ||
| c.SupressLogs = sl | ||
| return c | ||
| } | ||
|
|
||
| func DefaultConfig() *Config { | ||
| tmcfg := DefaultTendermintConfig() | ||
| tmcfg.Consensus.TargetHeightDuration = 1 * time.Millisecond | ||
| cfg := &Config{} | ||
| return cfg. | ||
| WithAccounts([]string{}). | ||
| WithChainID(tmrand.Str(6)). | ||
| WithTendermintConfig(DefaultTendermintConfig()). | ||
| WithAppConfig(DefaultAppConfig()). | ||
| WithConsensusParams(DefaultParams()). | ||
| WithAppOptions(DefaultAppOptions()). | ||
| WithGenesisOptions(). | ||
| WithAppCreator(cmd.NewAppServer). | ||
| WithSupressLogs(true) | ||
| } | ||
|
|
||
| type KVAppOptions struct { | ||
| options map[string]interface{} | ||
| } | ||
|
|
||
| // Get implements AppOptions | ||
| func (ao *KVAppOptions) Get(o string) interface{} { | ||
| return ao.options[o] | ||
| } | ||
|
|
||
| // Set adds an option to the KVAppOptions | ||
| func (ao *KVAppOptions) Set(o string, v interface{}) { | ||
| ao.options[o] = v | ||
| } | ||
|
|
||
| // DefaultAppOptions returns the default application options. | ||
| func DefaultAppOptions() *KVAppOptions { | ||
| opts := &KVAppOptions{options: make(map[string]interface{})} | ||
| opts.Set(server.FlagPruning, pruningtypes.PruningOptionNothing) | ||
| return opts | ||
| } | ||
|
|
||
| func DefaultParams() *tmproto.ConsensusParams { | ||
| cparams := types.DefaultConsensusParams() | ||
| cparams.Block.TimeIotaMs = 1 | ||
| cparams.Block.MaxBytes = appconsts.DefaultMaxBytes | ||
| return cparams | ||
| } | ||
|
|
||
| func DefaultTendermintConfig() *config.Config { | ||
| tmCfg := config.DefaultConfig() | ||
| // Reduce the target height duration so that blocks are produced faster | ||
| // during tests. | ||
| tmCfg.Consensus.TargetHeightDuration = 300 * time.Millisecond | ||
| tmCfg.Consensus.TimeoutPropose = 200 * time.Millisecond | ||
|
|
||
| // set the mempool's MaxTxBytes to allow the testnode to accept a | ||
| // transaction that fills the entire square. Any blob transaction larger | ||
| // than the square size will still fail no matter what. | ||
| tmCfg.Mempool.MaxTxBytes = appconsts.DefaultMaxBytes | ||
|
|
||
| // remove all barriers from the testnode being able to accept very large | ||
| // transactions and respond to very queries with large responses (~200MB was | ||
| // chosen only as an arbitrary large number). | ||
| tmCfg.RPC.MaxBodyBytes = 200_000_000 | ||
|
|
||
| return tmCfg | ||
| } |
Oops, something went wrong.