Add stop and formatting

grafana · Oct 17, 2024 · b0c9409 · b0c9409
1 parent 8856717
commit b0c9409
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/internal/component/prometheus/operator/common/component.go b/internal/component/prometheus/operator/common/component.go
@@ -71,12 +71,14 @@ type cancelHandler struct {
 func (ch *cancelHandler) Set(cncl context.CancelFunc) {
 	ch.mut.Lock()
 	defer ch.mut.Unlock()
+
 	ch.cancel = cncl
 }
 
 func (ch *cancelHandler) Cancel() {
 	ch.mut.Lock()
 	defer ch.mut.Unlock()
+
 	if ch.alreadyTrigger {
 		return
 	}

diff --git a/internal/component/prometheus/operator/common/component_test.go b/internal/component/prometheus/operator/common/component_test.go
@@ -109,6 +109,7 @@ func TestRunExit(t *testing.T) {
 	// Stop the component.
 	// It shouldn't stop immediately, because the CRD Manager is hung.
 	cancelFunc()
+	time.Sleep(5 * time.Second)
 	<-stopRun
 
 	require.Eventually(t, func() bool {