0

我有一个函数用于迭代查询的结果集：

func readRows(rows *sql.Rows, translator func(*sql.Rows) error) error {
    defer rows.Close()

    // Iterate over each row in the rows and scan each; if an error occurs then return
    for shouldScan := rows.Next(); shouldScan; {
        if err := translator(rows); err != nil {
            return err
        }
    }

    // Check if the rows had an error; if they did then return them. Otherwise,
    // close the rows and return an error if the close function fails
    if err := rows.Err(); err != nil {
        return err
    }

    return nil
}

该translator函数主要负责调用Scan对象*sql.Rows。这方面的一个例子是：

readRows(rows, func(scanner *sql.Rows) error {
    var entry gopb.TestObject

    // Embed the variables into a list that we can use to pull information out of the rows
    scanned := []interface{}{...}

    if err := scanner.Scan(scanned...); err != nil {
        return err
    }

    entries = append(entries, &entry)
    return nil
})

我为这段代码写了一个单元测试：

// Create the SQL mock and the RDS reqeuster
db, mock, _ := sqlmock.New()
requester := Requester{conn: db}
defer db.Close()

// Create the rows we'll use for testing the query
rows := sqlmock.NewRows([]string{"id", "data"}).
                AddRow(0, "data")

// Verify the command order for the transaction
mock.ExpectBegin()
mock.ExpectQuery(regexp.QuoteMeta("SELECT `id`, `data`, FROM `data`")).WillReturnRows(rows)
mock.ExpectRollback()

// Attempt to get the data
data, err := requester.GetData(context.TODO())

但是，它似乎Next被无限调用。我不确定这是sqlmock我的代码的问题还是问题。任何帮助，将不胜感激。

4

0 回答 0