wait for serve() to return before returning from server.Close()

2017-12-20 14:20:46 +07:00
parent 1692958f10
commit 96571b56e5
3 changed files with 22 additions and 11 deletions
--- a/server_test.go
+++ b/server_test.go
@@ -222,6 +222,7 @@ var _ = Describe("Server", func() {
 		})

 		It("closes sessions and the connection when Close is called", func() {
+			go serv.serve()
 			session, _ := newMockSession(nil, 0, 0, nil, nil, nil)
 			serv.sessions[1] = session
 			err := serv.Close()