forked from rcrowley/go-tigertonic
-
Notifications
You must be signed in to change notification settings - Fork 0
/
post_processor_test.go
32 lines (30 loc) · 773 Bytes
/
post_processor_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package tigertonic
import (
"io/ioutil"
"net/http"
"net/url"
"testing"
)
func TestPostProcessed(t *testing.T) {
w := &testResponseWriter{}
r, _ := http.NewRequest("GET", "http://example.com/foo", nil)
called := false
PostProcessed(Marshaled(func(*url.URL, http.Header) (int, http.Header, *testResponse, error) {
return http.StatusOK, http.Header{"X-Foo": []string{"bar"}}, &testResponse{"bar"}, nil
}), func(_ *http.Request, r *http.Response) {
called = true
if http.StatusOK != r.StatusCode {
t.Fatal(r.StatusCode)
}
if "bar" != r.Header.Get("X-Foo") {
t.Fatal(r.Header)
}
body, _ := ioutil.ReadAll(r.Body)
if "{\"foo\":\"bar\"}\n" != string(body) {
t.Fatal(w.Body.String())
}
}).ServeHTTP(w, r)
if !called {
t.Fatal(called)
}
}