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)
		})
	}
}