package discover

import (
	"errors"
	"reflect"
	"testing"
)

func TestConfigParse(t *testing.T) {
	tests := []struct {
		s   string
		c   Config
		err error
	}{
		// happy flows
		{``, nil, nil},
		{`key=a`, Config{"key": "a"}, nil},
		{`key=a key2=b`, Config{"key": "a", "key2": "b"}, nil},
		{`key=a+b key2=c/d`, Config{"key": "a+b", "key2": "c/d"}, nil},
		{` key=a    key2=b `, Config{"key": "a", "key2": "b"}, nil},
		{` key = a   key2 = b `, Config{"key": "a", "key2": "b"}, nil},
		{`  "k e \\\" y" = "a \" b" key2=c`, Config{`k e \" y`: `a " b`, "key2": "c"}, nil},

		{`provider=aws foo`, nil, errors.New(`foo: missing '='`)},
		{`project_name=Test zone_pattern=us-(?west|east).+ tag_value="consul server" credentials_file=xxx`,
			Config{
				"project_name":     "Test",
				"zone_pattern":     "us-(?west|east).+",
				"tag_value":        "consul server",
				"credentials_file": "xxx",
			},
			nil,
		},

		// errors
		{`key`, nil, errors.New(`key: missing '='`)},
		{`key=`, nil, errors.New(`key: missing value`)},
		{`key="a`, nil, errors.New(`key: unbalanced quotes`)},
		{`key="\`, nil, errors.New(`key: unterminated escape sequence`)},
		{`key=a key=b`, nil, errors.New(`key: duplicate key`)},
		{`key key2`, nil, errors.New(`key: missing '='`)},
	}

	for _, tt := range tests {
		t.Run(tt.s, func(t *testing.T) {
			c, err := Parse(tt.s)
			if got, want := err, tt.err; !reflect.DeepEqual(got, want) {
				t.Fatalf("got error %v want %v", got, want)
			}
			if got, want := c, tt.c; !reflect.DeepEqual(got, want) {
				t.Fatalf("got config %#v want %#v", got, want)
			}
		})
	}
}

func TestConfigString(t *testing.T) {
	tests := []struct {
		in, out string
	}{
		{``, ``},
		{`   `, ``},
		{`b=c "a a"="b b"`, `"a a"="b b" b=c`},
		{`a=b provider=foo x=y`, `provider=foo a=b x=y`},
	}

	for _, tt := range tests {
		t.Run(tt.in, func(t *testing.T) {
			c, err := Parse(tt.in)
			if err != nil {
				t.Fatal("Parse failed: ", err)
			}
			if got, want := c.String(), tt.out; got != want {
				t.Fatalf("got %q want %q", got, want)
			}
		})
	}
}
