NO-ISSUE Update tests

Author: linxGnu

errors.go

@@ -1,6 +1,7 @@
 package mssqlx
 
 import (
+	"database/sql"
 	"database/sql/driver"
 	"fmt"
 	"os"
@@ -13,7 +14,7 @@ import (
 // check bad connection
 func isErrBadConn(err error) bool {
 	if err != nil {
-		if err == driver.ErrBadConn || err == mysql.ErrInvalidConn {
+		if err == driver.ErrBadConn || err == sql.ErrConnDone || err == mysql.ErrInvalidConn {
 			return true
 		}
 

retry.go

@@ -12,8 +12,6 @@ func retryFunc(query string, f func() (interface{}, error)) (result interface{},
 		}
 
 		switch err {
-		case sql.ErrConnDone:
-
 		case sql.ErrTxDone, sql.ErrNoRows:
 			return
 
@@ -26,7 +24,7 @@ func retryFunc(query string, f func() (interface{}, error)) (result interface{},
 		}
 	}
 
-	if err == sql.ErrConnDone || isErrBadConn(err) {
+	if isErrBadConn(err) {
 		reportError(query, err)
 	}
 

retry_test.go

@@ -0,0 +1,15 @@
+package mssqlx
+
+import (
+	"database/sql"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRetry(t *testing.T) {
+	_, err := retryFunc("", func() (interface{}, error) {
+		return nil, sql.ErrConnDone
+	})
+	require.Equal(t, sql.ErrConnDone, err)
+}