GOMODULES
/
pgx
mirror of https://github.com/jackc/pgx.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add mock close

batch-wip
Jack Christensen 2017-05-22 08:51:23 -05:00
parent 749fdfe7d5
commit 21d2ed0934
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
stdlib/sql_test.go
View File

@ -1036,6 +1036,7 @@ func TestConnQueryContextCancel(t *testing.T) {
pgmock.ExpectMessage(&pgproto3.Sync{}), pgmock.ExpectMessage(&pgproto3.Sync{}),
pgmock.SendMessage(&pgproto3.BindComplete{}), pgmock.SendMessage(&pgproto3.BindComplete{}),
pgmock.WaitForClose(),
) )
server, err := pgmock.NewServer(script) server, err := pgmock.NewServer(script)
@ -1053,7 +1054,7 @@ func TestConnQueryContextCancel(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("sql.Open failed: %v", err) t.Fatalf("sql.Open failed: %v", err)
} }
// defer closeDB(t, db) // mock DB doesn't close correctly yet defer db.Close()
ctx, cancelFn := context.WithCancel(context.Background()) ctx, cancelFn := context.WithCancel(context.Background())