70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package exec
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type scannerSuite struct {
|
|
suite.Suite
|
|
}
|
|
|
|
func TestScanner(t *testing.T) {
|
|
suite.Run(t, &scannerSuite{})
|
|
}
|
|
|
|
func (s *scannerSuite) TestScanStructs() {
|
|
type StructWithTags struct {
|
|
Address string `db:"address"`
|
|
Name string `db:"name"`
|
|
}
|
|
db, mock, err := sqlmock.New()
|
|
s.Require().NoError(err)
|
|
|
|
mock.ExpectQuery(`SELECT \* FROM "items"`).
|
|
WithArgs().
|
|
WillReturnRows(sqlmock.NewRows([]string{"address", "name"}).
|
|
AddRow(testAddr1, testName1).
|
|
AddRow(testAddr2, testName2),
|
|
)
|
|
rows, err := db.Query(`SELECT * FROM "items"`)
|
|
s.Require().NoError(err)
|
|
|
|
sc := NewScanner(rows)
|
|
|
|
result := make([]StructWithTags, 0)
|
|
err = sc.ScanStructs(result)
|
|
s.Require().EqualError(err, errUnsupportedScanStructsType.Error())
|
|
|
|
err = sc.ScanStructs(&result)
|
|
s.Require().NoError(err)
|
|
s.Require().ElementsMatch(
|
|
[]StructWithTags{{Address: testAddr1, Name: testName1}, {Address: testAddr2, Name: testName2}},
|
|
result,
|
|
)
|
|
}
|
|
|
|
func (s *scannerSuite) TestScanVals() {
|
|
db, mock, err := sqlmock.New()
|
|
s.Require().NoError(err)
|
|
|
|
mock.ExpectQuery(`SELECT "id" FROM "items"`).
|
|
WithArgs().
|
|
WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1).AddRow(2))
|
|
|
|
rows, err := db.Query(`SELECT "id" FROM "items"`)
|
|
s.Require().NoError(err)
|
|
|
|
sc := NewScanner(rows)
|
|
|
|
result := make([]int, 0)
|
|
err = sc.ScanVals(result)
|
|
s.Require().EqualError(err, errUnsupportedScanValsType.Error())
|
|
|
|
err = sc.ScanVals(&result)
|
|
s.Require().NoError(err)
|
|
s.Require().ElementsMatch([]int{1, 2}, result)
|
|
}
|