| 8 | ) |
| 9 | |
| 10 | func TestContext(t *testing.T) { |
| 11 | wantHead := "bar" |
| 12 | wantRespHead := "baz" |
| 13 | f := func(ctx context.Context) { |
| 14 | if g := Request(ctx).Header.Get("Test-Key"); g != wantHead { |
| 15 | t.Errorf("header = %q want %q", g, wantHead) |
| 16 | } |
| 17 | ResponseWriter(ctx).Header().Set("Test-Resp-Key", wantRespHead) |
| 18 | } |
| 19 | |
| 20 | h, err := Handler(f, nil) |
| 21 | if err != nil { |
| 22 | t.Fatalf("err = %v", err) |
| 23 | } |
| 24 | |
| 25 | resp := httptest.NewRecorder() |
| 26 | req, _ := http.NewRequest("GET", "/", nil) |
| 27 | req.Header.Set("Test-Key", wantHead) |
| 28 | h.ServeHTTP(resp, req) |
| 29 | if g := resp.Header().Get("Test-Resp-Key"); g != wantRespHead { |
| 30 | t.Errorf("header = %q want %q", g, wantRespHead) |
| 31 | } |
| 32 | } |