如何模拟 gin.Context？

Question

嗨，我一直在尝试模拟一个 gin.Context，但我无法让它工作我正在尝试他们在此中所做的事情，但它不适用于我的路由器，这是我遇到的错误

r.POST("/urls", urlRepo.CreateUrl)

cannot use urlRepo.CreateUrl (value of type func(c controllers.Icontext)) as gin.HandlerFunc value in argument to r.POSTcompilerIncompatibleAssign

这是我为稍后模拟创建的接口以及我将在其中测试的方法

type Icontext interface {
  BindJSON(obj interface{}) error
  JSON(code int, obj interface{})
  AbortWithStatus(code int)
  AbortWithStatusJSON(code int, jsonObj interface{})

}

func (repository *UrlRepo) CreateUrl(c Icontext) {
    var url models.Url
    c.BindJSON(&url)
    if !validators.IsCreateJsonCorrect(url) {
        c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "Error format in Short or Full"})
        return
    }
    err := repository.reposito.CreateUrl(repository.Db, &url)
    if err != nil {
        c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err})
        return
    }
    c.JSON(http.StatusOK, url)
}

而不是

func (repository *UrlRepo) CreateUrl(c Icontext)

是

func (repository *UrlRepo) CreateUrl(c *gin.Context)

Answer 1

如果您使用 gin-gonic 作为您的 http 路由器，您入口点的参数应该是 *gin.Context。

因此，例如，您应该替换为：

func (repository *UrlRepo) CreateUrl(c Icontext) {

有了这个

func (repository *UrlRepo) CreateUrl(c *gin.Context) {

这样您应该能够使用模拟杜松子酒上下文作为单元测试的参数

Answer 2

严格来说，您不能以有意义的方式“模拟”*gin.Context，因为它是具有未导出字段和方法的 struct。

此外，您不能将类型不是 gin.HandlerFunc 的函数传递给 r.POST()，定义为 func(*gin.Context)。您的处理程序类型 CreateUrl(c Icontext) 根本不匹配。

如果您的目标是对 Gin 处理程序进行单元测试，您绝对不必模拟 *gin.Context。您应该做的是在其中设置测试值。为此，您可以简单地使用 gin.CreateTestContext() 并手动初始化其中的一些字段。更多信息 .

如果出于某些其他原因，您的目标是提供 *gin.Context 功能的替代实现以在您的处理程序中使用，您可以做的是使用您自己的替代方法定义您自己的类型并嵌入其中的*gin.Context。

实践中：

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{})
}

如何模拟 gin.Context？

How to mock a gin.Context?

go

testing

mocking

go-gin