Fix test connection retry

nats-io · Jan 25, 2025 · f2fd102 · f2fd102
1 parent d4fa354
commit f2fd102
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/tests/NATS.Client.Core.Tests/ClusterTests.cs b/tests/NATS.Client.Core.Tests/ClusterTests.cs
@@ -1,3 +1,5 @@
+using NATS.Client.TestUtilities2;
+
 namespace NATS.Client.Core.Tests;
 
 public class ClusterTests(ITestOutputHelper output)
@@ -47,6 +49,7 @@ public async Task Seed_urls_on_retry(bool userAuthInUrl)
             NoRandomize = true,
             Url = $"{url1},{url2}",
         });
+        await nats.ConnectRetryAsync();
 
         using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(60));
 

diff --git a/tests/NATS.Client.Core.Tests/TlsClientTest.cs b/tests/NATS.Client.Core.Tests/TlsClientTest.cs
@@ -2,6 +2,7 @@
 using System.Net.Sockets;
 using System.Security.Cryptography.X509Certificates;
 using System.Text;
+using NATS.Client.TestUtilities2;
 
 namespace NATS.Client.Core.Tests;
 
@@ -78,6 +79,7 @@ public async Task Client_cannot_connect_without_certificate(TransportType transp
         var clientOpts = server.ClientOpts(NatsOpts.Default);
         clientOpts = clientOpts with { TlsOpts = clientOpts.TlsOpts with { CertFile = null, KeyFile = null } };
         await using var nats = new NatsConnection(clientOpts);
+        await nats.ConnectRetryAsync();
 
         var exceptionTask = Assert.ThrowsAsync<NatsException>(async () => await nats.ConnectAsync());