go/go-sqlmock
1
0
Fork 0
mirror of https://github.com/DATA-DOG/go-sqlmock.git synced 2025-07-15 01:14:22 +02:00
Code Issues Releases Activity

closes #114 allow expecting rows to be closed

Browse Source
This commit is contained in:
gedi
2018-12-11 17:38:12 +02:00
parent a6e6646ad9
commit f7b0b9305b
7 changed files with 70 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
expectations.go
View File

@ -125,8 +125,10 @@ func (e *ExpectedRollback) String() string {
// Returned by *Sqlmock.ExpectQuery.
type ExpectedQuery struct {
queryBasedExpectation
rows driver.Rows
delay time.Duration
rows driver.Rows
delay time.Duration
rowsMustBeClosed bool
rowsWereClosed bool
}
// WithArgs will match given expected args to actual database query arguments.
@ -137,6 +139,12 @@ func (e *ExpectedQuery) WithArgs(args ...driver.Value) *ExpectedQuery {
return e
}
// RowsWillBeClosed expects this query rows to be closed.
func (e *ExpectedQuery) RowsWillBeClosed() *ExpectedQuery {
e.rowsMustBeClosed = true
return e
}
// WillReturnError allows to set an error for expected database query
func (e *ExpectedQuery) WillReturnError(err error) *ExpectedQuery {
e.err = err