使用 gosqlmock 并将参数插入模拟查询的问题

2024年 2月 11日 78.1k 0

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

在PHP开发中,数据库操作是非常常见的需求,而对于一些需要测试的场景,我们又不希望直接操作真实的数据库。这时候,我们可以使用go-sqlmock来模拟数据库查询,从而达到我们想要的测试效果。本文将向大家介绍如何使用go-sqlmock,并将参数插入模拟查询的问题。无论你是PHP开发初学者还是有一定经验的开发者,通过学习本文,你都能够轻松掌握这一技巧,提升你的开发效率。

问题内容

我正在尝试使用 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.gorepo_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
}
}

登录后复制

这里有两个小改进:

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

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

    登录后复制

    这里,您需要注意更多更改:

  • 在实例化 dbmock 时,您应该将 sqlmock.querymatcherequal 作为参数传递给。因此,它将完全匹配查询。
  • expectquery 方法现在使用准备好的语句功能并需要一个参数(例如本例中的 tcpdump)。
  • 重构了断言以利用 github.com/stretchr/testify/assert 包。
  • 我希望这可以帮助您解决问题,请告诉我!

    以上就是使用 go-sqlmock 并将参数插入模拟查询的问题的详细内容,更多请关注每日运维网(www.mryunwei.com)其它相关文章!

    相关文章

    JavaScript2024新功能:Object.groupBy、正则表达式v标志
    PHP trim 函数对多字节字符的使用和限制
    新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
    使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
    为React 19做准备:WordPress 6.6用户指南
    如何删除WordPress中的所有评论

    发布评论