1package http_test 2 3import ( 4 "context" 5 "net/http/httptest" 6 "testing" 7 8 httptransport "github.com/go-kit/kit/transport/http" 9) 10 11func TestSetHeader(t *testing.T) { 12 const ( 13 key = "X-Foo" 14 val = "12345" 15 ) 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) 20 } 21} 22 23func 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) 29 } 30} 31