diff --git a/client/opentracing_test.go b/client/opentracing_test.go index d293c9d3..650a437f 100644 --- a/client/opentracing_test.go +++ b/client/opentracing_test.go @@ -29,6 +29,9 @@ func (r tres) Message() string { func (r tres) GetHeader(_ string) string { return "the header" } +func (r tres) GetHeaders(_ string) []string { + return []string{"the headers", "the headers2"} +} func (r tres) Body() io.ReadCloser { return ioutil.NopCloser(bytes.NewBufferString("the content")) } diff --git a/client/response.go b/client/response.go index bd238588..b297a12f 100644 --- a/client/response.go +++ b/client/response.go @@ -39,6 +39,10 @@ func (r response) GetHeader(name string) string { return r.resp.Header.Get(name) } +func (r response) GetHeaders(name string) []string { + return r.resp.Header.Values(name) +} + func (r response) Body() io.ReadCloser { return r.resp.Body } diff --git a/client/response_test.go b/client/response_test.go index 5dc5db83..5d4694ee 100644 --- a/client/response_test.go +++ b/client/response_test.go @@ -37,5 +37,6 @@ func TestResponse(t *testing.T) { assert.EqualValues(t, under.StatusCode, resp.Code()) assert.Equal(t, under.Status, resp.Message()) assert.Equal(t, "blah blah", resp.GetHeader("blah")) + assert.Equal(t, []string{"blah blah"}, resp.GetHeaders("blah")) assert.Equal(t, under.Body, resp.Body()) } diff --git a/client_response.go b/client_response.go index d97e0ac6..c2e82f96 100644 --- a/client_response.go +++ b/client_response.go @@ -27,6 +27,7 @@ type ClientResponse interface { Code() int Message() string GetHeader(string) string + GetHeaders(string) []string Body() io.ReadCloser } diff --git a/client_response_test.go b/client_response_test.go index 7422d349..ee13d9df 100644 --- a/client_response_test.go +++ b/client_response_test.go @@ -35,6 +35,9 @@ func (r response) Message() string { func (r response) GetHeader(_ string) string { return "the header" } +func (r response) GetHeaders(_ string) []string { + return []string{"the headers", "the headers2"} +} func (r response) Body() io.ReadCloser { return ioutil.NopCloser(bytes.NewBufferString("the content")) }