From c7a188048a27a6b85d27a0cca6ce76fc721fc4d0 Mon Sep 17 00:00:00 2001
From: Kevin Franklin Kim <kevin.kim@bestbytes.com>
Date: Thu, 3 Feb 2022 09:02:34 +0100
Subject: [PATCH] test: add timeout

---
 server_test.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/server_test.go b/server_test.go
index eafcb31..f3d0c47 100644
--- a/server_test.go
+++ b/server_test.go
@@ -33,6 +33,7 @@ func (s *KeelTestSuite) SetupSuite() {
 
 // BeforeTest hook
 func (s *KeelTestSuite) BeforeTest(suiteName, testName string) {
+	s.l = keeltest.NewLogger(s.T()).Zap()
 	s.mux = http.NewServeMux()
 	s.mux.HandleFunc("/ok", func(w http.ResponseWriter, r *http.Request) {
 		w.WriteHeader(http.StatusOK)
@@ -65,6 +66,7 @@ func (s *KeelTestSuite) BeforeTest(suiteName, testName string) {
 // AfterTest hook
 func (s *KeelTestSuite) AfterTest(suiteName, testName string) {
 	s.cancel()
+	time.Sleep(time.Second * 3)
 }
 
 // TearDownSuite hook
@@ -194,7 +196,6 @@ func (s *KeelTestSuite) runServer() {
 	go func(waitChan chan string) {
 		waitChan <- "finished"
 		s.svr.Run()
-		time.Sleep(time.Second * 3)
 	}(waitChan)
 	l.Debug("waiting for server process to start")
 	<-waitChan