fix exporter unittest occasional error

Petrie · Jul 3, 2022 · ff2fdbe · ff2fdbe
1 parent c2dc940
commit ff2fdbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exporters/otlp/otlptrace/internal/otlptracetest/client.go b/exporters/otlp/otlptrace/internal/otlptracetest/client.go
@@ -69,7 +69,7 @@ func testClientStopHonorsTimeout(t *testing.T, client otlptrace.Client) {
 	defer cancel()
 	<-ctx.Done()
 
-	if err := e.Shutdown(ctx); !errors.Is(err, context.DeadlineExceeded) {
+	if err := e.Shutdown(ctx); !errors.Is(err, context.DeadlineExceeded) || err == nil {
 		t.Errorf("expected context DeadlineExceeded error, got %v", err)
 	}
 }
@@ -88,7 +88,7 @@ func testClientStopHonorsCancel(t *testing.T, client otlptrace.Client) {
 	ctx, cancel := context.WithCancel(context.Background())
 	cancel()
 
-	if err := e.Shutdown(ctx); !errors.Is(err, context.Canceled) {
+	if err := e.Shutdown(ctx); !errors.Is(err, context.Canceled) || err == nil {
 		t.Errorf("expected context canceled error, got %v", err)
 	}
 }