|
|
@ -11,13 +11,11 @@ import (
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|
|
|
"github.com/zeromicro/go-zero/rest/httpx"
|
|
|
|
|
|
|
|
"github.com/zeromicro/go-zero/rest/internal/header"
|
|
|
|
"github.com/zeromicro/go-zero/rest/pathvar"
|
|
|
|
"github.com/zeromicro/go-zero/rest/pathvar"
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
const (
|
|
|
|
const contentLength = "Content-Length"
|
|
|
|
applicationJsonWithUtf8 = "application/json; charset=utf-8"
|
|
|
|
|
|
|
|
contentLength = "Content-Length"
|
|
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
type mockedResponseWriter struct {
|
|
|
|
type mockedResponseWriter struct {
|
|
|
|
code int
|
|
|
|
code int
|
|
|
@ -167,7 +165,7 @@ func TestParseJsonPost(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(
|
|
|
@ -199,7 +197,7 @@ func TestParseJsonPostWithIntSlice(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
bytes.NewBufferString(`{"ages": [1, 2], "years": [3, 4]}`))
|
|
|
|
bytes.NewBufferString(`{"ages": [1, 2], "years": [3, 4]}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(
|
|
|
@ -227,7 +225,7 @@ func TestParseJsonPostError(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
bytes.NewBufferString(payload))
|
|
|
|
bytes.NewBufferString(payload))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
@ -255,7 +253,7 @@ func TestParseJsonPostInvalidRequest(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/",
|
|
|
|
bytes.NewBufferString(payload))
|
|
|
|
bytes.NewBufferString(payload))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/", http.HandlerFunc(
|
|
|
@ -277,7 +275,7 @@ func TestParseJsonPostRequired(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai"`))
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai"`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
@ -455,7 +453,7 @@ func TestParsePtrInRequest(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
bytes.NewBufferString(`{"audio": {"volume": 100}}`))
|
|
|
|
bytes.NewBufferString(`{"audio": {"volume": 100}}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
type (
|
|
|
|
type (
|
|
|
|
Request struct {
|
|
|
|
Request struct {
|
|
|
@ -603,7 +601,7 @@ func TestParseWrappedRequest(t *testing.T) {
|
|
|
|
func TestParseWrappedGetRequestWithJsonHeader(t *testing.T) {
|
|
|
|
func TestParseWrappedGetRequestWithJsonHeader(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodGet, "http://hello.com/kevin/2017", nil)
|
|
|
|
r, err := http.NewRequest(http.MethodGet, "http://hello.com/kevin/2017", nil)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, applicationJsonWithUtf8)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
type (
|
|
|
|
type (
|
|
|
|
Request struct {
|
|
|
|
Request struct {
|
|
|
@ -636,7 +634,7 @@ func TestParseWrappedGetRequestWithJsonHeader(t *testing.T) {
|
|
|
|
func TestParseWrappedHeadRequestWithJsonHeader(t *testing.T) {
|
|
|
|
func TestParseWrappedHeadRequestWithJsonHeader(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodHead, "http://hello.com/kevin/2017", nil)
|
|
|
|
r, err := http.NewRequest(http.MethodHead, "http://hello.com/kevin/2017", nil)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, applicationJsonWithUtf8)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
type (
|
|
|
|
type (
|
|
|
|
Request struct {
|
|
|
|
Request struct {
|
|
|
@ -702,7 +700,7 @@ func TestParseWithAll(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
@ -733,7 +731,7 @@ func TestParseWithAllUtf8(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
bytes.NewBufferString(`{"location": "shanghai", "time": 20170912}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, applicationJsonWithUtf8)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
@ -923,7 +921,7 @@ func TestParseWithMissingAllPaths(t *testing.T) {
|
|
|
|
func TestParseGetWithContentLengthHeader(t *testing.T) {
|
|
|
|
func TestParseGetWithContentLengthHeader(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodGet, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000", nil)
|
|
|
|
r, err := http.NewRequest(http.MethodGet, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000", nil)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, httpx.ApplicationJson)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
r.Header.Set(contentLength, "1024")
|
|
|
|
r.Header.Set(contentLength, "1024")
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
@ -951,7 +949,7 @@ func TestParseJsonPostWithTypeMismatch(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017?nickname=whatever&zipcode=200000",
|
|
|
|
bytes.NewBufferString(`{"time": "20170912"}`))
|
|
|
|
bytes.NewBufferString(`{"time": "20170912"}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, applicationJsonWithUtf8)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
@ -977,7 +975,7 @@ func TestParseJsonPostWithInt2String(t *testing.T) {
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
r, err := http.NewRequest(http.MethodPost, "http://hello.com/kevin/2017",
|
|
|
|
bytes.NewBufferString(`{"time": 20170912}`))
|
|
|
|
bytes.NewBufferString(`{"time": 20170912}`))
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
r.Header.Set(httpx.ContentType, applicationJsonWithUtf8)
|
|
|
|
r.Header.Set(httpx.ContentType, header.JsonContentType)
|
|
|
|
|
|
|
|
|
|
|
|
router := NewRouter()
|
|
|
|
router := NewRouter()
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|
err = router.Handle(http.MethodPost, "/:name/:year", http.HandlerFunc(
|
|
|
|