严格来说,你不能“嘲笑”*gin.Context
以一种有意义的方式,因为它是struct
具有未导出的字段和方法。
此外,你不能传递到r.POST()
类型不是 a 的函数gin.HandlerFunc
, 定义为func(*gin.Context)
。您的处理程序的类型CreateUrl(c Icontext)
根本不匹配。
如果您的目标是对 Gin 处理程序进行单元测试,那么您绝对不必模拟*gin.Context
。您应该做的就是在其中设置测试值。为此,您可以简单地使用gin.CreateTestContext()
并手动初始化其中一些字段。更多信息here https://stackoverflow.com/questions/66952761/how-to-unit-test-a-go-gin-handler-function/66953599#66953599.
如果出于其他原因,您的目标是提供以下功能的替代实现*gin.Context
为了在处理程序中使用,您可以做的是使用自己的替代方法定义自己的类型并嵌入*gin.Context
in it.
在实践中:
type MyGinContext struct {
*gin.Context
}
func (m *MyGinContext) BindJSON(obj interface{}) error {
fmt.Println("my own BindJSON")
return m.Context.BindJSON(obj) // or entirely alternative implementation
}
// Using the appropriate function signature now
func (repository *UrlRepo) CreateUrl(c *gin.Context) {
myCtx := &MyGinContext{c}
var url models.Url
_ = myCtx.BindJSON(&url) // will also print "my own BindJSON"
// ...
// other gin.Context methods are promoted and available on MyGinContext
myCtx.Status(200)
}
但老实说,我不确定为什么你想要覆盖某些方法*gin.Context
。如果您想提供不同的绑定逻辑,甚至不同的渲染,您可以实现库已经公开的接口。例如:
实现绑定:
c.ShouldBindWith()
将接口作为第二个参数binding.Binding
您可以实施:
type MyBinder struct {
}
func (m *MyBinder) Name() string {
return "foo"
}
func (m *MyBinder) Bind(*http.Request, interface{}) error {
// stuff
return nil
}
func MyHandler(c *gin.Context) {
var foo struct{/*fields*/}
c.ShouldBindWith(&foo, &MyBinder{})
}
实现渲染器:
type MyRenderer struct {
}
type Render interface {
func (m *MyRenderer) Render(http.ResponseWriter) error {
// ...
return nil
}
func (m *MyRenderer) WriteContentType(w http.ResponseWriter) {
header := w.Header()
if val := header["Content-Type"]; len(val) == 0 {
header["Content-Type"] = "application/foo+bar"
}
}
func MyHandler(c *gin.Context) {
c.Render(200, &MyRenderer{})
}