go/go-sqlmock
1
0
Fork 0
mirror of https://github.com/DATA-DOG/go-sqlmock.git synced 2025-07-03 00:36:52 +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

7
sqlmock.go
View File

@ -176,6 +176,13 @@ func (c *sqlmock) ExpectationsWereMet() error {
return fmt.Errorf("expected prepared statement to be closed, but it was not: %s", prep)
}
}
// must check whether all expected queried rows are closed
if query, ok := e.(*ExpectedQuery); ok {
if query.rowsMustBeClosed && !query.rowsWereClosed {
return fmt.Errorf("expected query rows to be closed, but it was not: %s", query)
}
}
}
return nil
}