36 lines
779 B
Go
36 lines
779 B
Go
|
//go:build unit
|
||
|
|
|
||
|
|
package testutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
gin.SetMode(gin.TestMode)
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewGinTestContext 创建一个 Gin 测试上下文和 ResponseRecorder。
|
||
|
|
// body 为空字符串时创建无 body 的请求。
|
||
|
|
func NewGinTestContext(method, path, body string) (*gin.Context, *httptest.ResponseRecorder) {
|
||
|
|
rec := httptest.NewRecorder()
|
||
|
|
c, _ := gin.CreateTestContext(rec)
|
||
|
|
|
||
|
|
var bodyReader io.Reader
|
||
|
|
if body != "" {
|
||
|
|
bodyReader = strings.NewReader(body)
|
||
|
|
}
|
||
|
|
|
||
|
|
c.Request = httptest.NewRequest(method, path, bodyReader)
|
||
|
|
if method == http.MethodPost || method == http.MethodPut || method == http.MethodPatch {
|
||
|
|
c.Request.Header.Set("Content-Type", "application/json")
|
||
|
|
}
|
||
|
|
|
||
|
|
return c, rec
|
||
|
|
}
|