gopkg/pgx
2
0
Fork 0
Code Activity

Add mock close

Browse Source
This commit is contained in:
Jack Christensen
2017-05-22 08:51:23 -05:00
parent 749fdfe7d5
commit 21d2ed0934
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stdlib/sql_test.go
+2 -1
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())