fix: test

VihasMakwana · Jul 23, 2024 · 091fff8 · 091fff8
1 parent fd6d312
commit 091fff8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/metricbeat/module/system/process/process_test.go b/metricbeat/module/system/process/process_test.go
@@ -20,6 +20,7 @@
 package process
 
 import (
+	"errors"
 	"os"
 	"testing"
 	"time"
@@ -38,15 +39,15 @@ func TestFetch(t *testing.T) {
 	f := mbtest.NewReportingMetricSetV2Error(t, getConfig())
 	events, errs := mbtest.ReportingFetchV2Error(f)
 	for _, err := range errs {
-		assert.True(t, process.CanDegrade(err))
+		assert.True(t, errors.Is(err, process.NonFatalErr{}))
 	}
 	assert.NotEmpty(t, events)
 
 	time.Sleep(2 * time.Second)
 
 	events, errs = mbtest.ReportingFetchV2Error(f)
 	for _, err := range errs {
-		assert.True(t, process.CanDegrade(err))
+		assert.True(t, errors.Is(err, process.NonFatalErr{}))
 	}
 	assert.NotEmpty(t, events)
 

diff --git a/metricbeat/module/system/process_summary/process_summary_test.go b/metricbeat/module/system/process_summary/process_summary_test.go
@@ -20,6 +20,7 @@
 package process_summary
 
 import (
+	"errors"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -47,7 +48,7 @@ func TestFetch(t *testing.T) {
 	events, errs := mbtest.ReportingFetchV2Error(f)
 
 	for _, err := range errs {
-		assert.True(t, process.CanDegrade(err))
+		assert.True(t, errors.Is(err, process.NonFatalErr{}))
 	}
 	require.NotEmpty(t, events)
 	event := events[0].BeatEvent("system", "process_summary").Fields
@@ -65,7 +66,7 @@ func TestStateNames(t *testing.T) {
 	events, errs := mbtest.ReportingFetchV2Error(f)
 
 	for _, err := range errs {
-		assert.True(t, process.CanDegrade(err))
+		assert.True(t, errors.Is(err, process.NonFatalErr{}))
 	}
 	require.NotEmpty(t, events)
 	event := events[0].BeatEvent("system", "process_summary").Fields