Expose ManagementClient instance to ServiceControl

Author: bording

src/NServiceBus.Transport.RabbitMQ/NServiceBus.Transport.RabbitMQ.csproj

@@ -28,6 +28,7 @@
   <ItemGroup>
     <InternalsVisibleTo Include="NServiceBus.Transport.RabbitMQ.AcceptanceTests" Key="$(NServiceBusTestsKey)" />
     <InternalsVisibleTo Include="NServiceBus.Transport.RabbitMQ.Tests" Key="$(NServiceBusTestsKey)" />
+    <InternalsVisibleTo Include="ServiceControl.Transports.RabbitMQ" Key="$(NServiceBusKey)" />
   </ItemGroup>
 
 </Project>

src/NServiceBus.Transport.RabbitMQ/RabbitMQTransport.cs

@@ -203,6 +203,8 @@ public void AddClusterNode(string hostName, int port, bool useTls)
             additionalClusterNodes.Add((hostName, port, useTls));
         }
 
+        internal ManagementClient ManagementClient { get; private set; }
+
         /// <inheritdoc />
         public override async Task<TransportInfrastructure> Initialize(HostSettings hostSettings, ReceiveSettings[] receivers, string[] sendingAddresses, CancellationToken cancellationToken = default)
         {
@@ -226,14 +228,12 @@ public override async Task<TransportInfrastructure> Initialize(HostSettings host
                 additionalClusterNodes
             );
 
-            ManagementClient managementClient = null;
-
             if (UseManagementApi)
             {
-                managementClient = new ManagementClient(ConnectionConfiguration, ManagementApiConfiguration);
+                ManagementClient = new ManagementClient(ConnectionConfiguration, ManagementApiConfiguration);
             }
 
-            var brokerVerifier = new BrokerVerifier(connectionFactory, UseManagementApi, managementClient);
+            var brokerVerifier = new BrokerVerifier(connectionFactory, UseManagementApi, ManagementClient);
             await brokerVerifier.Initialize(cancellationToken).ConfigureAwait(false);
 
             var channelProvider = new ChannelProvider(connectionFactory, NetworkRecoveryInterval, RoutingTopology);