From 21d2ed09349333d874bb165c37781595ab0a53b3 Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Mon, 22 May 2017 08:51:23 -0500 Subject: [PATCH] Add mock close --- stdlib/sql_test.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stdlib/sql_test.go b/stdlib/sql_test.go index 415864cd..bfeb07c6 100644 --- a/stdlib/sql_test.go +++ b/stdlib/sql_test.go @@ -1036,6 +1036,7 @@ func TestConnQueryContextCancel(t *testing.T) { pgmock.ExpectMessage(&pgproto3.Sync{}), pgmock.SendMessage(&pgproto3.BindComplete{}), + pgmock.WaitForClose(), ) server, err := pgmock.NewServer(script) @@ -1053,7 +1054,7 @@ func TestConnQueryContextCancel(t *testing.T) { if err != nil { 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())