41 lines
1021 B
Go
41 lines
1021 B
Go
package test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http/httptest"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// BodyEquals asserts the content from the response recorder with the encoded json of the provided instance.
|
|
func BodyEquals(t assert.TestingT, obj interface{}, recorder *httptest.ResponseRecorder) {
|
|
bytes, err := ioutil.ReadAll(recorder.Body)
|
|
assert.Nil(t, err)
|
|
actual := string(bytes)
|
|
|
|
JSONEquals(t, obj, actual)
|
|
}
|
|
|
|
// JSONEquals asserts the content of the string with the encoded json of the provided instance.
|
|
func JSONEquals(t assert.TestingT, obj interface{}, expected string) {
|
|
bytes, err := json.Marshal(obj)
|
|
assert.Nil(t, err)
|
|
objJSON := string(bytes)
|
|
|
|
assert.JSONEq(t, expected, objJSON)
|
|
}
|
|
|
|
type unreadableReader struct{}
|
|
|
|
func (c unreadableReader) Read([]byte) (int, error) {
|
|
return 0, errors.New("this reader cannot be read")
|
|
}
|
|
|
|
// UnreadableReader returns an unreadable reader, used to mock IO issues.
|
|
func UnreadableReader() io.Reader {
|
|
return unreadableReader{}
|
|
}
|