From 7cc3964a4dcf9bfb33e48c7a8831c580aa294b38 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Wed, 29 May 2024 16:11:58 +0200
Subject: [PATCH] Use mux for http server

Without it can not start multiple http
server instances for unit tests
---
 relay/server/listener/ws/listener.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/relay/server/listener/ws/listener.go b/relay/server/listener/ws/listener.go
index d93dfe0c3..b53e97505 100644
--- a/relay/server/listener/ws/listener.go
+++ b/relay/server/listener/ws/listener.go
@@ -36,10 +36,12 @@ func NewListener(address string) listener.Listener {
 // Listen todo: prevent multiple call
 func (l *Listener) Listen(acceptFn func(conn net.Conn)) error {
 	l.acceptFn = acceptFn
-	http.HandleFunc("/", l.onAccept)
+	mux := http.NewServeMux()
+	mux.HandleFunc("/", l.onAccept)
 
 	l.server = &http.Server{
-		Addr: l.address,
+		Addr:    l.address,
+		Handler: mux,
 	}
 
 	log.Infof("WS server is listening on address: %s", l.address)