Connect auth test subscriber sync fix (#110)

nats-io · Aug 11, 2023 · 638e2d3 · 638e2d3
1 parent 920a3ef
commit 638e2d3
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/tests/NATS.Client.Core.Tests/NatsConnectionTest.Auth.cs b/tests/NATS.Client.Core.Tests/NatsConnectionTest.Auth.cs
@@ -116,17 +116,23 @@ public async Task UserCredentialAuthTest(string name, string serverConfig, NatsO
         var signalComplete1 = new WaitSignal();
         var signalComplete2 = new WaitSignal();
 
+        var syncCount = 0;
         var natsSub = await subConnection.SubscribeAsync<int>(subject);
         var register = natsSub.Register(x =>
         {
+            Interlocked.Increment(ref syncCount);
             _output.WriteLine($"Received: {x}");
             if (x.Data == 1)
                 signalComplete1.Pulse();
             if (x.Data == 2)
                 signalComplete2.Pulse();
         });
 
-        await subConnection.PingAsync(); // wait for subscribe complete
+        var syncCount1 = Volatile.Read(ref syncCount);
+        await Retry.Until(
+            "subscribed",
+            () => syncCount1 != Volatile.Read(ref syncCount),
+            async () => await pubConnection.PublishAsync(subject, 0));
 
         _output.WriteLine("AUTHENTICATED CONNECTION");
         await pubConnection.PublishAsync(subject, 1);
@@ -146,6 +152,13 @@ public async Task UserCredentialAuthTest(string name, string serverConfig, NatsO
         await pubConnection.ConnectAsync(); // wait open again
 
         _output.WriteLine("AUTHENTICATED RE-CONNECTION");
+
+        var syncCount2 = Volatile.Read(ref syncCount);
+        await Retry.Until(
+            "re-subscribed",
+            () => syncCount2 != Volatile.Read(ref syncCount),
+            async () => await pubConnection.PublishAsync(subject, 0));
+
         await pubConnection.PublishAsync(subject, 2);
         await signalComplete2;