mirror of
https://github.com/DATA-DOG/go-sqlmock.git
synced 2025-04-02 21:45:24 +02:00
closes #159
This commit is contained in:
parent
7500ecac5e
commit
4ee1fdaf44
@ -474,3 +474,34 @@ func TestContextExecErrorDelay(t *testing.T) {
|
||||
t.Errorf("expecting a delay of less than %v before error, actual delay was %v", delay, elapsed)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStatementTX(t *testing.T) {
|
||||
t.Parallel()
|
||||
db, mock, err := New()
|
||||
if err != nil {
|
||||
t.Errorf("an error '%s' was not expected when opening a stub database connection", err)
|
||||
}
|
||||
defer db.Close()
|
||||
|
||||
prep := mock.ExpectPrepare("SELECT")
|
||||
mock.ExpectBegin()
|
||||
|
||||
prep.ExpectQuery().WithArgs(1).WillReturnError(errors.New("fast fail"))
|
||||
|
||||
stmt, err := db.Prepare("SELECT title, body FROM articles WHERE id = ?")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error on prepare: %v", err)
|
||||
}
|
||||
|
||||
tx, err := db.Begin()
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error on begin: %v", err)
|
||||
}
|
||||
|
||||
// upgrade connection for statement
|
||||
txStmt := tx.Stmt(stmt)
|
||||
_, err = txStmt.Query(1)
|
||||
if err == nil || err.Error() != "fast fail" {
|
||||
t.Fatalf("unexpected result: %v", err)
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user