discovery: address follow-up review feedback

Author: martavicentenavarro

pkg/discovery/module/comm_test.go

@@ -3,7 +3,7 @@
 // This product includes software developed at Datadog (https://www.datadoghq.com/).
 // Copyright 2024-present Datadog, Inc.
 
-//go:build test && linux_bpf
+//go:build test && linux_bpf && (!dd_discovery_rust || !cgo)
 
 package module
 

pkg/discovery/module/discovery_suite_test.go

@@ -0,0 +1,23 @@
+// Unless explicitly stated otherwise all files in this repository are licensed
+// under the Apache License Version 2.0.
+// This product includes software developed at Datadog (https://www.datadoghq.com/).
+// Copyright 2024-present Datadog, Inc.
+
+//go:build test && linux_bpf
+
+package module
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+)
+
+func TestDiscovery(t *testing.T) {
+	t.Run("module", func(t *testing.T) {
+		suite.Run(t, &discoveryTestSuite{setupModule: setupDiscoveryModule})
+	})
+	t.Run("system-probe-lite", func(t *testing.T) {
+		suite.Run(t, &discoveryTestSuite{setupModule: setupRustDiscoveryModule})
+	})
+}

pkg/discovery/module/impl_services_rust_linux_test.go

@@ -5,7 +5,7 @@
 
 // This doesn't need BPF, but it's built with this tag to only run with
 // system-probe tests.
-//go:build test && linux_bpf && (!dd_discovery_rust || !cgo)
+//go:build test && linux_bpf
 
 package module
 

pkg/discovery/module/impl_services_test.go

@@ -188,8 +188,8 @@ rust_test(
     name = "fallback_integration_test",
     srcs = ["tests/fallback_integration_test.rs"],
     data = [
-        "//pkg/discovery/module/rust/testdata:mock_system_probe",
         ":system-probe-lite",
+        "//pkg/discovery/module/rust/testdata:mock_system_probe",
     ],
     edition = "2024",
     # Set compile-time environment variable pointing to system-probe-lite binary

pkg/discovery/module/rust/BUILD.bazel

@@ -178,15 +178,6 @@ func (s *discoveryTestSuite) SetupTest() {
 	s.discovery = s.setupModule(s.T())
 }
 
-func TestDiscovery(t *testing.T) {
-	t.Run("go", func(t *testing.T) {
-		suite.Run(t, &discoveryTestSuite{setupModule: setupGoDiscoveryModule})
-	})
-	t.Run("rust", func(t *testing.T) {
-		suite.Run(t, &discoveryTestSuite{setupModule: setupRustDiscoveryModule})
-	})
-}
-
 func newDiscovery() *discovery {
 	mod, err := NewDiscoveryModule(nil, module.FactoryDependencies{})
 	if err != nil {