fix: server test

client.go

@@ -23,6 +23,8 @@ func (s *tcpClient) Connect() {
 		conn, err = net.Dial("tcp", s.address)
 	}
 
+	slog.Info("Connected to server", slog.String("address", conn.LocalAddr().String()))
+
 	defer conn.Close()
 
 	for {

server.go

@@ -36,6 +36,8 @@ func (s *tcpServer) Listen() (Server, error) {
 			continue
 		}
 
+		slog.Info("Client Connected", slog.String("host", conn.LocalAddr().String()))
+
 		s.Lock()
 		s.clients = append(s.clients, conn)
 		s.Unlock()

server_test.go

@@ -6,6 +6,7 @@ import (
 	"net"
 	"sync"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 )
@@ -37,11 +38,11 @@ func TestServer(t *testing.T) {
 		assert.Equal(t, "my-channel", request.Channel)
 		assert.Equal(t, "Hello there", request.Message.Extract())
 		wg.Done()
-
 	}()
 
+	time.Sleep(time.Millisecond)
+
 	msg := CreateMessage().SetBody("Hello there")
 	server.Publish("my-channel", msg)
 	wg.Wait()
-
 }