* Add wait on RegisteredAsync to see how the changes in #1750 affec…

…t the order of operations.
rabbitmq · Jan 3, 2025 · b021daa · b021daa
1 parent dc6a3f4
commit b021daa
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/projects/Test/Integration/TestAsyncConsumer.cs b/projects/Test/Integration/TestAsyncConsumer.cs
@@ -71,6 +71,7 @@ public async Task TestBasicRoundtripConcurrent()
 
             var consumer = new AsyncEventingBasicConsumer(_channel);
 
+            var consumerRegisteredTcs = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
             var publish1SyncSource = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
             var publish2SyncSource = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
 
@@ -105,6 +106,20 @@ public async Task TestBasicRoundtripConcurrent()
                     return Task.CompletedTask;
                 };
 
+                consumer.RegisteredAsync += (object sender, ConsumerEventArgs ea) =>
+                {
+                    if (ReferenceEquals(consumer, sender))
+                    {
+                        consumerRegisteredTcs.SetResult(true);
+                    }
+                    else
+                    {
+                        var ex = Xunit.Sdk.EqualException.ForMismatchedValues(consumer, sender);
+                        consumerRegisteredTcs.SetException(ex);
+                    }
+                    return Task.CompletedTask;
+                };
+
                 consumer.ReceivedAsync += (o, a) =>
                 {
                     if (ByteArraysEqual(a.Body.Span, body1))
@@ -126,6 +141,7 @@ public async Task TestBasicRoundtripConcurrent()
                 };
 
                 await _channel.BasicConsumeAsync(q.QueueName, true, string.Empty, false, false, null, consumer);
+                await consumerRegisteredTcs.Task.WaitAsync(WaitSpan);
 
                 try
                 {