1 package client
2
3 import (
4 "net/http"
5 "net/http/httptest"
6 "testing"
7
8 "github.com/stretchr/testify/assert"
9 )
10
11 func TestRoundTrip(t *testing.T) {
12 hdrs := map[string]string{
13 "hello": "world",
14 "foo": "bar",
15 }
16 tp := &Transport{
17 T: http.DefaultTransport,
18 Headers: hdrs,
19 }
20 cl := &http.Client{
21 Transport: tp,
22 }
23 svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
24 assert.Equal(t, hdrs["hello"], r.Header.Get("hello"))
25 assert.Equal(t, hdrs["foo"], r.Header.Get("foo"))
26 w.WriteHeader(http.StatusOK)
27 }))
28 defer svr.Close()
29 _, err := cl.Head(svr.URL)
30 assert.NoError(t, err)
31 }
32
View as plain text