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