unit-testing - 如何在 go-sqlmock 中添加多个结果集？

Question

我有一个返回 2 个结果集的数据库查询，我想对执行此查询的 go 函数进行单元测试。虽然我可以像这样添加和测试行:

myMockRows:=sqlmock.NewRows([]string{"col1","col2"}).AddRow("col1val1", "col2val2")
mock.ExpectQuery("my_stored_procedure").WithArgs(arg1, arg2).WillReturnRows(myMockRows)

我不确定如何继续在我的 rows 对象中创建多个结果集。我该怎么做？

Answer 1

我已经尝试过 @Nikhil Vandanapu 的回答，但我没有得到想要的输出。由于某种原因，它只使用了 myMockRows 而 myMockRows2 被忽略了。我做了一些阅读，我们可以执行以下操作以使其返回多行。

myMockRows:=sqlmock.NewRows([]string{"col1","col2"})
.AddRow("col1val1", "col2val2")
.AddRow("col1val1", "col2val2")

mock.ExpectQuery("my_stored_procedure").WillReturnRows(myMockRows)

根据 godoc .值 slice 返回相同的实例以执行后续操作。

添加这个 blog如果您想更轻松地阅读该主题，请发帖