Fix test on Travis

2017-07-22 08:41:13 -05:00
parent 2e5f5e0c9d
commit a147e0f3b8
1 changed files with 1 additions and 0 deletions
@@ -1211,6 +1211,7 @@ func TestStmtQueryContextCancel(t *testing.T) {
 		pgmock.ExpectMessage(&pgproto3.Sync{}),

 		pgmock.SendMessage(&pgproto3.BindComplete{}),
+		pgmock.WaitForClose(),
 	)

 	server, err := pgmock.NewServer(script)