Skip to content

Commit 4ee1fda

Browse files
committed
closes #159
1 parent 7500eca commit 4ee1fda

File tree

2 files changed

+33
-0
lines changed

2 files changed

+33
-0
lines changed

go.mod

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,3 @@
11
module github.com/DATA-DOG/go-sqlmock
2+
3+
go 1.12

sqlmock_go18_test.go

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,3 +474,34 @@ func TestContextExecErrorDelay(t *testing.T) {
474474
t.Errorf("expecting a delay of less than %v before error, actual delay was %v", delay, elapsed)
475475
}
476476
}
477+
478+
func TestStatementTX(t *testing.T) {
479+
t.Parallel()
480+
db, mock, err := New()
481+
if err != nil {
482+
t.Errorf("an error '%s' was not expected when opening a stub database connection", err)
483+
}
484+
defer db.Close()
485+
486+
prep := mock.ExpectPrepare("SELECT")
487+
mock.ExpectBegin()
488+
489+
prep.ExpectQuery().WithArgs(1).WillReturnError(errors.New("fast fail"))
490+
491+
stmt, err := db.Prepare("SELECT title, body FROM articles WHERE id = ?")
492+
if err != nil {
493+
t.Fatalf("unexpected error on prepare: %v", err)
494+
}
495+
496+
tx, err := db.Begin()
497+
if err != nil {
498+
t.Fatalf("unexpected error on begin: %v", err)
499+
}
500+
501+
// upgrade connection for statement
502+
txStmt := tx.Stmt(stmt)
503+
_, err = txStmt.Query(1)
504+
if err == nil || err.Error() != "fast fail" {
505+
t.Fatalf("unexpected result: %v", err)
506+
}
507+
}

0 commit comments

Comments
 (0)