Switch from glide to govendor
This commit is contained in:
149
vendor/github.com/go-kit/kit/transport/http/client_test.go
generated
vendored
149
vendor/github.com/go-kit/kit/transport/http/client_test.go
generated
vendored
@@ -1,149 +0,0 @@
|
||||
package http_test
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"net/url"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
|
||||
httptransport "github.com/go-kit/kit/transport/http"
|
||||
)
|
||||
|
||||
type TestResponse struct {
|
||||
Body io.ReadCloser
|
||||
String string
|
||||
}
|
||||
|
||||
func TestHTTPClient(t *testing.T) {
|
||||
var (
|
||||
testbody = "testbody"
|
||||
encode = func(context.Context, *http.Request, interface{}) error { return nil }
|
||||
decode = func(_ context.Context, r *http.Response) (interface{}, error) {
|
||||
buffer := make([]byte, len(testbody))
|
||||
r.Body.Read(buffer)
|
||||
return TestResponse{r.Body, string(buffer)}, nil
|
||||
}
|
||||
headers = make(chan string, 1)
|
||||
headerKey = "X-Foo"
|
||||
headerVal = "abcde"
|
||||
afterHeaderKey = "X-The-Dude"
|
||||
afterHeaderVal = "Abides"
|
||||
afterVal = ""
|
||||
afterFunc = func(ctx context.Context, r *http.Response) context.Context {
|
||||
afterVal = r.Header.Get(afterHeaderKey)
|
||||
return ctx
|
||||
}
|
||||
)
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
headers <- r.Header.Get(headerKey)
|
||||
w.Header().Set(afterHeaderKey, afterHeaderVal)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(testbody))
|
||||
}))
|
||||
|
||||
client := httptransport.NewClient(
|
||||
"GET",
|
||||
mustParse(server.URL),
|
||||
encode,
|
||||
decode,
|
||||
httptransport.ClientBefore(httptransport.SetRequestHeader(headerKey, headerVal)),
|
||||
httptransport.ClientAfter(afterFunc),
|
||||
)
|
||||
|
||||
res, err := client.Endpoint()(context.Background(), struct{}{})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
var have string
|
||||
select {
|
||||
case have = <-headers:
|
||||
case <-time.After(time.Millisecond):
|
||||
t.Fatalf("timeout waiting for %s", headerKey)
|
||||
}
|
||||
// Check that Request Header was successfully received
|
||||
if want := headerVal; want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
|
||||
// Check that Response header set from server was received in SetClientAfter
|
||||
if want, have := afterVal, afterHeaderVal; want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
|
||||
// Check that the response was successfully decoded
|
||||
response, ok := res.(TestResponse)
|
||||
if !ok {
|
||||
t.Fatal("response should be TestResponse")
|
||||
}
|
||||
if want, have := testbody, response.String; want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
|
||||
// Check that response body was closed
|
||||
b := make([]byte, 1)
|
||||
_, err = response.Body.Read(b)
|
||||
if err == nil {
|
||||
t.Fatal("wanted error, got none")
|
||||
}
|
||||
if doNotWant, have := io.EOF, err; doNotWant == have {
|
||||
t.Errorf("do not want %q, have %q", doNotWant, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTTPClientBufferedStream(t *testing.T) {
|
||||
var (
|
||||
testbody = "testbody"
|
||||
encode = func(context.Context, *http.Request, interface{}) error { return nil }
|
||||
decode = func(_ context.Context, r *http.Response) (interface{}, error) {
|
||||
return TestResponse{r.Body, ""}, nil
|
||||
}
|
||||
)
|
||||
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte(testbody))
|
||||
}))
|
||||
|
||||
client := httptransport.NewClient(
|
||||
"GET",
|
||||
mustParse(server.URL),
|
||||
encode,
|
||||
decode,
|
||||
httptransport.BufferedStream(true),
|
||||
)
|
||||
|
||||
res, err := client.Endpoint()(context.Background(), struct{}{})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Check that the response was successfully decoded
|
||||
response, ok := res.(TestResponse)
|
||||
if !ok {
|
||||
t.Fatal("response should be TestResponse")
|
||||
}
|
||||
|
||||
// Check that response body was NOT closed
|
||||
b := make([]byte, len(testbody))
|
||||
_, err = response.Body.Read(b)
|
||||
if want, have := io.EOF, err; have != want {
|
||||
t.Fatalf("want %q, have %q", want, have)
|
||||
}
|
||||
if want, have := testbody, string(b); want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func mustParse(s string) *url.URL {
|
||||
u, err := url.Parse(s)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return u
|
||||
}
|
56
vendor/github.com/go-kit/kit/transport/http/err_test.go
generated
vendored
56
vendor/github.com/go-kit/kit/transport/http/err_test.go
generated
vendored
@@ -1,56 +0,0 @@
|
||||
package http_test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
|
||||
httptransport "github.com/go-kit/kit/transport/http"
|
||||
)
|
||||
|
||||
func TestClientEndpointEncodeError(t *testing.T) {
|
||||
var (
|
||||
sampleErr = errors.New("Oh no, an error")
|
||||
enc = func(context.Context, *http.Request, interface{}) error { return sampleErr }
|
||||
dec = func(context.Context, *http.Response) (interface{}, error) { return nil, nil }
|
||||
)
|
||||
|
||||
u := &url.URL{
|
||||
Scheme: "https",
|
||||
Host: "localhost",
|
||||
Path: "/does/not/matter",
|
||||
}
|
||||
|
||||
c := httptransport.NewClient(
|
||||
"GET",
|
||||
u,
|
||||
enc,
|
||||
dec,
|
||||
)
|
||||
|
||||
_, err := c.Endpoint()(context.Background(), nil)
|
||||
if err == nil {
|
||||
t.Fatal("err == nil")
|
||||
}
|
||||
|
||||
e, ok := err.(httptransport.Error)
|
||||
if !ok {
|
||||
t.Fatal("err is not of type github.com/go-kit/kit/transport/http.Error")
|
||||
}
|
||||
|
||||
if want, have := sampleErr, e.Err; want != have {
|
||||
t.Fatalf("want %v, have %v", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func ExampleErrorOutput() {
|
||||
sampleErr := errors.New("oh no, an error")
|
||||
err := httptransport.Error{Domain: httptransport.DomainDo, Err: sampleErr}
|
||||
fmt.Println(err)
|
||||
// Output:
|
||||
// Do: oh no, an error
|
||||
}
|
31
vendor/github.com/go-kit/kit/transport/http/request_response_funcs_test.go
generated
vendored
31
vendor/github.com/go-kit/kit/transport/http/request_response_funcs_test.go
generated
vendored
@@ -1,31 +0,0 @@
|
||||
package http_test
|
||||
|
||||
import (
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
|
||||
httptransport "github.com/go-kit/kit/transport/http"
|
||||
)
|
||||
|
||||
func TestSetHeader(t *testing.T) {
|
||||
const (
|
||||
key = "X-Foo"
|
||||
val = "12345"
|
||||
)
|
||||
r := httptest.NewRecorder()
|
||||
httptransport.SetResponseHeader(key, val)(context.Background(), r)
|
||||
if want, have := val, r.Header().Get(key); want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetContentType(t *testing.T) {
|
||||
const contentType = "application/json"
|
||||
r := httptest.NewRecorder()
|
||||
httptransport.SetContentType(contentType)(context.Background(), r)
|
||||
if want, have := contentType, r.Header().Get("Content-Type"); want != have {
|
||||
t.Errorf("want %q, have %q", want, have)
|
||||
}
|
||||
}
|
120
vendor/github.com/go-kit/kit/transport/http/server_test.go
generated
vendored
120
vendor/github.com/go-kit/kit/transport/http/server_test.go
generated
vendored
@@ -1,120 +0,0 @@
|
||||
package http_test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
|
||||
httptransport "github.com/go-kit/kit/transport/http"
|
||||
)
|
||||
|
||||
func TestServerBadDecode(t *testing.T) {
|
||||
handler := httptransport.NewServer(
|
||||
context.Background(),
|
||||
func(context.Context, interface{}) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, *http.Request) (interface{}, error) { return struct{}{}, errors.New("dang") },
|
||||
func(context.Context, http.ResponseWriter, interface{}) error { return nil },
|
||||
)
|
||||
server := httptest.NewServer(handler)
|
||||
defer server.Close()
|
||||
resp, _ := http.Get(server.URL)
|
||||
if want, have := http.StatusBadRequest, resp.StatusCode; want != have {
|
||||
t.Errorf("want %d, have %d", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerBadEndpoint(t *testing.T) {
|
||||
handler := httptransport.NewServer(
|
||||
context.Background(),
|
||||
func(context.Context, interface{}) (interface{}, error) { return struct{}{}, errors.New("dang") },
|
||||
func(context.Context, *http.Request) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, http.ResponseWriter, interface{}) error { return nil },
|
||||
)
|
||||
server := httptest.NewServer(handler)
|
||||
defer server.Close()
|
||||
resp, _ := http.Get(server.URL)
|
||||
if want, have := http.StatusServiceUnavailable, resp.StatusCode; want != have {
|
||||
t.Errorf("want %d, have %d", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerBadEncode(t *testing.T) {
|
||||
handler := httptransport.NewServer(
|
||||
context.Background(),
|
||||
func(context.Context, interface{}) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, *http.Request) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, http.ResponseWriter, interface{}) error { return errors.New("dang") },
|
||||
)
|
||||
server := httptest.NewServer(handler)
|
||||
defer server.Close()
|
||||
resp, _ := http.Get(server.URL)
|
||||
if want, have := http.StatusInternalServerError, resp.StatusCode; want != have {
|
||||
t.Errorf("want %d, have %d", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerErrorEncoder(t *testing.T) {
|
||||
errTeapot := errors.New("teapot")
|
||||
code := func(err error) int {
|
||||
if e, ok := err.(httptransport.Error); ok && e.Err == errTeapot {
|
||||
return http.StatusTeapot
|
||||
}
|
||||
return http.StatusInternalServerError
|
||||
}
|
||||
handler := httptransport.NewServer(
|
||||
context.Background(),
|
||||
func(context.Context, interface{}) (interface{}, error) { return struct{}{}, errTeapot },
|
||||
func(context.Context, *http.Request) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, http.ResponseWriter, interface{}) error { return nil },
|
||||
httptransport.ServerErrorEncoder(func(_ context.Context, err error, w http.ResponseWriter) { w.WriteHeader(code(err)) }),
|
||||
)
|
||||
server := httptest.NewServer(handler)
|
||||
defer server.Close()
|
||||
resp, _ := http.Get(server.URL)
|
||||
if want, have := http.StatusTeapot, resp.StatusCode; want != have {
|
||||
t.Errorf("want %d, have %d", want, have)
|
||||
}
|
||||
}
|
||||
|
||||
func TestServerHappyPath(t *testing.T) {
|
||||
_, step, response := testServer(t)
|
||||
step()
|
||||
resp := <-response
|
||||
defer resp.Body.Close()
|
||||
buf, _ := ioutil.ReadAll(resp.Body)
|
||||
if want, have := http.StatusOK, resp.StatusCode; want != have {
|
||||
t.Errorf("want %d, have %d (%s)", want, have, buf)
|
||||
}
|
||||
}
|
||||
|
||||
func testServer(t *testing.T) (cancel, step func(), resp <-chan *http.Response) {
|
||||
var (
|
||||
ctx, cancelfn = context.WithCancel(context.Background())
|
||||
stepch = make(chan bool)
|
||||
endpoint = func(context.Context, interface{}) (interface{}, error) { <-stepch; return struct{}{}, nil }
|
||||
response = make(chan *http.Response)
|
||||
handler = httptransport.NewServer(
|
||||
ctx,
|
||||
endpoint,
|
||||
func(context.Context, *http.Request) (interface{}, error) { return struct{}{}, nil },
|
||||
func(context.Context, http.ResponseWriter, interface{}) error { return nil },
|
||||
httptransport.ServerBefore(func(ctx context.Context, r *http.Request) context.Context { return ctx }),
|
||||
httptransport.ServerAfter(func(ctx context.Context, w http.ResponseWriter) context.Context { return ctx }),
|
||||
)
|
||||
)
|
||||
go func() {
|
||||
server := httptest.NewServer(handler)
|
||||
defer server.Close()
|
||||
resp, err := http.Get(server.URL)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
return
|
||||
}
|
||||
response <- resp
|
||||
}()
|
||||
return cancelfn, func() { stepch <- true }, response
|
||||
}
|
Reference in New Issue
Block a user