diff --git a/tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/Azure/AzureEventHubsTests.cs b/tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/Azure/AzureEventHubsTests.cs index 690c7577f421..36eea1bac77c 100644 --- a/tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/Azure/AzureEventHubsTests.cs +++ b/tracer/test/Datadog.Trace.ClrProfiler.IntegrationTests/Azure/AzureEventHubsTests.cs @@ -101,7 +101,7 @@ public async Task TestEventHubsEnumerableIntegration(string packageVersion, stri using (var agent = EnvironmentHelper.GetMockAgent()) using (await RunSampleAndWaitForExit(agent, packageVersion: packageVersion)) { - var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000); + var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000, assertExpectedCount: false); using var s = new AssertionScope(); @@ -109,7 +109,7 @@ public async Task TestEventHubsEnumerableIntegration(string packageVersion, stri var receiveSpans = spans.Where(span => span.Name == "azure_eventhubs.receive").ToList(); sendSpans.Should().HaveCount(1, "Expected 1 SendAsync span with azure_eventhubs.send operation for enumerable"); - receiveSpans.Should().HaveCount(1, "Expected 1 receive span"); + receiveSpans.Should().HaveCountGreaterOrEqualTo(1, "Expected at least 1 receive span"); var enumerableSendSpans = sendSpans.Where(s => s.Resource == "samples-eventhubs-hub").ToList(); enumerableSendSpans.Should().HaveCount(1, "Expected 1 enumerable send span from SendAsync operation"); @@ -174,7 +174,7 @@ public async Task TestEventHubsEnumerableIntegrationWithoutBatchLinks(string pac using (var agent = EnvironmentHelper.GetMockAgent()) using (await RunSampleAndWaitForExit(agent, packageVersion: packageVersion)) { - var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000); + var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000, assertExpectedCount: false); using var s = new AssertionScope(); @@ -182,7 +182,7 @@ public async Task TestEventHubsEnumerableIntegrationWithoutBatchLinks(string pac var receiveSpans = spans.Where(span => span.Name == "azure_eventhubs.receive").ToList(); sendSpans.Should().HaveCount(1, "Expected 1 SendAsync span for enumerable (no individual event spans when batch links disabled)"); - receiveSpans.Should().HaveCount(1, "Expected 1 receive span"); + receiveSpans.Should().HaveCountGreaterOrEqualTo(1, "Expected at least 1 receive span"); var createSpans = spans.Where(span => span.Name == "azure_eventhubs.create").ToList(); createSpans.Should().BeEmpty("Expected no individual message spans (azure_eventhubs.create) when batch links disabled"); @@ -213,7 +213,7 @@ public async Task TestEventHubsBufferedProducerIntegration(string packageVersion using (var agent = EnvironmentHelper.GetMockAgent()) using (await RunSampleAndWaitForExit(agent, packageVersion: packageVersion)) { - var spans = await agent.WaitForSpansAsync(5, timeoutInMilliseconds: 30000); + var spans = await agent.WaitForSpansAsync(5, timeoutInMilliseconds: 30000, assertExpectedCount: false); using var s = new AssertionScope(); @@ -223,7 +223,7 @@ public async Task TestEventHubsBufferedProducerIntegration(string packageVersion createSpans.Should().HaveCount(3, "Expected 3 TryAdd spans with azure_eventhubs.create operation"); sendSpans.Should().HaveCount(1, "Expected 1 buffered send span with azure_eventhubs.send operation"); - receiveSpans.Should().HaveCount(1, "Expected 1 receive span"); + receiveSpans.Should().HaveCountGreaterOrEqualTo(1, "Expected at least 1 receive span"); var individualMessageSpans = createSpans.Where(s => s.Resource == "samples-eventhubs-hub").ToList(); individualMessageSpans.Should().HaveCount(3, "Expected 3 individual message spans from TryAdd operations"); @@ -255,7 +255,7 @@ public async Task TestEventHubsBufferedProducerIntegrationWithoutBatchLinks(stri using (var agent = EnvironmentHelper.GetMockAgent()) using (await RunSampleAndWaitForExit(agent, packageVersion: packageVersion)) { - var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000); + var spans = await agent.WaitForSpansAsync(2, timeoutInMilliseconds: 30000, assertExpectedCount: false); using var s = new AssertionScope(); @@ -263,7 +263,7 @@ public async Task TestEventHubsBufferedProducerIntegrationWithoutBatchLinks(stri var receiveSpans = spans.Where(span => span.Name == "azure_eventhubs.receive").ToList(); sendSpans.Should().HaveCount(1, "Expected 1 buffered send span (no individual TryAdd spans when batch links disabled)"); - receiveSpans.Should().HaveCount(1, "Expected 1 receive span"); + receiveSpans.Should().HaveCountGreaterOrEqualTo(1, "Expected at least 1 receive span"); var createSpans = spans.Where(span => span.Name == "azure_eventhubs.create").ToList(); createSpans.Should().BeEmpty("Expected no individual message spans (azure_eventhubs.create) when batch links disabled");