adjust tests

netbirdio · Jun 13, 2024 · fde1379 · fde1379
1 parent 2251486
commit fde1379
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/client/internal/routemanager/manager_test.go b/client/internal/routemanager/manager_test.go
@@ -435,7 +435,11 @@ func TestManagerUpdateRoutes(t *testing.T) {
 			_, _, err = routeManager.UpdateRoutes(testCase.inputSerial+uint64(len(testCase.inputInitRoutes)), testCase.inputRoutes)
 			require.NoError(t, err, "should update routes")
 
-			require.Len(t, routeManager.clientNetworks, testCase.clientNetworkWatchersExpectedAllowed, "client networks size should match")
+			expectedWatchers := testCase.clientNetworkWatchersExpected
+			if testCase.clientNetworkWatchersExpectedAllowed != 0 {
+				expectedWatchers = testCase.clientNetworkWatchersExpectedAllowed
+			}
+			require.Len(t, routeManager.clientNetworks, expectedWatchers, "client networks size should match")
 
 			if runtime.GOOS == "linux" && routeManager.serverRouter != nil {
 				sr := routeManager.serverRouter.(*defaultServerRouter)