You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.0 KiB
78 lines
2.0 KiB
package debug
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
// mux is the HTTP request multiplexer used with the test server.
|
|
mux *http.ServeMux
|
|
|
|
// server is a test HTTP server used to provide mock API responses.
|
|
server *httptest.Server
|
|
)
|
|
|
|
// setup sets up a test HTTP server along with a cos.Client that is
|
|
// configured to talk to that test server. Tests should register handlers on
|
|
// mux which provide mock responses for the API method being tested.
|
|
func setup() {
|
|
// test server
|
|
mux = http.NewServeMux()
|
|
server = httptest.NewServer(mux)
|
|
}
|
|
|
|
// teardown closes the test HTTP server.
|
|
func teardown() {
|
|
server.Close()
|
|
}
|
|
|
|
func TestDebugRequestTransport(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Add("X-Test-Response", "2333")
|
|
w.WriteHeader(http.StatusBadGateway)
|
|
w.Write([]byte("test response body"))
|
|
})
|
|
|
|
w := bytes.NewBufferString("")
|
|
client := http.Client{}
|
|
|
|
client.Transport = &DebugRequestTransport{
|
|
RequestHeader: true,
|
|
RequestBody: true,
|
|
ResponseHeader: true,
|
|
ResponseBody: true,
|
|
Writer: w,
|
|
}
|
|
|
|
body := bytes.NewReader([]byte("test_request body"))
|
|
req, _ := http.NewRequest("GET", server.URL, body)
|
|
req.Header.Add("X-Test-Debug", "123")
|
|
client.Do(req)
|
|
|
|
b := make([]byte, 800)
|
|
w.Read(b)
|
|
info := string(b)
|
|
if !strings.Contains(info, "GET / HTTP/1.1\r\n") ||
|
|
!strings.Contains(info, "X-Test-Debug: 123\r\n") {
|
|
t.Errorf("DebugRequestTransport debug info %#v don't contains request header", info)
|
|
}
|
|
if !strings.Contains(info, "\r\n\r\ntest_request body") {
|
|
t.Errorf("DebugRequestTransport debug info %#v don't contains request body", info)
|
|
}
|
|
|
|
if !strings.Contains(info, "HTTP/1.1 502 Bad Gateway\r\n") ||
|
|
!strings.Contains(info, "X-Test-Response: 2333\r\n") {
|
|
t.Errorf("DebugRequestTransport debug info %#v don't contains response header", info)
|
|
}
|
|
|
|
if !strings.Contains(info, "\r\n\r\ntest response body") {
|
|
t.Errorf("DebugRequestTransport debug info %#v don't contains response body", info)
|
|
}
|
|
}
|