app/forkjoin: fix race in test (#2010)

Fixes race in `TestPanic` test. category: misc ticket: none
ObolNetwork · Apr 11, 2023 · 46c6bb4 · 46c6bb4
1 parent 3ee9463
commit 46c6bb4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/app/forkjoin/forkjoin_test.go b/app/forkjoin/forkjoin_test.go
@@ -109,10 +109,9 @@ func TestForkJoin(t *testing.T) {
 func TestPanic(t *testing.T) {
 	defer goleak.VerifyNone(t)
 
-	fork, join, cancel := forkjoin.New[int, int](context.Background(), nil)
-	defer cancel()
-
+	fork, join, cancel := forkjoin.New[int, int](context.Background(), nil, forkjoin.WithWaitOnCancel())
 	join()
+	cancel()
 
 	// Calling fork after join panics
 	require.Panics(t, func() {