3

sqlmock用来模拟雪花查询执行。这可能看起来很奇怪,但是对于以下查询:

rows, err = db.Query(sqlQuery)

我需要模拟返回nil和。请注意,我想要的不是一个空集。这是我的功能。我也尝试了评论中的陈述。rowserrnilrows

func mockSnowflake(t *testing.T, sqlQuery string) (*sql.DB, sqlmock.Sqlmock) {
    db, mock, err := sqlmock.New()
    // 5 is because I'll be calling db.Query() 5 times
    for i := 1; i <= 5; i++ {
        //mock.ExpectQuery(sqlQuery).WillReturnRows(sqlmock.NewRows(nil))
        //mock.ExpectQuery(sqlQuery).WillReturnRows(sqlmock.NewRows([]string {"Test Column"}).AddRow(nil))
        mock.ExpectQuery(sqlQuery).WillReturnRows(nil)
    }
    return db, mock
}

我应该如何更改功能?

4

0 回答 0