Codebase list golang-github-alecthomas-kong / fresh-releases/main config_test.go
fresh-releases/main

Tree @fresh-releases/main (Download .tar.gz)

config_test.go @fresh-releases/mainraw · history · blame

package kong_test

import (
	"encoding/json"
	"io/ioutil"
	"os"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/alecthomas/kong"
)

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" required:""`
	}

	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() // nolint: gosec
	err = json.NewEncoder(w).Encode(config)
	require.NoError(t, err)
	return w.Name(), func() { os.Remove(w.Name()) }
}