[chore]: use testify instead of testing.Fatal or testing.Error in exp…

…orter (#36835)

exporter/alertmanagerexporter/alertmanager_exporter_test.go

@@ -325,9 +325,7 @@ func TestAlertManagerPostAlert(t *testing.T) {
 
 	err = am.postAlert(context.Background(), alerts)
 	assert.NoError(t, err)
-	if mock.fooCalledSuccessfully == false {
-		t.Errorf("mock server wasn't called")
-	}
+	assert.True(t, mock.fooCalledSuccessfully, "mock server wasn't called")
 }
 
 func TestClientConfig(t *testing.T) {

exporter/clickhouseexporter/exporter_logs_test.go

@@ -32,10 +32,7 @@ func TestLogsExporter_New(t *testing.T) {
 	_ = func(want error) validate {
 		return func(t *testing.T, exporter *logsExporter, err error) {
 			require.Nil(t, exporter)
-			require.Error(t, err)
-			if !errors.Is(err, want) {
-				t.Fatalf("Expected error '%v', but got '%v'", want, err)
-			}
+			require.ErrorIs(t, err, want, "Expected error '%v', but got '%v'", want, err)
 		}
 	}
 

exporter/datadogexporter/internal/logs/sender_test.go

@@ -13,6 +13,7 @@ import (
 	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
 	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"go.opentelemetry.io/collector/config/confighttp"
 	"go.opentelemetry.io/collector/config/configtls"
 	"go.uber.org/zap/zaptest"
@@ -192,9 +193,7 @@ func TestSubmitLogs(t *testing.T) {
 			})
 			defer server.Close()
 			s := NewSender(server.URL, logger, confighttp.ClientConfig{Timeout: time.Second * 10, TLSSetting: configtls.ClientConfig{InsecureSkipVerify: true}}, true, "")
-			if err := s.SubmitLogs(context.Background(), tt.payload); err != nil {
-				t.Fatal(err)
-			}
+			require.NoError(t, s.SubmitLogs(context.Background(), tt.payload))
 			assert.Equal(t, calls, tt.numRequests)
 		})
 	}

exporter/datadogexporter/internal/metrics/sketches/sketches_test.go

@@ -74,14 +74,10 @@ func TestSketchSeriesListMarshal(t *testing.T) {
 	}
 
 	b, err := sl.Marshal()
-	if err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, err)
 
 	pl := new(gogen.SketchPayload)
-	if err := pl.Unmarshal(b); err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, pl.Unmarshal(b))
 
 	require.Len(t, pl.Sketches, len(sl))
 

exporter/datadogexporter/traces_exporter_test.go

@@ -116,11 +116,10 @@ func TestTracesSource(t *testing.T) {
 			return
 		}
 		buf := new(bytes.Buffer)
-		if _, err := buf.ReadFrom(r.Body); err != nil {
-			t.Fatalf("Metrics server handler error: %v", err)
-		}
+		_, err := buf.ReadFrom(r.Body)
+		assert.NoError(t, err, "Metrics server handler error: %v", err)
 		reqs <- buf.Bytes()
-		_, err := w.Write([]byte("{\"status\": \"ok\"}"))
+		_, err = w.Write([]byte("{\"status\": \"ok\"}"))
 		assert.NoError(t, err)
 	}))
 	defer metricsServer.Close()

exporter/logzioexporter/jsonlog_test.go

@@ -111,10 +111,6 @@ func TestSetTimeStamp(t *testing.T) {
 	requests := strings.Split(string(decoded), "\n")
 	require.NoError(t, json.Unmarshal([]byte(requests[0]), &jsonLog))
 	require.NoError(t, json.Unmarshal([]byte(requests[1]), &jsonLogNoTimestamp))
-	if jsonLogNoTimestamp["@timestamp"] != nil {
-		t.Fatalf("did not expect @timestamp")
-	}
-	if jsonLog["@timestamp"] == nil {
-		t.Fatalf("@timestamp does not exist")
-	}
+	require.Nil(t, jsonLogNoTimestamp["@timestamp"], "did not expect @timestamp")
+	require.NotNil(t, jsonLog["@timestamp"], "@timestamp does not exist")
 }

exporter/logzioexporter/logziospan_test.go

@@ -10,54 +10,41 @@ import (
 	"testing"
 
 	"github.com/jaegertracing/jaeger/model"
+	"github.com/stretchr/testify/require"
 )
 
 func TestTransformToLogzioSpanBytes(tester *testing.T) {
 	inStr, err := os.ReadFile("./testdata/span.json")
-	if err != nil {
-		tester.Fatalf("error opening sample span file %s", err.Error())
-	}
+	require.NoError(tester, err, "error opening sample span file")
 
 	var span model.Span
 	err = json.Unmarshal(inStr, &span)
 	if err != nil {
 		fmt.Println("json.Unmarshal")
 	}
 	newSpan, err := transformToLogzioSpanBytes(&span)
-	if err != nil {
-		tester.Fatal(err.Error())
-	}
+	require.NoError(tester, err)
 	m := make(map[string]any)
 	err = json.Unmarshal(newSpan, &m)
-	if err != nil {
-		tester.Fatal(err.Error())
-	}
+	require.NoError(tester, err)
 	if _, ok := m["JaegerTag"]; !ok {
 		tester.Error("error converting span to logzioSpan, JaegerTag is not found")
 	}
 }
 
 func TestTransformToDbModelSpan(tester *testing.T) {
 	inStr, err := os.ReadFile("./testdata/span.json")
-	if err != nil {
-		tester.Fatalf("error opening sample span file %s", err.Error())
-	}
+	require.NoError(tester, err, "error opening sample span file")
 	var span model.Span
 	err = json.Unmarshal(inStr, &span)
 	if err != nil {
 		fmt.Println("json.Unmarshal")
 	}
 	newSpan, err := transformToLogzioSpanBytes(&span)
-	if err != nil {
-		tester.Fatal(err.Error())
-	}
+	require.NoError(tester, err)
 	var testLogzioSpan logzioSpan
 	err = json.Unmarshal(newSpan, &testLogzioSpan)
-	if err != nil {
-		tester.Fatal(err.Error())
-	}
+	require.NoError(tester, err)
 	dbModelSpan := testLogzioSpan.transformToDbModelSpan()
-	if len(dbModelSpan.References) != 3 {
-		tester.Fatalf("Error converting logzio span to dbmodel span")
-	}
+	require.Len(tester, dbModelSpan.References, 3, "Error converting logzio span to dbmodel span")
 }

exporter/mezmoexporter/exporter_test.go

@@ -122,9 +122,7 @@ type (
 func createHTTPServer(params *testServerParams) testServer {
 	httpServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		body, err := io.ReadAll(r.Body)
-		if err != nil {
-			params.t.Fatal(err)
-		}
+		assert.NoError(params.t, err)
 
 		var logBody mezmoLogBody
 		if err = json.Unmarshal(body, &logBody); err != nil {

exporter/otelarrowexporter/internal/arrow/exporter_test.go

@@ -922,9 +922,7 @@ func benchmarkPrioritizer(b *testing.B, numStreams int, pname PrioritizerName) {
 
 	wg.Add(1)
 	defer func() {
-		if err := tc.exporter.Shutdown(bg); err != nil {
-			b.Errorf("shutdown failed: %v", err)
-		}
+		assert.NoError(b, tc.exporter.Shutdown(bg), "shutdown failed")
 		wg.Done()
 		wg.Wait()
 	}()

exporter/prometheusexporter/end_to_end_test.go

@@ -48,9 +48,7 @@ func TestEndToEndSummarySupport(t *testing.T) {
 	defer dropWizardServer.Close()
 
 	srvURL, err := url.Parse(dropWizardServer.URL)
-	if err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, err)
 
 	ctx, cancel := context.WithCancel(context.Background())
 	defer cancel()
@@ -67,12 +65,8 @@ func TestEndToEndSummarySupport(t *testing.T) {
 	exporterFactory := NewFactory()
 	set := exportertest.NewNopSettings()
 	exporter, err := exporterFactory.CreateMetrics(ctx, set, exporterCfg)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err = exporter.Start(ctx, nil); err != nil {
-		t.Fatalf("Failed to start the Prometheus exporter: %v", err)
-	}
+	require.NoError(t, err)
+	require.NoError(t, exporter.Start(ctx, nil), "Failed to start the Prometheus exporter")
 	t.Cleanup(func() { require.NoError(t, exporter.Shutdown(ctx)) })
 
 	// 3. Create the Prometheus receiver scraping from the DropWizard mock server and
@@ -89,9 +83,7 @@ func TestEndToEndSummarySupport(t *testing.T) {
                 - targets: ['%s']
         `, srvURL.Host))
 	receiverConfig := new(prometheusreceiver.PromConfig)
-	if err = yaml.Unmarshal(yamlConfig, receiverConfig); err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, yaml.Unmarshal(yamlConfig, receiverConfig))
 
 	receiverFactory := prometheusreceiver.NewFactory()
 	receiverCreateSet := receivertest.NewNopSettings()
@@ -100,26 +92,18 @@ func TestEndToEndSummarySupport(t *testing.T) {
 	}
 	// 3.5 Create the Prometheus receiver and pass in the previously created Prometheus exporter.
 	prometheusReceiver, err := receiverFactory.CreateMetrics(ctx, receiverCreateSet, rcvCfg, exporter)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err = prometheusReceiver.Start(ctx, nil); err != nil {
-		t.Fatalf("Failed to start the Prometheus receiver: %v", err)
-	}
+	require.NoError(t, err)
+	require.NoError(t, prometheusReceiver.Start(ctx, nil), "Failed to start the Prometheus receiver")
 	t.Cleanup(func() { require.NoError(t, prometheusReceiver.Shutdown(ctx)) })
 
 	// 4. Scrape from the Prometheus receiver to ensure that we export summary metrics
 	wg.Wait()
 
 	res, err := http.Get("http://" + exporterCfg.Endpoint + "/metrics")
-	if err != nil {
-		t.Fatalf("Failed to scrape from the exporter: %v", err)
-	}
+	require.NoError(t, err, "Failed to scrape from the exporter")
 	prometheusExporterScrape, err := io.ReadAll(res.Body)
 	res.Body.Close()
-	if err != nil {
-		t.Fatal(err)
-	}
+	require.NoError(t, err)
 
 	// 5. Verify that we have the summary metrics and that their values make sense.
 	wantLineRegexps := []string{
@@ -171,9 +155,7 @@ func TestEndToEndSummarySupport(t *testing.T) {
 	// After this replacement, there should ONLY be newlines present.
 	prometheusExporterScrape = bytes.ReplaceAll(prometheusExporterScrape, []byte("\n"), []byte(""))
 	// Now assert that NO output was left over.
-	if len(prometheusExporterScrape) != 0 {
-		t.Fatalf("Left-over unmatched Prometheus scrape content: %q\n", prometheusExporterScrape)
-	}
+	require.Empty(t, prometheusExporterScrape, "Left-over unmatched Prometheus scrape content: %q\n", prometheusExporterScrape)
 }
 
 // the following triggers G101: Potential hardcoded credentials