Fix flaky test TestBufferExporter/Shutdown/ContextCancelled (#5261)

open-telemetry · Apr 25, 2024 · 19ee6d4 · 19ee6d4
1 parent fd6d4db
commit 19ee6d4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/sdk/log/exporter_test.go b/sdk/log/exporter_test.go
@@ -378,6 +378,13 @@ func TestBufferExporter(t *testing.T) {
 		})
 
 		t.Run("ContextCancelled", func(t *testing.T) {
+			// Discard error logs.
+			defer func(orig otel.ErrorHandler) {
+				otel.SetErrorHandler(orig)
+			}(otel.GetErrorHandler())
+			handler := otel.ErrorHandlerFunc(func(err error) {})
+			otel.SetErrorHandler(handler)
+
 			exp := newTestExporter(assert.AnError)
 			t.Cleanup(exp.Stop)
 
@@ -386,6 +393,9 @@ func TestBufferExporter(t *testing.T) {
 			t.Cleanup(func() { close(trigger) })
 			e := newBufferExporter(exp, 1)
 
+			// Make sure there is something to flush.
+			require.True(t, e.EnqueueExport(make([]Record, 1)))
+
 			ctx, cancel := context.WithCancel(context.Background())
 			cancel()