Tests

Author: huangpf

src/Common/Commands.ScenarioTest/Commands.ScenarioTest.csproj

@@ -236,6 +236,9 @@
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunAzurePlatformVMImageNegativeTest.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>
+    <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunEnableAndDisableDataCollectionTests.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="SessionRecords\Microsoft.WindowsAzure.Commands.ScenarioTest.ServiceManagementTests\RunNewAzureComputeArgumentListTests.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>

src/Common/Commands.ScenarioTest/Resources/ServiceManagement/ServiceManagementTests.ps1

@@ -590,4 +590,22 @@ function Run-ServiceDeploymentExtensionCmdletTests
         # Cleanup
         Cleanup-CloudService $svcName;
     }
-}
+}
+
+# Run Data Collection Cmdlet Tests
+function Run-EnableAndDisableDataCollectionTests
+{
+    $st = Enable-AzureDataCollection;
+
+    $locations = Get-AzureLocation;
+    foreach ($loc in $locations)
+    {
+        $svcName = getAssetName;
+        $st = New-AzureService -ServiceName $svcName -Location $loc.Name;
+        
+        # Cleanup
+        Cleanup-CloudService $svcName
+    }
+
+    $st = Disable-AzureDataCollection;
+}

src/Common/Commands.ScenarioTest/ServiceManagement/ScenarioTests.cs

@@ -125,5 +125,14 @@ public void RunServiceDeploymentExtensionCmdletTests()
         {
             this.RunPowerShellTest("Run-ServiceDeploymentExtensionCmdletTests");
         }
+
+        [Fact]
+        [Trait(Category.Service, Category.ServiceManagement)]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        [Trait(Category.AcceptanceType, Category.BVT)]
+        public void RunEnableAndDisableDataCollectionTests()
+        {
+            this.RunPowerShellTest("Run-EnableAndDisableDataCollectionTests");
+        }
     }
 }