* Still trying to track down GHA issues

Author: lukebakken

projects/Test/Integration/TestToxiproxy.cs

@@ -46,6 +46,8 @@ namespace Test.Integration
     public class TestToxiproxy : IntegrationFixture
     {
         private readonly TimeSpan _heartbeatTimeout = TimeSpan.FromSeconds(1);
+        private ToxiproxyManager _toxiproxyManager;
+        private int _proxyPort;
 
         public TestToxiproxy(ITestOutputHelper output) : base(output)
         {
@@ -59,7 +61,15 @@ public override Task InitializeAsync()
             Assert.Null(_conn);
             Assert.Null(_channel);
 
-            return Task.CompletedTask;
+            _toxiproxyManager = new ToxiproxyManager(_testDisplayName, IsRunningInCI, IsWindows);
+            _proxyPort = _toxiproxyManager.ProxyPort;
+            return _toxiproxyManager.InitializeAsync();
+        }
+
+        public override async Task DisposeAsync()
+        {
+            await _toxiproxyManager.DisposeAsync();
+            await base.DisposeAsync();
         }
 
         [SkippableFact]
@@ -68,11 +78,8 @@ public async Task TestCloseConnection()
         {
             Skip.IfNot(AreToxiproxyTestsEnabled, "RABBITMQ_TOXIPROXY_TESTS is not set, skipping test");
 
-            using var pm = new ToxiproxyManager(_testDisplayName, IsRunningInCI, IsWindows);
-            await pm.InitializeAsync();
-
             ConnectionFactory cf = CreateConnectionFactory();
-            cf.Port = pm.ProxyPort;
+            cf.Port = _proxyPort;
             cf.AutomaticRecoveryEnabled = true;
             cf.NetworkRecoveryInterval = TimeSpan.FromSeconds(1);
             cf.RequestedHeartbeat = TimeSpan.FromSeconds(1);
@@ -172,11 +179,11 @@ async Task PublishLoop()
 
             Assert.True(await messagePublishedTcs.Task);
 
-            Task disableProxyTask = pm.DisableAsync();
+            Task disableProxyTask = _toxiproxyManager.DisableAsync();
 
             await Task.WhenAll(disableProxyTask, connectionShutdownTcs.Task);
 
-            Task enableProxyTask = pm.EnableAsync();
+            Task enableProxyTask = _toxiproxyManager.EnableAsync();
 
             Task whenAllTask = Task.WhenAll(enableProxyTask, recoverySucceededTcs.Task);
             await whenAllTask.WaitAsync(TimeSpan.FromSeconds(15));
@@ -193,11 +200,8 @@ public async Task TestThatStoppedSocketResultsInHeartbeatTimeout()
         {
             Skip.IfNot(AreToxiproxyTestsEnabled, "RABBITMQ_TOXIPROXY_TESTS is not set, skipping test");
 
-            using var pm = new ToxiproxyManager(_testDisplayName, IsRunningInCI, IsWindows);
-            await pm.InitializeAsync();
-
             ConnectionFactory cf = CreateConnectionFactory();
-            cf.Port = pm.ProxyPort;
+            cf.Port = _proxyPort;
             cf.RequestedHeartbeat = _heartbeatTimeout;
             cf.AutomaticRecoveryEnabled = false;
 
@@ -229,7 +233,7 @@ public async Task TestThatStoppedSocketResultsInHeartbeatTimeout()
             timeoutToxic.Attributes.Timeout = 0;
             timeoutToxic.Toxicity = 1.0;
 
-            Task<TimeoutToxic> addToxicTask = pm.AddToxicAsync(timeoutToxic);
+            Task<TimeoutToxic> addToxicTask = _toxiproxyManager.AddToxicAsync(timeoutToxic);
 
             await Assert.ThrowsAsync<AlreadyClosedException>(() =>
             {
@@ -243,11 +247,8 @@ public async Task TestTcpReset_GH1464()
         {
             Skip.IfNot(AreToxiproxyTestsEnabled, "RABBITMQ_TOXIPROXY_TESTS is not set, skipping test");
 
-            using var pm = new ToxiproxyManager(_testDisplayName, IsRunningInCI, IsWindows);
-            await pm.InitializeAsync();
-
             ConnectionFactory cf = CreateConnectionFactory();
-            cf.Endpoint = new AmqpTcpEndpoint(IPAddress.Loopback.ToString(), pm.ProxyPort);
+            cf.Endpoint = new AmqpTcpEndpoint(IPAddress.Loopback.ToString(), _proxyPort);
             cf.RequestedHeartbeat = TimeSpan.FromSeconds(5);
             cf.AutomaticRecoveryEnabled = true;
 
@@ -283,11 +284,11 @@ public async Task TestTcpReset_GH1464()
             resetPeerToxic.Attributes.Timeout = 500;
             resetPeerToxic.Toxicity = 1.0;
 
-            Task<ResetPeerToxic> addToxicTask = pm.AddToxicAsync(resetPeerToxic);
+            Task<ResetPeerToxic> addToxicTask = _toxiproxyManager.AddToxicAsync(resetPeerToxic);
 
             await Task.WhenAll(addToxicTask, connectionShutdownTcs.Task);
 
-            await pm.RemoveToxicAsync(toxicName);
+            await _toxiproxyManager.RemoveToxicAsync(toxicName);
 
             await recoveryTask;
         }
@@ -302,11 +303,8 @@ public async Task TestPublisherConfirmationThrottling()
             const int MaxOutstandingConfirms = 8;
             const int BatchSize = MaxOutstandingConfirms * 2;
 
-            using var pm = new ToxiproxyManager(_testDisplayName, IsRunningInCI, IsWindows);
-            await pm.InitializeAsync();
-
             ConnectionFactory cf = CreateConnectionFactory();
-            cf.Endpoint = new AmqpTcpEndpoint(IPAddress.Loopback.ToString(), pm.ProxyPort);
+            cf.Endpoint = new AmqpTcpEndpoint(IPAddress.Loopback.ToString(), _proxyPort);
             cf.RequestedHeartbeat = TimeSpan.FromSeconds(5);
             cf.AutomaticRecoveryEnabled = true;
 
@@ -371,7 +369,7 @@ public async Task TestPublisherConfirmationThrottling()
 
             await Task.Delay(TimeSpan.FromSeconds(1));
 
-            Task<BandwidthToxic> addToxicTask = pm.AddToxicAsync(bandwidthToxic);
+            Task<BandwidthToxic> addToxicTask = _toxiproxyManager.AddToxicAsync(bandwidthToxic);
 
             while (true)
             {
@@ -387,7 +385,7 @@ public async Task TestPublisherConfirmationThrottling()
             }
 
             await addToxicTask.WaitAsync(WaitSpan);
-            await pm.RemoveToxicAsync(toxicName).WaitAsync(WaitSpan);
+            await _toxiproxyManager.RemoveToxicAsync(toxicName).WaitAsync(WaitSpan);
 
             await messagesPublishedTcs.Task.WaitAsync(WaitSpan);
             await publishTask.WaitAsync(WaitSpan);

projects/Test/Integration/ToxiproxyManager.cs

@@ -9,7 +9,7 @@
 
 namespace Integration
 {
-    public class ToxiproxyManager : IDisposable
+    public class ToxiproxyManager : IAsyncDisposable
     {
         private const string ProxyNamePrefix = "rmq";
         private const ushort ProxyPortStart = 55669;
@@ -21,7 +21,7 @@ public class ToxiproxyManager : IDisposable
         private readonly Client _proxyClient;
         private readonly Proxy _proxy;
 
-        private bool _disposedValue;
+        private bool _disposedValue = false;
 
         public ToxiproxyManager(string testDisplayName, bool isRunningInCI, bool isWindows)
         {
@@ -115,30 +115,24 @@ public Task DisableAsync()
             return _proxyClient.UpdateAsync(_proxy);
         }
 
-        public void Dispose()
-        {
-            // Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method
-            Dispose(disposing: true);
-            GC.SuppressFinalize(this);
-        }
-
-        protected virtual void Dispose(bool disposing)
+        public async ValueTask DisposeAsync()
         {
             if (!_disposedValue)
             {
-                if (disposing)
+                try
                 {
-                    try
-                    {
-                        _proxyClient.DeleteAsync(_proxy).GetAwaiter().GetResult();
-                        _proxyConnection.Dispose();
-                    }
-                    catch
-                    {
-                    }
+                    await _proxyClient.DeleteAsync(_proxy);
+                    _proxyConnection.Dispose();
+                }
+                catch (Exception ex)
+                {
+                    string now = DateTime.Now.ToString("o", CultureInfo.InvariantCulture);
+                    Console.Error.WriteLine("{0} [ERROR] error disposing proxy '{1}': {2}", now, _proxy.Name, ex);
+                }
+                finally
+                {
+                    _disposedValue = true;
                 }
-
-                _disposedValue = true;
             }
         }
     }