测试 SQL 表值参数在sqlmock中

一分耕耘，一分收获！既然都打开这篇《测试 SQL 表值参数在sqlmock中》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

我有一个函数，旨在使用表值参数和过程将大量元素插入 mssql 数据库。

func (requester *requester) doquery(ctx context.context, dtos interface{}) error {
    conn, err := requester.conn.conn(ctx)
    if err != nil {
        return err
    }

    defer func() {
        if clerr := conn.close(); clerr != nil {
            err = clerr
        }
    }()

    tx, err := conn.begintx(ctx, &sql.txoptions{isolation: sql.levelrepeatableread, readonly: false})
    if err != nil {
        return err
    }

    defer func() {
        if p := recover(); p != nil {
            tx.rollback()
            panic(p)
        } else if err != nil {
            tx.rollback()
        } else {
            tx.commit()
    }()

    param := sql.named("tvp", mssql.tvp{
        typename: "mytypename",
        value:    dtos,
    })

    return tx.execcontext(ctx, "exec [dbo].[usp_insertconsumption] @tvp", param)
}

我为此函数编写的测试如下（请注意，它取决于 ginkgo 和 gomega）：

describe("sql tests", func() {

    it("addconsumption - no failures - added", func() {

        db, mock, _ := sqlmock.new()
        requester := requester{conn: db}
        defer db.close()

        mock.expectbegin()
        mock.expectexec(regexp.quotemeta("exec [dbo].[usp_insertconsumption] @tvp")).
            withargs("").willreturnresult(sqlmock.newresult(1, 1))
        mock.expectexec(regexp.quotemeta("exec [dbo].[usp_inserttags] @tvp")).
            withargs("").willreturnresult(sqlmock.newresult(1, 1))
        mock.expectcommit()

        err := requester.doquery(context.todo(), generatedata())
        expect(err).shouldnot(haveoccurred())
        expect(mock.expectationsweremet()).shouldnot(haveoccurred())
    })
})

现在，这段代码是为 mysql 上下文编写的，由于我已将代码移植到 mssql，所以我遇到了一个奇怪的错误：

sql: converting argument with name \"TVP\" type: unsupported type mssql.TVP, a struct

sqlmock 似乎正在尝试在 tvp 对象上调用 convertvalue，但该调用无效。那么，如何让 sqlmock 正确处理这个值，以便我可以围绕查询进行单元测试？

解决方案

我在这里发现sqlmock有一个名为valueconverteroption的函数，接受driver.valueconverter接口的实现。这将用于代替每次调用 convertvalue 的标准函数。如果您想在 execcontext 函数接收非标准参数（本例中为 tvp）时对其进行测试，那么您可以使用此函数将自定义转换逻辑注入 sqlmock。

type mocktvpconverter struct {}

func (converter *mocktvpconverter) convertvalue(raw interface{}) (driver.value, error) {

    // since this function will take the place of every call of convertvalue, we will inevitably
    // the fake string we return from this function so we need to check whether we've recieved
    // that or a tvp. more extensive logic may be required
    switch inner := raw.(type) {
    case string:
        return raw.(string), nil
    case mssql.tvp:

        // first, verify the type name
        expect(inner.typename).should(equal("mytypename"))

        // verification logic here

        // finally, return a fake value that we can use when verifying the arguments
        return "passed", nil
    }

    // we had an invalid type; return an error
    return nil, fmt.errorf("invalid type")
}

这意味着测试将变为：

Describe("SQL Tests", func() {

    It("AddConsumption - No failures - Added", func() {

        db, mock, _ := sqlmock.New(sqlmock.ValueConverterOption(&mockTvpConverter{}))
        requester := Requester{conn: db}
        defer db.Close()

        mock.ExpectBegin()
        mock.ExpectExec(regexp.QuoteMeta("EXEC [dbo].[usp_InsertConsumption] @TVP")).
            WithArgs("PASSED").WillReturnResult(sqlmock.NewResult(1, 1))
        mock.ExpectExec(regexp.QuoteMeta("EXEC [dbo].[usp_InsertTags] @TVP")).
            WithArgs("PASSED").WillReturnResult(sqlmock.NewResult(1, 1))
        mock.ExpectCommit()

        err := requester.doQuery(context.TODO(), generateData())
        Expect(err).ShouldNot(HaveOccurred())
        Expect(mock.ExpectationsWereMet()).ShouldNot(HaveOccurred())
    })
})

本篇关于《测试 SQL 表值参数在sqlmock中》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！