postgresql - 如何在go中模拟sql更新

Question

我正在尝试编写一个单元测试来更新 golang 中 postgresql 中的记录

数据库功能是

CREATE OR REPLACE FUNCTION test.update(param_id text, param_data jsonb, id2 text) RETURNS void AS
$$
BEGIN
    UPDATE test.test_table
    SET data = param_data,
    WHERE id = (SELECT id FROM test.test_table2 WHERE name = id2)
END;
$$
LANGUAGE plpgsql;

单元测试是

rows := sqlmock.NewRows([]string{"result"}).AddRow("")

mock.ExpectBegin()
mock.ExpectQuery(fmt.Sprintf("SELECT %v.update(.+)", schema)).WithArgs(1, data).WillReturnRows(rows)
mock.ExpectCommit()

我还需要在里面模拟选择查询吗？如何模拟 void 函数？

Answer 1

我用过下面的包

import (sql "database/sql")

这使我能够检查以下内容，

....
var rows sql.Result
...
<execute query >
...
rows.RowsAffected()


rows.RowsAffected() 提供更新的行数。