From 82e210a8876724786e2379f0cac00a532f542ede Mon Sep 17 00:00:00 2001 From: Eloy Toro Date: Mon, 3 Feb 2025 15:43:13 +0100 Subject: [PATCH] add tests --- .../datadogagent/feature/apm/feature_test.go | 40 +++++++++++++++++++ 1 file changed, 40 insertions(+) diff --git a/internal/controller/datadogagent/feature/apm/feature_test.go b/internal/controller/datadogagent/feature/apm/feature_test.go index ea0d3673c..a33e9eef4 100644 --- a/internal/controller/datadogagent/feature/apm/feature_test.go +++ b/internal/controller/datadogagent/feature/apm/feature_test.go @@ -363,6 +363,17 @@ func TestAPMFeature(t *testing.T) { } }, }, + { + Name: "error tracking standalone enabled", + DDA: v2alpha1test.NewDatadogAgentBuilder(). + WithAPMEnabled(true). + WithAPMHostPortEnabled(false, apiutils.NewInt32Pointer(8126)). + WithAPMUDSEnabled(true, apmSocketHostPath). + WithErrorTrackingStandalone(true). + Build(), + WantConfigure: true, + Agent: testAgentErrorTrackingStandalone(), + }, } tests.Run(t, buildAPMFeature) @@ -444,6 +455,35 @@ func testAgentHostPortOnly() *test.ComponentTest { ) } +func testAgentErrorTrackingStandalone() *test.ComponentTest { + return test.NewDefaultComponentTest().WithWantFunc( + func(t testing.TB, mgrInterface feature.PodTemplateManagers) { + mgr := mgrInterface.(*fake.PodTemplateManagers) + + agentEnvs := mgr.EnvVarMgr.EnvVarsByC[apicommon.TraceAgentContainerName] + expectedAgentEnvs := []*corev1.EnvVar{ + { + Name: apicommon.DDAPMEnabled, + Value: "true", + }, + { + Name: apicommon.DDAPMReceiverSocket, + Value: apmSocketLocalPath, + }, + { + Name: apicommon.DDAPMErrorTrackingStandaloneEnabled, + Value: "true", + }, + } + assert.True( + t, + apiutils.IsEqualStruct(agentEnvs, expectedAgentEnvs), + "Trace Agent ENVs \ndiff = %s", cmp.Diff(agentEnvs, expectedAgentEnvs), + ) + }, + ) +} + func testAgentUDSOnly(agentContainerName apicommon.AgentContainerName) *test.ComponentTest { return test.NewDefaultComponentTest().WithWantFunc( func(t testing.TB, mgrInterface feature.PodTemplateManagers) {