diff --git a/tests/Agent/IntegrationTests/IntegrationTests/AwsLambda/AwsLambdaSmokeTest.cs b/tests/Agent/IntegrationTests/IntegrationTests/AwsLambda/AwsLambdaSmokeTest.cs index 7765c02da..5e0398289 100644 --- a/tests/Agent/IntegrationTests/IntegrationTests/AwsLambda/AwsLambdaSmokeTest.cs +++ b/tests/Agent/IntegrationTests/IntegrationTests/AwsLambda/AwsLambdaSmokeTest.cs @@ -23,6 +23,14 @@ protected AwsLambdaSmokeTestBase(string expectedTransactionName, T fixture, ITes _fixture = fixture; _fixture.TestLogger = output; + + _fixture.AdditionalSetupConfiguration = () => + { + var configPath = fixture.DestinationNewRelicConfigFilePath; + var configModifier = new NewRelicConfigModifier(configPath); + configModifier.ForceTransactionTraces(); + }; + _fixture.Actions( exerciseApplication: () => { @@ -51,8 +59,8 @@ public void Test() () => Assert.NotNull(serverlessPayload.Telemetry.MetricsPayload), () => Assert.NotNull(serverlessPayload.Telemetry.TransactionEventsPayload), () => Assert.NotNull(serverlessPayload.Telemetry.SpanEventsPayload), + () => Assert.NotNull(serverlessPayload.Telemetry.TransactionTracePayload), () => Assert.Null(serverlessPayload.Telemetry.SqlTracePayload), - () => Assert.Null(serverlessPayload.Telemetry.TransactionTracePayload), () => Assert.Null(serverlessPayload.Telemetry.ErrorTracePayload), () => Assert.Null(serverlessPayload.Telemetry.ErrorEventsPayload), () => Assert.Equal(_expectedTransactionName, serverlessPayload.Telemetry.TransactionEventsPayload.TransactionEvents.Single().IntrinsicAttributes["name"]) diff --git a/tests/Agent/IntegrationTests/IntegrationTests/RemoteServiceFixtures/AwsLambda/LambdaTestToolFixture.cs b/tests/Agent/IntegrationTests/IntegrationTests/RemoteServiceFixtures/AwsLambda/LambdaTestToolFixture.cs index 1a583b15b..4cd13e1ba 100644 --- a/tests/Agent/IntegrationTests/IntegrationTests/RemoteServiceFixtures/AwsLambda/LambdaTestToolFixture.cs +++ b/tests/Agent/IntegrationTests/IntegrationTests/RemoteServiceFixtures/AwsLambda/LambdaTestToolFixture.cs @@ -14,6 +14,7 @@ namespace NewRelic.Agent.IntegrationTests.RemoteServiceFixtures.AwsLambda public class LambdaTestToolFixture : RemoteApplicationFixture { public DotnetTool LambdaTestTool { get; set; } + public Action AdditionalSetupConfiguration { get; set; } public LambdaTestToolFixture(RemoteApplication remoteApplication, string newRelicLambdaHandler, string lambdaHandler, string lambdaName, string lambdaVersion, string lambdaExecutionEnvironment) : base(remoteApplication) { @@ -41,6 +42,8 @@ public LambdaTestToolFixture(RemoteApplication remoteApplication, string newReli // Finest level logs are necessary to read the uncompressed payloads from the agent logs remoteApplication.NewRelicConfig.SetLogLevel("finest"); + AdditionalSetupConfiguration?.Invoke(); + if (LambdaTestTool.IsRunning) { return;