Fix API test

api/client_test.go

@@ -0,0 +1,72 @@
+package api
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConfigSetAddress(t *testing.T) {
+	type test struct {
+		name    string
+		input   string
+		address string
+		err     string
+	}
+
+	tests := []test{
+		{
+			"bare",
+			"http://127.0.0.1:9200",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"bare with version",
+			"http://127.0.0.1:9200/v1",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"bare with version and trailing slash",
+			"http://127.0.0.1:9200/v1/",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"valid top level scope",
+			"http://127.0.0.1:9200/v1/scopes",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"valid scope",
+			"http://127.0.0.1:9200/v1/scopes/scopeid",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"longer path project",
+			"http://127.0.0.1:9200/v1/auth-methods",
+			"http://127.0.0.1:9200",
+			"",
+		},
+		{
+			"valid project",
+			"http://127.0.0.1:9200/my-install",
+			"http://127.0.0.1:9200/my-install",
+			"",
+		},
+	}
+
+	for _, v := range tests {
+		t.Run(v.name, func(t *testing.T) {
+			var c Config
+			err := c.setAddr(v.input)
+			if err != nil {
+				assert.Equal(t, v.err, err.Error())
+			}
+			assert.Equal(t, v.address, c.Addr)
+		})
+	}
+}

api/go.mod

@@ -11,6 +11,7 @@ require (
 	github.com/hashicorp/go-kms-wrapping v0.5.16
 	github.com/hashicorp/go-retryablehttp v0.6.7
 	github.com/hashicorp/go-rootcerts v1.0.2
+	github.com/stretchr/testify v1.6.1
 	golang.org/x/time v0.0.0-20200630173020-3af7569d3a1e
 	google.golang.org/grpc v1.32.0
 )