feat: initial commit

ctl/transport_test.go

@@ -0,0 +1,48 @@
+package ctl
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_headerTransport_RoundTrip(t1 *testing.T) {
+	type fields struct {
+		apiKey string
+	}
+	tests := []struct {
+		name    string
+		fields  fields
+		handler func(t *testing.T) http.Handler
+		want    *http.Response
+		wantErr bool
+	}{
+		{
+			name: "apiKey is set as header",
+			fields: fields{
+				apiKey: "abc123",
+			},
+			handler: func(t *testing.T) http.Handler {
+				return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+					assert.Equal(t, "abc123", req.Header.Get("x-api-key"))
+					_, err := w.Write([]byte("OK"))
+					assert.NoError(t, err)
+				})
+			},
+			want:    nil,
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t1.Run(tt.name, func(t1 *testing.T) {
+			srv := httptest.NewServer(tt.handler(t1))
+			defer srv.Close()
+			t := NewTransport(http.DefaultTransport, tt.fields.apiKey)
+			c := http.Client{Transport: t}
+			_, err := c.Get(srv.URL)
+			assert.NoError(t1, err)
+		})
+	}
+}