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.
172 lines
4.5 KiB
172 lines
4.5 KiB
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
cfg "github.com/tendermint/tendermint/config"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
tmos "github.com/tendermint/tendermint/libs/os"
|
|
)
|
|
|
|
var (
|
|
defaultRoot = os.ExpandEnv("$HOME/.some/test/dir")
|
|
)
|
|
|
|
// clearConfig clears env vars, the given root dir, and resets viper.
|
|
func clearConfig(dir string) {
|
|
if err := os.Unsetenv("TMHOME"); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := os.Unsetenv("TM_HOME"); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := os.RemoveAll(dir); err != nil {
|
|
panic(err)
|
|
}
|
|
viper.Reset()
|
|
config = cfg.DefaultConfig()
|
|
}
|
|
|
|
// prepare new rootCmd
|
|
func testRootCmd() *cobra.Command {
|
|
rootCmd := &cobra.Command{
|
|
Use: RootCmd.Use,
|
|
PersistentPreRunE: RootCmd.PersistentPreRunE,
|
|
Run: func(cmd *cobra.Command, args []string) {},
|
|
}
|
|
registerFlagsRootCmd(rootCmd)
|
|
var l string
|
|
rootCmd.PersistentFlags().String("log", l, "Log")
|
|
return rootCmd
|
|
}
|
|
|
|
func testSetup(rootDir string, args []string, env map[string]string) error {
|
|
clearConfig(defaultRoot)
|
|
|
|
rootCmd := testRootCmd()
|
|
cmd := cli.PrepareBaseCmd(rootCmd, "TM", defaultRoot)
|
|
|
|
// run with the args and env
|
|
args = append([]string{rootCmd.Use}, args...)
|
|
return cli.RunWithArgs(cmd, args, env)
|
|
}
|
|
|
|
func TestRootHome(t *testing.T) {
|
|
newRoot := filepath.Join(defaultRoot, "something-else")
|
|
cases := []struct {
|
|
args []string
|
|
env map[string]string
|
|
root string
|
|
}{
|
|
{nil, nil, defaultRoot},
|
|
{[]string{"--home", newRoot}, nil, newRoot},
|
|
{nil, map[string]string{"TMHOME": newRoot}, newRoot},
|
|
}
|
|
|
|
for i, tc := range cases {
|
|
idxString := strconv.Itoa(i)
|
|
|
|
err := testSetup(defaultRoot, tc.args, tc.env)
|
|
require.Nil(t, err, idxString)
|
|
|
|
assert.Equal(t, tc.root, config.RootDir, idxString)
|
|
assert.Equal(t, tc.root, config.P2P.RootDir, idxString)
|
|
assert.Equal(t, tc.root, config.Consensus.RootDir, idxString)
|
|
assert.Equal(t, tc.root, config.Mempool.RootDir, idxString)
|
|
}
|
|
}
|
|
|
|
func TestRootFlagsEnv(t *testing.T) {
|
|
|
|
// defaults
|
|
defaults := cfg.DefaultConfig()
|
|
defaultLogLvl := defaults.LogLevel
|
|
|
|
cases := []struct {
|
|
args []string
|
|
env map[string]string
|
|
logLevel string
|
|
}{
|
|
{[]string{"--log", "debug"}, nil, defaultLogLvl}, // wrong flag
|
|
{[]string{"--log-level", "debug"}, nil, "debug"}, // right flag
|
|
{nil, map[string]string{"TM_LOW": "debug"}, defaultLogLvl}, // wrong env flag
|
|
{nil, map[string]string{"MT_LOG_LEVEL": "debug"}, defaultLogLvl}, // wrong env prefix
|
|
{nil, map[string]string{"TM_LOG_LEVEL": "debug"}, "debug"}, // right env
|
|
}
|
|
|
|
for i, tc := range cases {
|
|
idxString := strconv.Itoa(i)
|
|
|
|
err := testSetup(defaultRoot, tc.args, tc.env)
|
|
require.Nil(t, err, idxString)
|
|
|
|
assert.Equal(t, tc.logLevel, config.LogLevel, idxString)
|
|
}
|
|
}
|
|
|
|
func TestRootConfig(t *testing.T) {
|
|
|
|
// write non-default config
|
|
nonDefaultLogLvl := "abc:debug"
|
|
cvals := map[string]string{
|
|
"log-level": nonDefaultLogLvl,
|
|
}
|
|
|
|
cases := []struct {
|
|
args []string
|
|
env map[string]string
|
|
|
|
logLvl string
|
|
}{
|
|
{nil, nil, nonDefaultLogLvl}, // should load config
|
|
{[]string{"--log-level=abc:info"}, nil, "abc:info"}, // flag over rides
|
|
{nil, map[string]string{"TM_LOG_LEVEL": "abc:info"}, "abc:info"}, // env over rides
|
|
}
|
|
|
|
for i, tc := range cases {
|
|
idxString := strconv.Itoa(i)
|
|
clearConfig(defaultRoot)
|
|
|
|
// XXX: path must match cfg.defaultConfigPath
|
|
configFilePath := filepath.Join(defaultRoot, "config")
|
|
err := tmos.EnsureDir(configFilePath, 0700)
|
|
require.Nil(t, err)
|
|
|
|
// write the non-defaults to a different path
|
|
// TODO: support writing sub configs so we can test that too
|
|
err = WriteConfigVals(configFilePath, cvals)
|
|
require.Nil(t, err)
|
|
|
|
rootCmd := testRootCmd()
|
|
cmd := cli.PrepareBaseCmd(rootCmd, "TM", defaultRoot)
|
|
|
|
// run with the args and env
|
|
tc.args = append([]string{rootCmd.Use}, tc.args...)
|
|
err = cli.RunWithArgs(cmd, tc.args, tc.env)
|
|
require.Nil(t, err, idxString)
|
|
|
|
assert.Equal(t, tc.logLvl, config.LogLevel, idxString)
|
|
}
|
|
}
|
|
|
|
// WriteConfigVals writes a toml file with the given values.
|
|
// It returns an error if writing was impossible.
|
|
func WriteConfigVals(dir string, vals map[string]string) error {
|
|
data := ""
|
|
for k, v := range vals {
|
|
data += fmt.Sprintf("%s = \"%s\"\n", k, v)
|
|
}
|
|
cfile := filepath.Join(dir, "config.toml")
|
|
return ioutil.WriteFile(cfile, []byte(data), 0600)
|
|
}
|