feat: backend core - auth, user, role, permission, device, webhook, monitoring, cache, repository, service, middleware, API handlers
This commit is contained in:
35
internal/testutil/httptest.go
Normal file
35
internal/testutil/httptest.go
Normal file
@@ -0,0 +1,35 @@
|
||||
//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
|
||||
}
|
||||
Reference in New Issue
Block a user