package statusgo import ( "errors" "testing" "github.com/stretchr/testify/require" ) type nonJSON struct{} func (*nonJSON) MarshalJSON() ([]byte, error) { return nil, errors.New("invalid JSON") } func TestPrepareJSONResponseErrorWithResult(t *testing.T) { data := prepareJSONResponse("0x123", nil) require.Equal(t, `{"result":"0x123"}`, data) data = prepareJSONResponse(&nonJSON{}, nil) require.Contains(t, data, `{"error":{"code":1,"message":`) } func TestPrepareJSONResponseErrorWithError(t *testing.T) { data := prepareJSONResponse("0x123", errors.New("some error")) require.Contains(t, data, `{"error":{"message":"some error"}}`) }