Use t.Cleanup to make sure shutdown is called.

open-telemetry · Jan 4, 2024 · 16246d7 · 16246d7
1 parent cb5d261
commit 16246d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exporter/exporterhelper/queue_sender_test.go b/exporter/exporterhelper/queue_sender_test.go
@@ -325,10 +325,10 @@ func TestQueuedRetryPersistentEnabled_NoDataLossOnShutdown(t *testing.T) {
 		newNoopObsrepSender, WithRetry(rCfg), WithQueue(qCfg))
 	require.NoError(t, err)
 	require.NoError(t, be.Start(context.Background(), host))
+	t.Cleanup(func() { require.NoError(t, be.Shutdown(context.Background())) })
 
 	// wait for the item to be consumed from the queue
 	replacedReq.checkNumRequests(t, 1)
-	require.NoError(t, be.Shutdown(context.Background()))
 }
 
 func TestQueueSenderNoStartShutdown(t *testing.T) {