Fix slow mocked tests for service management networking

Author: markcowl

src/Common/Commands.ScenarioTests.Common/Mocks/MockClientFactory.cs

@@ -19,6 +19,8 @@
 using System.Net;
 using System.Net.Http;
 using System.Net.Http.Headers;
+using System.Threading;
+using System.Threading.Tasks;
 using Hyak.Common;
 using Microsoft.Azure.Test.HttpRecorder;
 using Microsoft.Azure.Common;
@@ -96,7 +98,7 @@ public TClient CreateCustomClient<TClient>(params object[] parameters) where TCl
                 {
                     throw new ArgumentException(
                         string.Format("TestManagementClientHelper class wasn't initialized with the {0} client.",
-                            typeof(TClient).Name));
+                            typeof (TClient).Name));
                 }
                 else
                 {
@@ -107,6 +109,13 @@ public TClient CreateCustomClient<TClient>(params object[] parameters) where TCl
                     return newRealClient;
                 }
             }
+            else
+            {
+                // Use the WithHandler method to create an extra reference to the http client
+                // this will prevent the httpClient from being disposed in a long-runnign test using 
+                // the same client for multiple cmdlets
+                client = client.WithHandler(new PassThroughDelegatingHandler());
+            }
 
             return client;
         }
@@ -162,5 +171,17 @@ public void AddUserAgent(string productName)
         }
 
         public HashSet<ProductInfoHeaderValue> UserAgents { get; set; }
+
+        /// <summary>
+        /// This class exists to allow adding an additional reference to the httpClient to prevent the client 
+        /// from being disposed.  Should not be used execpt in this mocked context
+        /// </summary>
+        class PassThroughDelegatingHandler : DelegatingHandler
+        {
+            protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
+            {
+                return base.SendAsync(request, cancellationToken);
+            }
+        }
     }
 }

src/ServiceManagement/Network/Commands.Network.Test/ScenarioTests/MultiVip/MultiVip.cs

@@ -12,6 +12,9 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using Microsoft.WindowsAzure.Management.Compute;
+using Microsoft.WindowsAzure.Management.Storage;
+
 namespace Microsoft.WindowsAzure.Commands.ServiceManagement.Network.Test.ScenarioTests
 {
     using Microsoft.Azure.Common.Authentication;
@@ -66,7 +69,9 @@ protected void SetupManagementClients()
         {
             var client = TestBase.GetServiceClient<NetworkManagementClient>(new RDFETestEnvironmentFactory());
             var client2 = TestBase.GetServiceClient<ManagementClient>(new RDFETestEnvironmentFactory());
-            helper.SetupSomeOfManagementClients(client, client2);
+            var client3 = TestBase.GetServiceClient<StorageManagementClient>(new RDFETestEnvironmentFactory());
+            var client4 = TestBase.GetServiceClient<ComputeManagementClient>(new RDFETestEnvironmentFactory());
+            helper.SetupManagementClients(client, client2, client3, client4);
         }
 
         protected void RunPowerShellTest(params string[] scripts)

src/ServiceManagement/Network/Commands.Network.Test/ScenarioTests/ReservedIPs/ReservedIP.cs

@@ -12,6 +12,9 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using Microsoft.WindowsAzure.Management.Compute;
+using Microsoft.WindowsAzure.Management.Storage;
+
 namespace Microsoft.WindowsAzure.Commands.ServiceManagement.Network.Test.ScenarioTests
 {
     using Microsoft.Azure.Common.Authentication;
@@ -73,7 +76,9 @@ protected void SetupManagementClients()
         {
             var client = TestBase.GetServiceClient<NetworkManagementClient>(new RDFETestEnvironmentFactory());
             var client2 = TestBase.GetServiceClient<ManagementClient>(new RDFETestEnvironmentFactory());
-            helper.SetupSomeOfManagementClients(client, client2);
+            var client3 = TestBase.GetServiceClient<ComputeManagementClient>(new RDFETestEnvironmentFactory());
+            var client4 = TestBase.GetServiceClient<StorageManagementClient>(new RDFETestEnvironmentFactory());
+            helper.SetupManagementClients(client, client2, client3, client4);
         }
 
         protected void RunPowerShellTest(params string[] scripts)