Add extension methods to enable and disable data collection for tests

Author: cormacpayne

src/Common/Commands.Common/Extensions/CmdletExtensions.cs

@@ -12,6 +12,7 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using Microsoft.WindowsAzure.Commands.Common;
 using System;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
@@ -216,6 +217,18 @@ public static void InvokeEndProcessing(this PSCmdlet cmdlt)
             dynMethod.Invoke(cmdlt, null);
         }
 
+        public static void EnableDataCollection(this AzurePSCmdlet cmdlt)
+        {
+            FieldInfo dynField = (typeof(AzurePSCmdlet)).GetField("_dataCollectionProfile", BindingFlags.NonPublic | BindingFlags.Static);
+            dynField.SetValue(cmdlt, new AzurePSDataCollectionProfile(true));
+        }
+
+        public static void DisableDataCollection(this AzurePSCmdlet cmdlt)
+        {
+            FieldInfo dynField = (typeof(AzurePSCmdlet)).GetField("_dataCollectionProfile", BindingFlags.NonPublic | BindingFlags.Static);
+            dynField.SetValue(cmdlt, new AzurePSDataCollectionProfile(false));
+        }
+
         #endregion
     }
 }

src/ResourceManager/Profile/Commands.Profile.Test/ContextCmdletTests.cs

@@ -44,7 +44,6 @@ public ContextCmdletTests(ITestOutputHelper output)
             AzureSession.Instance.DataStore = dataStore;
             commandRuntimeMock = new MockCommandRuntime();
             AzureSession.Instance.AuthenticationFactory = new MockTokenAuthenticationFactory();
-            Environment.SetEnvironmentVariable(AzurePSDataCollectionProfile.EnvironmentVariableName, "false");
         }
 
         [Fact]
@@ -75,6 +74,7 @@ public void GetAzureContextNoLogin()
 
             // Setup
             cmdlt.CommandRuntime = commandRuntimeMock;
+            cmdlt.DisableDataCollection();
             var profile = AzureRmProfileProvider.Instance.Profile;
             AzureRmProfileProvider.Instance.Profile = new AzureRmProfile();