test only relay credentials push updates

netbirdio · Jan 8, 2025 · 70b0925 · 70b0925
1 parent 3d7003b
commit 70b0925
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/management/server/token_mgr_test.go b/management/server/token_mgr_test.go
@@ -133,11 +133,14 @@ loop:
 			}
 		}
 		if relay := update.Update.GetWiretrusteeConfig().GetRelay(); relay != nil {
-			relayUpdates++
-			if relayUpdates == 1 {
-				firstRelayUpdate = relay
-			} else {
-				secondRelayUpdate = relay
+			// avoid updating on turn updates since they also send relay credentials
+			if update.Update.GetWiretrusteeConfig().GetTurns() == nil {
+				relayUpdates++
+				if relayUpdates == 1 {
+					firstRelayUpdate = relay
+				} else {
+					secondRelayUpdate = relay
+				}
 			}
 		}
 	}