Skip to content

Commit 64f1ac4

Browse files
committed
updated tests for messageid tag
1 parent 20a71f4 commit 64f1ac4

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

projects/Test/SequentialIntegration/TestActivitySource.cs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ public async Task TestPublisherWithPublicationAddressAndConsumerActivityTagsAsyn
293293
[InlineData(false, true)]
294294
[InlineData(true, false)]
295295
[InlineData(false, false)]
296-
public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOperationName, bool verifyMessageIdTag)
296+
public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOperationName, bool useMessageId)
297297
{
298298
RabbitMQActivitySource.UseRoutingKeyAsOperationName = useRoutingKeyAsOperationName;
299299
var activities = new List<Activity>();
@@ -302,7 +302,7 @@ public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOpera
302302
string queue = $"queue-{Guid.NewGuid()}";
303303
const string msg = "for basic.get";
304304

305-
var basicProps = new BasicProperties() { MessageId = Guid.NewGuid().ToString() };
305+
var basicProps = useMessageId ? new BasicProperties() { MessageId = Guid.NewGuid().ToString() } : new BasicProperties();
306306

307307
try
308308
{
@@ -315,7 +315,7 @@ public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOpera
315315
ok = await _channel.QueueDeclarePassiveAsync(queue);
316316
Assert.Equal(0u, ok.MessageCount);
317317
await Task.Delay(500);
318-
AssertActivityData(useRoutingKeyAsOperationName, queue, activities, false, verifyMessageIdTag);
318+
AssertActivityData(useRoutingKeyAsOperationName, queue, activities, false, basicProps.MessageId);
319319
}
320320
finally
321321
{
@@ -404,7 +404,7 @@ private static ActivityListener StartActivityListener(List<Activity> activities)
404404
}
405405

406406
private void AssertActivityData(bool useRoutingKeyAsOperationName, string queueName,
407-
List<Activity> activityList, bool isDeliver = false, bool verifyMessageIdTag = false)
407+
List<Activity> activityList, bool isDeliver = false, string messageId = null)
408408
{
409409
string childName = isDeliver ? "deliver" : "fetch";
410410
Activity[] activities = activityList.ToArray();
@@ -449,10 +449,10 @@ private void AssertActivityData(bool useRoutingKeyAsOperationName, string queueN
449449
AssertIntTagGreaterThanZero(sendActivity, RabbitMQActivitySource.MessagingBodySize);
450450
AssertIntTagGreaterThanZero(receiveActivity, RabbitMQActivitySource.MessagingBodySize);
451451

452-
if (verifyMessageIdTag)
452+
if (messageId is not null)
453453
{
454-
AssertStringTagNotNullOrEmpty(sendActivity, RabbitMQActivitySource.MessageId);
455-
AssertStringTagNotNullOrEmpty(receiveActivity, RabbitMQActivitySource.MessageId);
454+
AssertStringTagEquals(sendActivity, RabbitMQActivitySource.MessageId, messageId);
455+
AssertStringTagEquals(receiveActivity, RabbitMQActivitySource.MessageId, messageId);
456456
}
457457
}
458458
}

projects/Test/SequentialIntegration/TestOpenTelemetry.cs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ public async Task TestPublisherWithCachedStringsAndConsumerActivityTagsAsync(boo
305305
[InlineData(false, true)]
306306
[InlineData(true, false)]
307307
[InlineData(false, false)]
308-
public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOperationName, bool verifyMessageIdTag)
308+
public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOperationName, bool useMessageId)
309309
{
310310
var exportedItems = new List<Activity>();
311311
using var tracer = Sdk.CreateTracerProviderBuilder()
@@ -320,7 +320,7 @@ public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOpera
320320
string queue = $"queue-{Guid.NewGuid()}";
321321
const string msg = "for basic.get";
322322

323-
var basicProps = new BasicProperties() { MessageId = Guid.NewGuid().ToString() };
323+
var basicProps = useMessageId ? new BasicProperties() { MessageId = Guid.NewGuid().ToString() } : new BasicProperties();
324324

325325
try
326326
{
@@ -335,7 +335,7 @@ public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOpera
335335
ok = await _channel.QueueDeclarePassiveAsync(queue);
336336
Assert.Equal(0u, ok.MessageCount);
337337
await Task.Delay(500);
338-
AssertActivityData(useRoutingKeyAsOperationName, queue, exportedItems, false, verifyMessageIdTag);
338+
AssertActivityData(useRoutingKeyAsOperationName, queue, exportedItems, false, basicProps.MessageId);
339339
}
340340
finally
341341
{
@@ -344,7 +344,7 @@ public async Task TestPublisherAndBasicGetActivityTags(bool useRoutingKeyAsOpera
344344
}
345345

346346
private void AssertActivityData(bool useRoutingKeyAsOperationName, string queueName,
347-
List<Activity> activityList, bool isDeliver = false, bool verifyMessageIdTag = false)
347+
List<Activity> activityList, bool isDeliver = false, string messageId = null)
348348
{
349349
string childName = isDeliver ? "deliver" : "fetch";
350350
string childType = isDeliver ? "process" : "receive";
@@ -390,10 +390,10 @@ private void AssertActivityData(bool useRoutingKeyAsOperationName, string queueN
390390
AssertStringTagEquals(sendActivity, RabbitMQActivitySource.MessagingOperationType, "send");
391391
AssertStringTagEquals(sendActivity, RabbitMQActivitySource.MessagingOperationName, "publish");
392392

393-
if (verifyMessageIdTag)
393+
if (messageId is not null)
394394
{
395-
AssertStringTagNotNullOrEmpty(sendActivity, RabbitMQActivitySource.MessageId);
396-
AssertStringTagNotNullOrEmpty(receiveActivity, RabbitMQActivitySource.MessageId);
395+
AssertStringTagEquals(sendActivity, RabbitMQActivitySource.MessageId, messageId);
396+
AssertStringTagEquals(receiveActivity, RabbitMQActivitySource.MessageId, messageId);
397397
}
398398
}
399399
}

0 commit comments

Comments
 (0)