Fix test errors

newrelic · Aug 18, 2023 · 4994c11 · 4994c11
1 parent 1ec843d
commit 4994c11
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/pkg/metrics/process/sampler_darwin_test.go b/pkg/metrics/process/sampler_darwin_test.go
@@ -4,12 +4,15 @@ package process
 
 import (
 	"errors"
+	"testing"
+
+	"github.com/newrelic/infrastructure-agent/pkg/metrics"
+
 	"github.com/newrelic/infrastructure-agent/internal/agent/mocks"
 	"github.com/newrelic/infrastructure-agent/pkg/config"
 	"github.com/newrelic/infrastructure-agent/pkg/metrics/types"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
-	"testing"
 )
 
 func TestProcessSampler_Sample(t *testing.T) {
@@ -110,7 +113,7 @@ func TestProcessSampler_Sample_DockerDecorator(t *testing.T) {
 	harvester := &HarvesterMock{}
 	sampler := NewProcessSampler(ctx).(*processSampler)
 	sampler.harvest = harvester
-	sampler.containerSampler = &fakeContainerSampler{}
+	sampler.containerSamplers = []metrics.ContainerSampler{&fakeContainerSampler{}}
 
 	samples := []*types.ProcessSample{
 		{

diff --git a/pkg/metrics/process/sampler_linux_test.go b/pkg/metrics/process/sampler_linux_test.go
@@ -18,6 +18,7 @@ import (
 	"github.com/newrelic/infrastructure-agent/pkg/config"
 	"github.com/newrelic/infrastructure-agent/pkg/entity"
 	"github.com/newrelic/infrastructure-agent/pkg/entity/host"
+	"github.com/newrelic/infrastructure-agent/pkg/metrics"
 	"github.com/newrelic/infrastructure-agent/pkg/metrics/types"
 	"github.com/newrelic/infrastructure-agent/pkg/plugins/ids"
 	"github.com/newrelic/infrastructure-agent/pkg/sample"
@@ -41,7 +42,7 @@ func TestProcessSampler_DockerDecorator(t *testing.T) {
 			ProcessDisplayName: "Bye",
 		},
 	}}
-	ps.containerSampler = &fakeContainerSampler{}
+	ps.containerSamplers = []metrics.ContainerSampler{&fakeContainerSampler{}}
 
 	// When asking for the process samples
 	samples, err := ps.Sample()

diff --git a/pkg/metrics/procs_windows.go b/pkg/metrics/procs_windows.go
@@ -662,7 +662,7 @@ func (self *ProcsMonitor) Sample() (results sample.EventBatch, err error) {
 		}
 
 		var containerDecorators []ProcessDecorator
-		for _, containerSampler := range ps.containerSamplers {
+		for _, containerSampler := range self.containerSamplers {
 			if containerSampler.Enabled() {
 				decorator, err := containerSampler.NewDecorator()
 				if err != nil {