1
2
3 package sqlmock
4
5 import (
6 "errors"
7 "testing"
8 )
9
10 func TestExpectedPreparedStatementCloseError(t *testing.T) {
11 conn, mock, err := New()
12 if err != nil {
13 t.Fatal("failed to open sqlmock database:", err)
14 }
15
16 mock.ExpectBegin()
17 want := errors.New("STMT ERROR")
18 mock.ExpectPrepare("SELECT").WillReturnCloseError(want)
19
20 txn, err := conn.Begin()
21 if err != nil {
22 t.Fatal("unexpected error while opening transaction:", err)
23 }
24
25 stmt, err := txn.Prepare("SELECT")
26 if err != nil {
27 t.Fatal("unexpected error while preparing a statement:", err)
28 }
29
30 if err := stmt.Close(); err != want {
31 t.Fatalf("got = %v, want = %v", err, want)
32 }
33 }
34
View as plain text