tests: minor

kataras · Nov 5, 2023 · 11f04ff · 11f04ff
1 parent 09da695
commit 11f04ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/db_information_example_test.go b/db_information_example_test.go
@@ -16,7 +16,7 @@ func handleExampleError(err error) {
 }
 
 func ExampleDB_ListColumns() {
-	db, err := openTestConnection(false)
+	db, err := openTestConnection(true)
 	if err != nil {
 		handleExampleError(err)
 		return

diff --git a/db_table_listener.go b/db_table_listener.go
@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"net"
 	"strings"
 	"sync/atomic"
 
@@ -234,7 +235,7 @@ func (db *DB) ListenTable(ctx context.Context, opts *ListenTableOptions, callbac
 
 			notification, err := conn.Accept(ctx)
 			if err != nil {
-				if errors.Is(err, io.ErrUnexpectedEOF) {
+				if errors.Is(err, io.ErrUnexpectedEOF) || errors.Is(err, net.ErrClosed) {
 					return // may produced by close.
 				}