test: fix flaky CSP and Lambda integration tests (#3505)

- CSP tests: Replace GetTransactionSamples().FirstOrDefault() with
  TryGetTransactionSample(transactionName) to target the GetWithData
  transaction. Previously, if the ThrowException trace was harvested
  first, the test failed asserting response.status 200 vs actual 500.

- Lambda tests: Add initial 5s delay and increase warmup retry count
  from 3 to 5 in LambdaTestToolFixture.WarmUpTestTool(), giving the
  mock Lambda test tool more time to start on loaded CI runners.

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: tippmar-nr

tests/Agent/IntegrationTests/IntegrationTests/CSP/AspNetCoreLocalHSMEnabledAndServerSideHSMEnabledTests.cs

@@ -3,7 +3,6 @@
 
 
 using System.Collections.Generic;
-using System.Linq;
 using NewRelic.Agent.IntegrationTestHelpers;
 using NewRelic.Agent.Tests.TestSerializationHelpers.Models;
 using NewRelic.Testing.Assertions;
@@ -84,7 +83,7 @@ public void Test()
         };
 
 
-        var transactionSample = _fixture.AgentLog.GetTransactionSamples().FirstOrDefault();
+        var transactionSample = _fixture.AgentLog.TryGetTransactionSample("WebTransaction/MVC/Home/Query/{data}");
         var getDataTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/Home/Query/{data}");
         var getExceptionTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/Home/ThrowException");
 
@@ -104,4 +103,4 @@ public void Test()
             () => Assertions.TransactionEventHasAttributes(expectedErrorTransactionEventAttributes, TransactionEventAttributeType.Intrinsic, getExceptionTransactionEvent)
         );
     }
-}
+}

tests/Agent/IntegrationTests/IntegrationTests/CSP/HighSecurityModeDisabled.cs

@@ -74,7 +74,7 @@ public void Test()
 
         var displayHost = _fixture.AgentLog.GetConnectData().DisplayHost;
         var getDataTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/DefaultController/Query");
-        var transactionSample = _fixture.AgentLog.GetTransactionSamples().FirstOrDefault();
+        var transactionSample = _fixture.AgentLog.TryGetTransactionSample("WebTransaction/MVC/DefaultController/Query");
         var errorEvents = _fixture.AgentLog.GetErrorEvents().ToList();
         var errorTraces = _fixture.AgentLog.GetErrorTraces().ToList();
         var firstErrorEvent = errorEvents.FirstOrDefault();

tests/Agent/IntegrationTests/IntegrationTests/CSP/HighSecurityModeEnabled.cs

@@ -104,7 +104,7 @@ public void Test()
         const string originalErrorMessage = "!Exception~Message!";
 
         var displayHost = _fixture.AgentLog.GetConnectData().DisplayHost;
-        var transactionSample = _fixture.AgentLog.GetTransactionSamples().FirstOrDefault();
+        var transactionSample = _fixture.AgentLog.TryGetTransactionSample("WebTransaction/MVC/DefaultController/Query");
         var getDataTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/DefaultController/Query");
         var getExceptionTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/DefaultController/ThrowException");
 

tests/Agent/IntegrationTests/IntegrationTests/CSP/SecurityPoliciesMostRestrictiveTests.cs

@@ -91,7 +91,7 @@ public void Test()
 
         const string originalErrorMessage = "!Exception~Message!";
 
-        var transactionSample = _fixture.AgentLog.GetTransactionSamples().FirstOrDefault();
+        var transactionSample = _fixture.AgentLog.TryGetTransactionSample("WebTransaction/MVC/DefaultController/Query");
         var getDataTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/DefaultController/Query");
         var getExceptionTransactionEvent = _fixture.AgentLog.TryGetTransactionEvent("WebTransaction/MVC/DefaultController/ThrowException");
 

tests/Agent/IntegrationTests/IntegrationTests/RemoteServiceFixtures/AwsLambda/LambdaTestToolFixture.cs

@@ -78,8 +78,12 @@ private void AddAdditionalEnvironmentVariableIfNotNull(string name, string value
 
     private void WarmUpTestTool()
     {
+        const int maxAttempts = 5;
 
-        for (var attempt = 1; attempt <= 3; attempt++)
+        // Give the tool a moment to start listening before the first probe
+        Thread.Sleep(TimeSpan.FromSeconds(5));
+
+        for (var attempt = 1; attempt <= maxAttempts; attempt++)
         {
             try
             {
@@ -93,7 +97,7 @@ private void WarmUpTestTool()
             catch (Exception e)
             {
                 TestLogger?.WriteLine($"Unable to warm up lambda test tool during attempt {attempt}. Exception: {e}");
-                if (attempt == 3)
+                if (attempt == maxAttempts)
                 {
                     // Halt the test because the test tool was unable to start
                     throw;