[tests] add graceful shutdown to mock collectors listener (#4847)

add graceful shutdown in test server, fix received log records count logging

Author: lachmatt

test/IntegrationTests/Helpers/MockLogsCollector.cs

@@ -21,6 +21,7 @@ internal sealed class MockLogsCollector : IDisposable
     private readonly TestHttpServer _listener;
     private readonly BlockingCollection<LogRecord> _logs = new(100); // bounded to avoid memory leak
     private readonly List<Expectation> _expectations = new();
+    private int _logsReceived;
 
     private CollectedExpectation? _collectedExpectation;
 
@@ -44,10 +45,10 @@ public MockLogsCollector(ITestOutputHelper output, string host = "localhost")
 
     public void Dispose()
     {
-        WriteOutput($"Shutting down. Total logs requests received: '{_logs.Count}'");
+        _listener.Dispose();
+        WriteOutput($"Shutting down. Total logs received: '{_logsReceived}'");
         ResourceExpector.Dispose();
         _logs.Dispose();
-        _listener.Dispose();
     }
 
     public void Expect(Func<LogRecord, bool> predicate, string? description = null)
@@ -213,6 +214,7 @@ private void HandleLogsMessage(ExportLogsServiceRequest logsMessage)
             {
                 foreach (var logRecord in scopeLogs.LogRecords ?? Enumerable.Empty<LogRecord>())
                 {
+                    Interlocked.Increment(ref _logsReceived);
                     _logs.Add(logRecord);
                 }
             }

test/IntegrationTests/Helpers/TestHttpServer.AspNetCore.cs

@@ -42,7 +42,8 @@ public TestHttpServer(ITestOutputHelper output, string name, params PathHandler[
                                 x.Run(pathHandler.Delegate);
                             });
                         }
-                    });
+                    })
+                    .UseShutdownTimeout(TimeSpan.FromSeconds(5));
             })
             .Build();
 
@@ -69,7 +70,8 @@ public static TestHttpServer CreateDefaultTestServer(ITestOutputHelper output)
 
     public void Dispose()
     {
-        WriteOutput($"Shutting down");
+        WriteOutput("Shutting down");
+        _listener.StopAsync().GetAwaiter().GetResult();
         _listener.Dispose();
     }