mirror of
https://github.com/gravitational/teleport.git
synced 2026-09-24 16:17:11 +08:00
Update following packages: * Replace Sirupsen/log with sirupsen/log everywhere * Update etcd client to 3.2.4 * Update docker/term to moby/term * Update kr/pty to v1.0.0 release * Update K8s client to 2.0
35 lines
691 B
Go
35 lines
691 B
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type UtilsSuite struct{}
|
|
|
|
var _ = Suite(&UtilsSuite{})
|
|
|
|
func (s *UtilsSuite) TestDefaultHandlerErrors(c *C) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
h := w.(http.Hijacker)
|
|
conn, _, _ := h.Hijack()
|
|
conn.Close()
|
|
}))
|
|
defer srv.Close()
|
|
|
|
request, err := http.NewRequest("GET", srv.URL, strings.NewReader(""))
|
|
c.Assert(err, IsNil)
|
|
|
|
_, err = http.DefaultTransport.RoundTrip(request)
|
|
|
|
w := NewBufferWriter(NopWriteCloser(&bytes.Buffer{}))
|
|
|
|
DefaultHandler.ServeHTTP(w, nil, err)
|
|
|
|
c.Assert(w.Code, Equals, http.StatusBadGateway)
|
|
}
|