Using slices.ContainsFunc in HasConnection

stackrox · Nov 14, 2024 · 9b4f72e · 9b4f72e
1 parent 4583ed4
commit 9b4f72e
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/integration-tests/pkg/mock_sensor/server.go b/integration-tests/pkg/mock_sensor/server.go
@@ -5,6 +5,7 @@ import (
 	"log"
 	"net"
 	"os"
+	"slices"
 	"strings"
 	"sync"
 	"time"
@@ -166,11 +167,9 @@ func (m *MockSensor) HasConnection(containerID string, conn types.NetworkInfo) b
 	defer m.networkMutex.Unlock()
 
 	if conns, ok := m.connections[containerID]; ok {
-		for _, connection := range conns {
-			if connection.Equal(conn) {
-				return true
-			}
-		}
+		return slices.ContainsFunc(conns, func(c types.NetworkInfo) bool {
+			return c.Equal(conn)
+		})
 	}
 
 	return false