Fix test logic

ksysoev · May 19, 2024 · 7bc2645 · 7bc2645
1 parent f2e16c2
commit 7bc2645
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/server/server.go b/server/server.go
@@ -21,11 +21,11 @@ var ErrServerAlreadyRunning = fmt.Errorf("server is already running")
 
 type Server struct {
 	baseCtx  context.Context
+	listener net.Listener
 	mutex    *sync.Mutex
 	handler  *http.Server
 	addr     string
 	channels []wasabi.Channel
-	listener net.Listener
 }
 
 type Option func(*Server)

diff --git a/server/server_test.go b/server/server_test.go
@@ -2,7 +2,6 @@ package server
 
 import (
 	"context"
-	"fmt"
 	"net/http"
 	"testing"
 	"time"
@@ -256,7 +255,8 @@ func TestServer_Addr(t *testing.T) {
 		t.Error("Expected server to start")
 	}
 
-	fmt.Println(server.listener)
+	// Wait for the server to fully start
+	time.Sleep(1 * time.Millisecond)
 
 	addr := server.Addr()