go/go-sqlmock
1
0
Fork 0
mirror of https://github.com/DATA-DOG/go-sqlmock.git synced 2025-06-19 00:07:38 +02:00
Code Issues Releases Activity

delay for WillDelayFor() duration, before returning error set by WillReturnError()

Browse Source 
If both WillDelayFor() and WillReturnError() are called, there should
be a delay before the error is returned.  Applied to Begin, Exec,
Query, and Prepare.  Also, the context versions of same.

And a couple of unit tests.
This commit is contained in:
Wachowski
2018-02-20 17:30:20 -07:00
parent a7b3087be7
commit 32675e3ad1
4 changed files with 158 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
sqlmock.go
View File

@ -168,11 +168,13 @@ func (c *sqlmock) ExpectationsWereMet() error {
// Begin meets http://golang.org/pkg/database/sql/driver/#Conn interface
func (c *sqlmock) Begin() (driver.Tx, error) {
ex, err := c.begin()
if ex != nil {
time.Sleep(ex.delay)
}
if err != nil {
return nil, err
}
time.Sleep(ex.delay)
return c, nil
}
@ -228,11 +230,13 @@ func (c *sqlmock) Exec(query string, args []driver.Value) (driver.Result, error)
}
ex, err := c.exec(query, namedArgs)
if ex != nil {
time.Sleep(ex.delay)
}
if err != nil {
return nil, err
}
time.Sleep(ex.delay)
return ex.result, nil
}
@ -283,7 +287,7 @@ func (c *sqlmock) exec(query string, args []namedValue) (*ExpectedExec, error) {
expected.triggered = true
if expected.err != nil {
return nil, expected.err // mocked to return error
return expected, expected.err // mocked to return error
}
if expected.result == nil {
@ -304,11 +308,13 @@ func (c *sqlmock) ExpectExec(sqlRegexStr string) *ExpectedExec {
// Prepare meets http://golang.org/pkg/database/sql/driver/#Conn interface
func (c *sqlmock) Prepare(query string) (driver.Stmt, error) {
ex, err := c.prepare(query)
if ex != nil {
time.Sleep(ex.delay)
}
if err != nil {
return nil, err
}
time.Sleep(ex.delay)
return &statement{c, ex, query}, nil
}
@ -385,11 +391,13 @@ func (c *sqlmock) Query(query string, args []driver.Value) (driver.Rows, error)
}
ex, err := c.query(query, namedArgs)
if ex != nil {
time.Sleep(ex.delay)
}
if err != nil {
return nil, err
}
time.Sleep(ex.delay)
return ex.rows, nil
}
@ -442,7 +450,7 @@ func (c *sqlmock) query(query string, args []namedValue) (*ExpectedQuery, error)
expected.triggered = true
if expected.err != nil {
return nil, expected.err // mocked to return error
return expected, expected.err // mocked to return error
}
if expected.rows == nil {