From 616f9ac9bc50ee8f280a8e614ad589cda9751d11 Mon Sep 17 00:00:00 2001 From: Stephen Wakely Date: Wed, 4 Dec 2024 15:11:30 +0000 Subject: [PATCH] Fixed demultiplexer test Signed-off-by: Stephen Wakely --- cmd/dogstatsd/subcommands/start/command.go | 1 + .../demultiplexerimpl/demultiplexer_fake_sampler_mock.go | 2 ++ 2 files changed, 3 insertions(+) diff --git a/cmd/dogstatsd/subcommands/start/command.go b/cmd/dogstatsd/subcommands/start/command.go index 975bee035d48d..4a0442222bcd9 100644 --- a/cmd/dogstatsd/subcommands/start/command.go +++ b/cmd/dogstatsd/subcommands/start/command.go @@ -55,6 +55,7 @@ import ( "github.com/DataDog/datadog-agent/comp/metadata/runner" metadatarunnerimpl "github.com/DataDog/datadog-agent/comp/metadata/runner/runnerimpl" compressionfx "github.com/DataDog/datadog-agent/comp/serializer/compression/fx" + compressionfxfactory "github.com/DataDog/datadog-agent/comp/serializer/compression/fx-factory" "github.com/DataDog/datadog-agent/pkg/serializer" "github.com/DataDog/datadog-agent/pkg/status/health" "github.com/DataDog/datadog-agent/pkg/util" diff --git a/comp/aggregator/demultiplexer/demultiplexerimpl/demultiplexer_fake_sampler_mock.go b/comp/aggregator/demultiplexer/demultiplexerimpl/demultiplexer_fake_sampler_mock.go index 5f0c1db21316f..c5c4c90a38d80 100644 --- a/comp/aggregator/demultiplexer/demultiplexerimpl/demultiplexer_fake_sampler_mock.go +++ b/comp/aggregator/demultiplexer/demultiplexerimpl/demultiplexer_fake_sampler_mock.go @@ -17,6 +17,7 @@ import ( "github.com/DataDog/datadog-agent/comp/core/hostname" log "github.com/DataDog/datadog-agent/comp/core/log/def" compression "github.com/DataDog/datadog-agent/comp/serializer/compression/def" + compressionfx "github.com/DataDog/datadog-agent/comp/serializer/compression/fx-mock" "github.com/DataDog/datadog-agent/pkg/aggregator" "github.com/DataDog/datadog-agent/pkg/util/fxutil" ) @@ -25,6 +26,7 @@ import ( func FakeSamplerMockModule() fxutil.Module { return fxutil.Component( fx.Provide(newFakeSamplerMock), + compressionfx.MockModuleFactory(), fx.Provide(func(demux demultiplexerComp.FakeSamplerMock) aggregator.Demultiplexer { return demux }),