8 httptransport "github.com/go-kit/kit/transport/http"
11 func TestSetHeader(t *testing.T) {
16 r := httptest.NewRecorder()
17 httptransport.SetResponseHeader(key, val)(context.Background(), r)
18 if want, have := val, r.Header().Get(key); want != have {
19 t.Errorf("want %q, have %q", want, have)
23 func TestSetContentType(t *testing.T) {
24 const contentType = "application/json"
25 r := httptest.NewRecorder()
26 httptransport.SetContentType(contentType)(context.Background(), r)
27 if want, have := contentType, r.Header().Get("Content-Type"); want != have {
28 t.Errorf("want %q, have %q", want, have)