当前位置：首页 > 文章列表 > Golang > Go问答 > 问题出在httptest.NewRequest的查询参数上

问题出在httptest.NewRequest的查询参数上

来源：stackoverflow 2024-03-19 10:57:31 0浏览收藏

在使用 Gin 框架、PostgreSQL 数据库和 GORM ORM 测试处理程序时，出现了查询参数未正确提取的问题。测试函数中使用 httptest.NewRequest 创建请求时，查询参数似乎未被使用，导致 ORM 运行错误的 SQL 查询。而浏览器中直接访问该路由时，查询参数可以正常提取。通过检查代码发现，处理程序的拼写错误导致了这个问题，正确的拼写应该是 handler := handlers.GetArtworks(db) 而不是 handler := handlers.GetArtwork(db)。

问题内容

我使用 gin 框架和 postgres db 以及 gorm 作为 orm。

其中一个路由接受查询参数。当我在浏览器中搜索它时，我得到了预期的结果：一个 json 对象数组。这是路线： /artworks/?limit=10&last_id=1

但是，当我尝试测试该路由使用的处理程序时，出现以下错误： routes_test.go:184: [错误] 无法将数据解组到艺术品：json: 无法将对象解组到 []models.artwork 类型的 go 值

orm 尝试在测试函数中运行的查询如下： select * from "artwork_migrate_artwork" where id = ''

因此，当我在浏览器中运行请求时，它会正确提取查询参数，然后 orm 运行正确的 sql 查询。但是当使用 httptest.newrequest 时，似乎没有使用查询参数。

这是我的测试函数：

func testgetartworks(t *testing.t) {
    dsn := fmt.sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s timezone=%s", env_var.host, env_var.user, env_var.password, env_var.dbname, env_var.port, env_var.sslmode, env_var.timezone)
    db, err := gorm.open(postgres.open(dsn), &gorm.config{})
    if err != nil {
        panic("failed to connect to db")
    }

    route := "/artworks/"
    handler := handlers.getartwork(db)
    router := setupgetrouter(handler, route)
    writer := httptest.newrecorder()
    req := httptest.newrequest(http.methodget, "/artworks/?limit=10&last_id=1", nil)
    fmt.println(req)
    router.servehttp(writer, req)

    assert.equal(t, 200, writer.code)

    data, err := ioutil.readall(writer.body)
    if err != nil {
        t.errorf("\u001b[31m[error] unable to read writer.body: %s", err)
    }

    // no body can be unmarshalled
    fmt.println("here is the body:", writer.body.string())

    var artworks []models.artwork
    if err := json.unmarshal(data, &artworks); err != nil {
        t.errorf("\u001b[31m[error] unable to unmarshal data to artworks: %s", err)
    }

    assert.equal(t, 10, len(artworks))
}

这是我的路线处理程序：

func getartworks(db *gorm.db) gin.handlerfunc {
    return func(c *gin.context) {
        limit, err := strconv.atoi(c.query("limit"))
        if err != nil {
            panic(err)
        }
        last_id := c.query("last_id")

        var artworks []models.artwork
        db.where("id > ?", last_id).limit(limit).find(&artworks)

        c.json(http.statusok, artworks)
    }
}
router.get("/artworks", han.getartworks(db))

这是模型结构：

type Artwork struct {
    ID              int       `json:"id"`
    Title           string    `json:"title"`
    Nationality     string    `json:"nationality"`
    Artist_Bio      string    `json:"artist_bio"`
    Desc            string    `json:"desc"`
    Culture         string    `json:"culture"`
    Gender          string    `json:"gender"`
    Nation          string    `json:"nation"`
    Medium          string    `json:"medium"`
    Date_of_Release string    `json:"date_of_release"`
    Image           string    `json:"image"`
    Image_Small     string    `json:"image_small"`
    Last_Modified   time.Time `json:"last_modified"`
    Artist_ID       int       `json:"artist_id"`
    Source_ID       int       `json:"source_id"`
}