当前位置： > > > > 使用 go-sqlmock 并将参数插入模拟查询的问题

使用 go-sqlmock 并将参数插入模拟查询的问题

来源：stackoverflow
2024-04-19 12:06:32
0浏览
收藏

从现在开始，努力学习吧！本文《使用 go-sqlmock 并将参数插入模拟查询的问题》主要讲解了等等相关知识点，我会在米云中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

问题内容

我正在尝试使用 go-sqlmock 模拟我的查询函数并类似地复制数据库表。但是，我没有得到我期望的结果。查询的行为不正常，参数没有插入到查询中并且实际结果不正确。我在这里做错了什么？

这是我正在嘲笑的函数和查询：

func (y *yumdatabase) gettransactionid(pkg string) (int, error) {
    var id int

    queryfortid := "select tid from trans_cmdline where cmdline like '%install " + pkg + "%' order by tid desc limit 1"
    row := y.db.queryrow(queryfortid)
    switch err := row.scan(&id); err {
    case sql.errnorows:
        fmt.println("no rows were returned")
        return 0, err
    case nil:
        return id, nil
    default:
        return 0, err
    }
}

这是模拟测试功能：

func testgettransactionid(t *testing.t) {
    db, mock, err := sqlmock.new()
    if err != nil {
        t.fatalf("err not expected: %v", err)
    }
    pkg := "tcpdump"
    rows := sqlmock.newrows([]string{"tid"}).addrow("1").addrow("3")
    mock.expectquery("select tid from trans_cmdline where cmdline like '%install " + pkg + "%' order by tid desc limit 1").willreturnrows(rows)

    mockdb := &yumdatabase{
        db: db,
    }

    got, err := mockdb.gettransactionid("tcpdump")
    assert.equal(t, 3, got)
}

如果上述工作按预期进行，我会在“got”中返回“3”，但我会返回“1”

其次，是否可以将 rows 更改为以下内容：

rows := sqlmock.newrows([]string{"tid", "cmdline"}).addrow("1", "install test").addrow("3", "delete test2")

实际上进行了比较“where cmdline like ‘%install xyz%’”，因为我尝试了这个，并且收到了以下错误（所有主要代码都构建并工作，包括查询，所以这是一个问题我猜是我写的模拟代码）：

error sql: expected 2 destination arguments in Scan, not 1

我希望看到从 sql 查询返回的最高 tid，而不是“addrow”中指定的第一个 tid，并且我希望查询实现对模拟中的“cmdline”行的检查。

正确答案

我就是这样管理你的需求的。首先，让我分享代码，然后，我将引导您完成所有相关更改。该代码包含在两个文件中：repo.go 和 repo_test.go。

repo.go 文件

package repo

import (
    "database/sql"
    "fmt"
)

func gettransactionid(db *sql.db, pkg string) (int, error) {
    var id int
    row := db.queryrow("select tid from trans_cmdline where cmdline like '%install $1%' order by tid desc limit 1", pkg)
    switch err := row.scan(&id); err {
    case sql.errnorows:
        fmt.println("no rows were returned")
        return 0, err
    case nil:
        return id, nil
    default:
        return 0, err
    }
}

这里有两个小改进：

1. *sql.db 作为参数传入。正如最佳实践所建议的，函数是一等公民。这就是为什么我更愿意尽可能坚持使用它们。
2. 我使用准备好的语句来传递查询的参数。不是简单的字符串连接。因此，可以更轻松地拦截传递给查询的参数并对其设置期望

现在让我们切换到测试代码。

repo_test.go 文件

package repo

import (
    "database/sql"
    "testing"

    "github.com/DATA-DOG/go-sqlmock"
    "github.com/stretchr/testify/assert"
)

func TestGetTransactionId(t *testing.T) {
    db, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))
    if err != nil {
        t.Fatalf("err not expected while opening mock db, %v", err)
    }
    t.Run("HappyPath", func(t *testing.T) {
        rows := sqlmock.NewRows([]string{"tid"}).AddRow("1")
        mock.ExpectQuery("SELECT tid FROM trans_cmdline WHERE cmdline LIKE '%install $1%' ORDER BY tid DESC LIMIT 1").
            WithArgs("tcpdump").
            WillReturnRows(rows)

        got, err := GetTransactionId(db, "tcpdump")

        assert.Equal(t, 1, got)
        assert.Nil(t, err)
    })

    t.Run("NoRowsReturned", func(t *testing.T) {
        mock.ExpectQuery("SELECT tid FROM trans_cmdline WHERE cmdline LIKE '%install $1%' ORDER BY tid DESC LIMIT 1").
            WithArgs("tcpdump").
            WillReturnError(sql.ErrNoRows)

        got, err := GetTransactionId(db, "tcpdump")

        assert.Equal(t, 0, got)
        assert.Equal(t, sql.ErrNoRows, err)
    })
}

这里，您需要注意更多更改：

1. 在实例化 db 和 mock 时，您应该将 sqlmock.querymatcherequal 作为参数传递给。因此，它将完全匹配查询。
2. expectquery 方法现在使用准备好的语句功能并需要一个参数（例如本例中的 tcpdump）。
3. 重构了断言以利用 github.com/stretchr/testify/assert 包。

我希望这可以帮助您解决问题，请告诉我！

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持米云！更多关于Golang的相关知识，也可关注米云公众号。