Fix sporadic in probe tests

internal/probe/agent_test.go

@@ -4,33 +4,16 @@
 package probe_test
 
 import (
-	"context"
 	"encoding/json"
 	"fmt"
 	"net/http"
-	"time"
-
-	"github.com/ironcore-dev/metal-operator/internal/probe"
 
 	"github.com/ironcore-dev/metal-operator/internal/api/registry"
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
 var _ = Describe("ProbeAgent", func() {
-	BeforeEach(func() {
-		By("Starting the probe agent")
-		ctx, cancel := context.WithCancel(context.Background())
-		DeferCleanup(cancel)
-
-		// Initialize your probe agent
-		probeAgent = probe.NewAgent(systemUUID, registryURL, 100*time.Millisecond)
-		go func() {
-			defer GinkgoRecover()
-			Expect(probeAgent.Start(ctx)).To(Succeed(), "failed to start probe agent")
-		}()
-	})
-
 	It("should ensure the correct endpoints have been registered", func() {
 		By("performing a GET request to the /systems/{uuid} endpoint")
 		var resp *http.Response

internal/probe/probe_suite_test.go

@@ -5,6 +5,7 @@ package probe_test
 
 import (
 	"context"
+	"net/http"
 	"testing"
 	"time"
 
@@ -43,6 +44,11 @@ var _ = BeforeSuite(func() {
 		Expect(registryServer.Start(ctx)).To(Succeed(), "failed to start registry agent")
 	}()
 
+	Eventually(func() error {
+		_, err := http.Get(registryAddr)
+		return err
+	}).Should(Succeed())
+
 	// Initialize your probe server
 	probeAgent = probe.NewAgent(systemUUID, registryURL, 100*time.Millisecond)
 	go func() {

internal/registry/registry_suite_test.go

@@ -5,6 +5,7 @@ package registry_test
 
 import (
 	"context"
+	"net/http"
 	"testing"
 
 	"github.com/ironcore-dev/metal-operator/internal/registry"
@@ -36,4 +37,9 @@ var _ = BeforeSuite(func() {
 		defer GinkgoRecover()
 		Expect(server.Start(ctx)).To(Succeed(), "failed to start registry server")
 	}()
+
+	Eventually(func() error {
+		_, err := http.Get(testServerURL)
+		return err
+	}).Should(Succeed())
 })