From ef5027ab2a8f57da989265b54720727fa3e3d3ad Mon Sep 17 00:00:00 2001
From: Viktor Liu <viktor@netbird.io>
Date: Mon, 2 Jun 2025 19:43:07 +0200
Subject: [PATCH] Fix test

---
 client/internal/routemanager/manager_test.go  | 5 ++---
 client/internal/routemanager/server/server.go | 6 ++++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/client/internal/routemanager/manager_test.go b/client/internal/routemanager/manager_test.go
index 318ef5ae5..680bd813f 100644
--- a/client/internal/routemanager/manager_test.go
+++ b/client/internal/routemanager/manager_test.go
@@ -4,7 +4,6 @@ import (
 	"context"
 	"fmt"
 	"net/netip"
-	"runtime"
 	"testing"
 
 	"github.com/pion/transport/v3/stdnet"
@@ -454,8 +453,8 @@ func TestManagerUpdateRoutes(t *testing.T) {
 			}
 			require.Len(t, routeManager.clientNetworks, expectedWatchers, "client networks size should match")
 
-			if runtime.GOOS == "linux" && routeManager.serverRouter != nil {
-				require.Len(t, routeManager.serverRouter.routes, testCase.serverRoutesExpected, "server networks size should match")
+			if routeManager.serverRouter != nil {
+				require.Equal(t, testCase.serverRoutesExpected, routeManager.serverRouter.RoutesCount(), "server networks size should match")
 			}
 		})
 	}
diff --git a/client/internal/routemanager/server/server.go b/client/internal/routemanager/server/server.go
index 09149feb9..e674c80cd 100644
--- a/client/internal/routemanager/server/server.go
+++ b/client/internal/routemanager/server/server.go
@@ -133,6 +133,12 @@ func (r *Router) CleanUp() {
 	r.statusRecorder.CleanLocalPeerStateRoutes()
 }
 
+func (r *Router) RoutesCount() int {
+	r.mux.Lock()
+	defer r.mux.Unlock()
+	return len(r.routes)
+}
+
 func routeToRouterPair(route *route.Route, useNewDNSRoute bool) firewall.RouterPair {
 	source := getDefaultPrefix(route.Network)
 	destination := firewall.Network{}