mirror of
https://github.com/openfaas/faas.git
synced 2025-06-20 13:06:40 +00:00
Changes from pull request feedback
1. Use httptest.ResponseRecorder instead of custom implementation StringResponseWriter. 2. Remove verbose log line in infohandler Signed-off-by: Edward Wilde <ewilde@gmail.com>
This commit is contained in:
@ -1,10 +1,5 @@
|
||||
package types
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// GatewayInfo provides information about the gateway and it's connected components
|
||||
type GatewayInfo struct {
|
||||
Provider *ProviderInfo `json:"provider"`
|
||||
@ -24,35 +19,3 @@ type VersionInfo struct {
|
||||
SHA string `json:"sha"`
|
||||
Release string `json:"release"`
|
||||
}
|
||||
|
||||
// StringResponseWriter captures the handlers HTTP response in a buffer
|
||||
type StringResponseWriter struct {
|
||||
body *bytes.Buffer
|
||||
headerCode int
|
||||
header http.Header
|
||||
}
|
||||
|
||||
// NewStringResponseWriter create a new StringResponseWriter
|
||||
func NewStringResponseWriter() *StringResponseWriter {
|
||||
return &StringResponseWriter{body: &bytes.Buffer{}, header: make(http.Header)}
|
||||
}
|
||||
|
||||
// Header capture the Header information
|
||||
func (s StringResponseWriter) Header() http.Header {
|
||||
return s.header
|
||||
}
|
||||
|
||||
// Write captures the response data
|
||||
func (s StringResponseWriter) Write(data []byte) (int, error) {
|
||||
return s.body.Write(data)
|
||||
}
|
||||
|
||||
// WriteHeader captures the status code of the response
|
||||
func (s StringResponseWriter) WriteHeader(statusCode int) {
|
||||
s.headerCode = statusCode
|
||||
}
|
||||
|
||||
// Body returns the response body bytes
|
||||
func (s StringResponseWriter) Body() []byte {
|
||||
return s.body.Bytes()
|
||||
}
|
||||
|
Reference in New Issue
Block a user