flag_kv_test.go

package command

import (
	"flag"
	"io/ioutil"
	"reflect"
	"testing"
)

func TestFlagKV_impl(t *testing.T) {
	var _ flag.Value = new(FlagKV)
}

func TestFlagKV(t *testing.T) {
	cases := []struct {
		Input  string
		Output map[string]string
		Error  bool
	}{
		{
			"key=value",
			map[string]string{"key": "value"},
			false,
		},

		{
			"key=",
			map[string]string{"key": ""},
			false,
		},

		{
			"key=foo=bar",
			map[string]string{"key": "foo=bar"},
			false,
		},

		{
			"map.key=foo",
			map[string]string{"map.key": "foo"},
			false,
		},

		{
			"key",
			nil,
			true,
		},
	}

	for _, tc := range cases {
		f := new(FlagKV)
		err := f.Set(tc.Input)
		if err != nil != tc.Error {
			t.Fatalf("bad error. Input: %#v", tc.Input)
		}

		actual := map[string]string(*f)
		if !reflect.DeepEqual(actual, tc.Output) {
			t.Fatalf("bad: %#v", actual)
		}
	}
}

func TestFlagKVFile_impl(t *testing.T) {
	var _ flag.Value = new(FlagKVFile)
}

func TestFlagKVFile(t *testing.T) {
	inputLibucl := `
foo = "bar"
`

	inputJson := `{
		"foo": "bar"}`

	cases := []struct {
		Input  string
		Output map[string]string
		Error  bool
	}{
		{
			inputLibucl,
			map[string]string{"foo": "bar"},
			false,
		},

		{
			inputJson,
			map[string]string{"foo": "bar"},
			false,
		},

		{
			`map.key = "foo"`,
			map[string]string{"map.key": "foo"},
			false,
		},
	}

	path := testTempFile(t)

	for _, tc := range cases {
		if err := ioutil.WriteFile(path, []byte(tc.Input), 0644); err != nil {
			t.Fatalf("err: %s", err)
		}

		f := new(FlagKVFile)
		err := f.Set(path)
		if err != nil != tc.Error {
			t.Fatalf("bad error. Input: %#v, err: %s", tc.Input, err)
		}

		actual := map[string]string(*f)
		if !reflect.DeepEqual(actual, tc.Output) {
			t.Fatalf("bad: %#v", actual)
		}
	}
}