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