You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
140 lines
3.8 KiB
140 lines
3.8 KiB
// Import this in all *_test.go files to initialize ~/.tendermint_test.
|
|
|
|
package tendermint_test
|
|
|
|
import (
|
|
"github.com/naoina/toml"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
. "github.com/tendermint/tendermint/common"
|
|
cfg "github.com/tendermint/tendermint/config"
|
|
)
|
|
|
|
func init() {
|
|
// Creates ~/.tendermint_test/*
|
|
config := GetConfig("")
|
|
cfg.ApplyConfig(config)
|
|
}
|
|
|
|
func getTMRoot(rootDir string) string {
|
|
if rootDir == "" {
|
|
rootDir = os.Getenv("HOME") + "/.tendermint_test"
|
|
}
|
|
return rootDir
|
|
}
|
|
|
|
func initTMRoot(rootDir string) {
|
|
rootDir = getTMRoot(rootDir)
|
|
EnsureDir(rootDir)
|
|
|
|
configFilePath := path.Join(rootDir, "config.toml")
|
|
genesisFilePath := path.Join(rootDir, "genesis.json")
|
|
privValFilePath := path.Join(rootDir, "priv_validator.json")
|
|
|
|
// Write default config file if missing.
|
|
if !FileExists(configFilePath) {
|
|
// Ask user for moniker
|
|
moniker := cfg.Prompt("Type hostname: ", "anonymous")
|
|
MustWriteFile(configFilePath, []byte(defaultConfig(moniker)))
|
|
}
|
|
if !FileExists(genesisFilePath) {
|
|
MustWriteFile(genesisFilePath, []byte(defaultGenesis))
|
|
}
|
|
if !FileExists(privValFilePath) {
|
|
MustWriteFile(privValFilePath, []byte(privValFilePath))
|
|
}
|
|
}
|
|
|
|
func GetConfig(rootDir string) cfg.Config {
|
|
rootDir = getTMRoot(rootDir)
|
|
initTMRoot(rootDir)
|
|
|
|
var mapConfig = cfg.MapConfig(make(map[string]interface{}))
|
|
configFilePath := path.Join(rootDir, "config.toml")
|
|
configFileBytes := MustReadFile(configFilePath)
|
|
err := toml.Unmarshal(configFileBytes, mapConfig)
|
|
if err != nil {
|
|
Exit(Fmt("Could not read config: %v", err))
|
|
}
|
|
|
|
// Set defaults or panic
|
|
if !mapConfig.IsSet("network") {
|
|
Exit("Must set 'network'")
|
|
}
|
|
if mapConfig.IsSet("version") {
|
|
Exit("Cannot set 'version' via config.toml")
|
|
}
|
|
// mapConfig.SetDefault("network", "tendermint_testnet0")
|
|
mapConfig.SetDefault("version", "0.2.1")
|
|
mapConfig.SetDefault("genesis_file", rootDir+"/genesis.json")
|
|
mapConfig.SetDefault("moniker", "anonymous")
|
|
mapConfig.SetDefault("node_laddr", "0.0.0.0:36656")
|
|
mapConfig.SetDefault("fast_sync", true)
|
|
mapConfig.SetDefault("addrbook_file", rootDir+"/addrbook.json")
|
|
mapConfig.SetDefault("priv_validator_file", rootDir+"/priv_validator.json")
|
|
mapConfig.SetDefault("db_backend", "memdb")
|
|
mapConfig.SetDefault("db_dir", rootDir+"/data")
|
|
mapConfig.SetDefault("log_level", "debug")
|
|
mapConfig.SetDefault("rpc_laddr", "0.0.0.0:36657")
|
|
return mapConfig
|
|
}
|
|
|
|
func ensureDefault(mapConfig cfg.MapConfig, key string, value interface{}) {
|
|
if !mapConfig.IsSet(key) {
|
|
mapConfig[key] = value
|
|
}
|
|
}
|
|
|
|
var defaultConfigTmpl = `# This is a TOML config file.
|
|
# For more information, see https://github.com/toml-lang/toml
|
|
|
|
network = "tendermint_test"
|
|
moniker = "__MONIKER__"
|
|
node_laddr = "0.0.0.0:36656"
|
|
seeds = ""
|
|
fast_sync = true
|
|
db_backend = "memdb"
|
|
log_level = "debug"
|
|
rpc_laddr = "0.0.0.0:36657"
|
|
`
|
|
|
|
func defaultConfig(moniker string) (defaultConfig string) {
|
|
defaultConfig = strings.Replace(defaultConfigTmpl, "__MONIKER__", moniker, -1)
|
|
return
|
|
}
|
|
|
|
var defaultGenesis = `{
|
|
"accounts": [
|
|
{
|
|
"address": "1D7A91CB32F758A02EBB9BE1FB6F8DEE56F90D42",
|
|
"amount": 200000000
|
|
},
|
|
{
|
|
"address": "AC89A6DDF4C309A89A2C4078CE409A5A7B282270",
|
|
"amount": 200000000
|
|
}
|
|
],
|
|
"validators": [
|
|
{
|
|
"pub_key": [1, "06FBAC4E285285D1D91FCBC7E91C780ADA11516F67462340B3980CE2B94940E8"],
|
|
"amount": 1000000,
|
|
"unbond_to": [
|
|
{
|
|
"address": "1D7A91CB32F758A02EBB9BE1FB6F8DEE56F90D42",
|
|
"amount": 100000
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}`
|
|
|
|
var defaultPrivValidator = `{
|
|
"address": "1D7A91CB32F758A02EBB9BE1FB6F8DEE56F90D42",
|
|
"pub_key": [1,"06FBAC4E285285D1D91FCBC7E91C780ADA11516F67462340B3980CE2B94940E8"],
|
|
"priv_key": [1,"C453604BD6480D5538B4C6FD2E3E314B5BCE518D75ADE4DA3DA85AB8ADFD819606FBAC4E285285D1D91FCBC7E91C780ADA11516F67462340B3980CE2B94940E8"],
|
|
"last_height":0,
|
|
"last_round":0,
|
|
"last_step":0
|
|
}`
|