1
0
mirror of https://github.com/DATA-DOG/go-sqlmock.git synced 2025-04-02 21:45:24 +02:00
This commit is contained in:
gedi 2019-06-07 09:26:40 +03:00
parent 7500ecac5e
commit 4ee1fdaf44
No known key found for this signature in database
GPG Key ID: 56604CDCCC201556
2 changed files with 33 additions and 0 deletions

2
go.mod
View File

@ -1 +1,3 @@
module github.com/DATA-DOG/go-sqlmock
go 1.12

View File

@ -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)
}
}