package mhttp import ( "bytes" "io" "io/ioutil" "net/http" "net/http/httptest" . "testing" "github.com/mediocregopher/mediocre-go-lib/mtest" "github.com/mediocregopher/mediocre-go-lib/mtest/massert" ) func TestMListenAndServe(t *T) { cmp := mtest.Component() srv := InstListeningServer(cmp, http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { io.Copy(rw, r.Body) })) mtest.Run(cmp, t, func() { body := bytes.NewBufferString("HELLO") resp, err := http.Post("http://"+srv.Addr, "text/plain", body) if err != nil { t.Fatal(err) } defer resp.Body.Close() respBody, err := ioutil.ReadAll(resp.Body) if err != nil { t.Fatal(err) } else if string(respBody) != "HELLO" { t.Fatalf("unexpected respBody: %q", respBody) } }) } func TestAddXForwardedFor(t *T) { assertXFF := func(prev []string, ipStr, expected string) massert.Assertion { r := httptest.NewRequest("GET", "/", nil) for i := range prev { r.Header.Add("X-Forwarded-For", prev[i]) } AddXForwardedFor(r, ipStr) var a massert.Assertion if expected == "" { a = massert.Length(r.Header["X-Forwarded-For"], 0) } else { a = massert.All( massert.Length(r.Header["X-Forwarded-For"], 1), massert.Equal(expected, r.Header["X-Forwarded-For"][0]), ) } return massert.Comment(a, "prev:%#v ipStr:%q", prev, ipStr) } massert.Require(t, assertXFF(nil, "invalid", ""), assertXFF(nil, "::1", ""), assertXFF([]string{"8.0.0.0"}, "invalid", "8.0.0.0"), assertXFF([]string{"8.0.0.0"}, "::1", "8.0.0.0"), assertXFF(nil, "8.0.0.0", "8.0.0.0"), assertXFF([]string{"8.0.0.0"}, "8.0.0.1", "8.0.0.0, 8.0.0.1"), assertXFF([]string{"8.0.0.0, 8.0.0.1"}, "8.0.0.2", "8.0.0.0, 8.0.0.1, 8.0.0.2"), assertXFF([]string{"8.0.0.0, 8.0.0.1", "8.0.0.2"}, "8.0.0.3", "8.0.0.0, 8.0.0.1, 8.0.0.2, 8.0.0.3"), ) }