如何在go-sqlmock中添加多个结果集？

Question

I have a db query that returns 2 result sets and I would like to unit test the go function that performs this query. While I can add and test rows like this:

myMockRows:=sqlmock.NewRows([]string{"col1","col2"}).AddRow("col1val1", "col2val2")
mock.ExpectQuery("my_stored_procedure").WithArgs(arg1, arg2).WillReturnRows(myMockRows)

I am not sure how to proceed with creating multiple result sets in my rows object. How do I do that?

Answer 1

Do something like this:

myMockRows:=sqlmock.NewRows([]string{"col1","col2"}).AddRow("col1val1", "col2val2")
myMockRows2:=sqlmock.NewRows([]string{"col3","col4"}).AddRow("col3val1", "col4val2")
mock.ExpectQuery("my_stored_procedure").WithArgs(arg1, arg2).WillReturnRows(myMockRows, myMockRows2)

Since WillReturnRows accepts multiple row objects and forms a slice, use it to construct the next Result Set.