* Close connection mid-integration test for OAuth2

Author: lukebakken

projects/Test/OAuth2/TestOAuth2.cs

@@ -36,6 +36,7 @@
 using RabbitMQ.Client;
 using RabbitMQ.Client.Events;
 using RabbitMQ.Client.OAuth2;
+using Test;
 using Xunit;
 using Xunit.Abstractions;
 
@@ -211,6 +212,8 @@ private Task OnCredentialsRefreshedAsync(Credentials? credentials, Exception? ex
         [Fact]
         public async void IntegrationTest()
         {
+            Task? closeConnectionTask = null;
+
             using (IChannel publishChannel = await DeclarePublishChannelAsync())
             {
                 using (IChannel consumeChannel = await DeclareConsumeChannelAsync())
@@ -225,7 +228,20 @@ public async void IntegrationTest()
                             var delaySpan = TimeSpan.FromSeconds(_tokenExpiresInSeconds + 10);
                             _testOutputHelper.WriteLine("{0} [INFO] wait '{1}' until Token expires. Attempt #{1}",
                                 DateTime.Now, delaySpan, (i + 1));
+
+                            if (i == 2)
+                            {
+                                closeConnectionTask = Util.CloseConnectionAsync(_connection);
+                            }
+
                             await Task.Delay(delaySpan);
+
+                            if (closeConnectionTask != null)
+                            {
+                                await closeConnectionTask;
+                                closeConnectionTask = null;
+                            }
+
                             _testOutputHelper.WriteLine("{0} [INFO] Resuming ...", DateTime.Now);
 
                             await PublishAsync(publishChannel);