1
0
mirror of https://github.com/DATA-DOG/go-sqlmock.git synced 2024-11-28 08:49:01 +02:00
go-sqlmock/statement_test.go

34 lines
655 B
Go

// +build go1.6
package sqlmock
import (
"errors"
"testing"
)
func TestExpectedPreparedStatemtCloseError(t *testing.T) {
conn, mock, err := New()
if err != nil {
t.Fatal("failed to open sqlmock database:", err)
}
mock.ExpectBegin()
want := errors.New("STMT ERROR")
mock.ExpectPrepare("SELECT").WillReturnCloseError(want)
txn, err := conn.Begin()
if err != nil {
t.Fatal("unexpected error while opening transaction:", err)
}
stmt, err := txn.Prepare("SELECT")
if err != nil {
t.Fatal("unexpected error while preparing a statement:", err)
}
if err := stmt.Close(); err != want {
t.Fatalf("Got = %v, want = %v", err, want)
}
}