A few extra fixes and tests.
This commit is contained in:
@@ -0,0 +1,54 @@
|
|||||||
|
package kong_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/alecthomas/kong"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMultipleConfigLoading(t *testing.T) {
|
||||||
|
var cli struct {
|
||||||
|
Flag string `json:"flag,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
cli.Flag = "first"
|
||||||
|
first, cleanFirst := makeConfig(t, &cli)
|
||||||
|
defer cleanFirst()
|
||||||
|
|
||||||
|
cli.Flag = ""
|
||||||
|
second, cleanSecond := makeConfig(t, &cli)
|
||||||
|
defer cleanSecond()
|
||||||
|
|
||||||
|
p := mustNew(t, &cli, kong.Configuration(kong.JSON, first, second))
|
||||||
|
_, err := p.Parse(nil)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, "first", cli.Flag)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestConfigValidation(t *testing.T) {
|
||||||
|
var cli struct {
|
||||||
|
Flag string `json:"flag,omitempty" enum:"valid"`
|
||||||
|
}
|
||||||
|
|
||||||
|
cli.Flag = "invalid"
|
||||||
|
conf, cleanConf := makeConfig(t, &cli)
|
||||||
|
defer cleanConf()
|
||||||
|
|
||||||
|
p := mustNew(t, &cli, kong.Configuration(kong.JSON, conf))
|
||||||
|
_, err := p.Parse(nil)
|
||||||
|
require.Error(t, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeConfig(t *testing.T, config interface{}) (path string, cleanup func()) {
|
||||||
|
t.Helper()
|
||||||
|
w, err := ioutil.TempFile("", "")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer w.Close()
|
||||||
|
err = json.NewEncoder(w).Encode(config)
|
||||||
|
require.NoError(t, err)
|
||||||
|
return w.Name(), func() { os.Remove(w.Name()) }
|
||||||
|
}
|
||||||
@@ -326,7 +326,6 @@ func (k *Kong) FatalIfErrorf(err error, args ...interface{}) {
|
|||||||
if err, ok := err.(*ParseError); ok && k.usageOnError {
|
if err, ok := err.(*ParseError); ok && k.usageOnError {
|
||||||
fmt.Fprintln(k.Stdout)
|
fmt.Fprintln(k.Stdout)
|
||||||
options := k.helpOptions
|
options := k.helpOptions
|
||||||
options.Summary = true
|
|
||||||
_ = k.help(options, err.Context)
|
_ = k.help(options, err.Context)
|
||||||
}
|
}
|
||||||
k.Exit(1)
|
k.Exit(1)
|
||||||
|
|||||||
@@ -707,3 +707,12 @@ func TestDefaultValueIsHyphen(t *testing.T) {
|
|||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
require.Equal(t, "-", cli.Flag)
|
require.Equal(t, "-", cli.Flag)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDefaultEnumValidated(t *testing.T) {
|
||||||
|
var cli struct {
|
||||||
|
Flag string `default:"invalid" enum:"valid"`
|
||||||
|
}
|
||||||
|
p := mustNew(t, &cli)
|
||||||
|
_, err := p.Parse(nil)
|
||||||
|
require.Error(t, err)
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,9 +1,6 @@
|
|||||||
package kong
|
package kong
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
@@ -21,35 +18,6 @@ func TestOptions(t *testing.T) {
|
|||||||
require.Nil(t, p.Exit)
|
require.Nil(t, p.Exit)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestConfigLoading(t *testing.T) {
|
|
||||||
first, err := ioutil.TempFile("", "")
|
|
||||||
require.NoError(t, err)
|
|
||||||
defer first.Close()
|
|
||||||
defer os.Remove(first.Name())
|
|
||||||
second, err := ioutil.TempFile("", "")
|
|
||||||
require.NoError(t, err)
|
|
||||||
defer second.Close()
|
|
||||||
defer os.Remove(second.Name())
|
|
||||||
|
|
||||||
var cli struct {
|
|
||||||
Flag string `json:"flag,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
cli.Flag = "first"
|
|
||||||
err = json.NewEncoder(first).Encode(&cli)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
cli.Flag = ""
|
|
||||||
err = json.NewEncoder(second).Encode(&cli)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
p, err := New(&cli, Configuration(JSON, first.Name(), second.Name()))
|
|
||||||
require.NoError(t, err)
|
|
||||||
_, err = p.Parse(nil)
|
|
||||||
require.NoError(t, err)
|
|
||||||
require.Equal(t, "first", cli.Flag)
|
|
||||||
}
|
|
||||||
|
|
||||||
type impl string
|
type impl string
|
||||||
|
|
||||||
func (impl) Method() {}
|
func (impl) Method() {}
|
||||||
|
|||||||
@@ -74,6 +74,24 @@ func TestEnvarsTag(t *testing.T) {
|
|||||||
require.Equal(t, []int{5, 2, 9}, cli.Slice)
|
require.Equal(t, []int{5, 2, 9}, cli.Slice)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestEnvarsWithDefault(t *testing.T) {
|
||||||
|
var cli struct {
|
||||||
|
Flag string `env:"KONG_FLAG" default:"default"`
|
||||||
|
}
|
||||||
|
parser, restoreEnv := newEnvParser(t, &cli, envMap{})
|
||||||
|
defer restoreEnv()
|
||||||
|
|
||||||
|
_, err := parser.Parse(nil)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, "default", cli.Flag)
|
||||||
|
|
||||||
|
parser, restoreEnv = newEnvParser(t, &cli, envMap{"KONG_FLAG": "moo"})
|
||||||
|
defer restoreEnv()
|
||||||
|
_, err = parser.Parse(nil)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, "moo", cli.Flag)
|
||||||
|
}
|
||||||
|
|
||||||
func TestJSONBasic(t *testing.T) {
|
func TestJSONBasic(t *testing.T) {
|
||||||
var cli struct {
|
var cli struct {
|
||||||
String string
|
String string
|
||||||
|
|||||||
Reference in New Issue
Block a user