Add DataMigrationTestRunner to replace DataMigrationTestController (#…

…18103) * Add DataMigrationTestRunner to replace DataMigrationTestController * Add DataMigrationTestRunner to replace DataMigrationTestController Co-authored-by: Ziyue Zheng <[email protected]>
Azure · May 12, 2022 · 8c840b2 · 8c840b2
1 parent 6c78b5c
commit 8c840b2
Show file tree

Hide file tree

Showing 3 changed files with 81 additions and 183 deletions.
diff --git a/src/DataMigration/DataMigration.Test/ScenarioTests/DataMigrationTestController.cs b/src/DataMigration/DataMigration.Test/ScenarioTests/DataMigrationTestController.cs
diff --git a/src/DataMigration/DataMigration.Test/ScenarioTests/DataMigrationTestRunner.cs b/src/DataMigration/DataMigration.Test/ScenarioTests/DataMigrationTestRunner.cs
@@ -0,0 +1,56 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using System.Collections.Generic;
+using Microsoft.Azure.Commands.TestFx;
+using Xunit.Abstractions;
+
+namespace Microsoft.Azure.Commands.ScenarioTest.DmsTest
+{
+    public class DataMigrationTestRunner
+    {
+        protected readonly ITestRunner TestRunner;
+
+        protected DataMigrationTestRunner(ITestOutputHelper output)
+        {
+            TestRunner = TestManager.CreateInstance(output)
+                .WithNewPsScriptFilename($"{GetType().Name}.ps1")
+                .WithProjectSubfolderForTests("ScenarioTests")
+                .WithCommonPsScripts(new[]
+                {
+                    @"Common.ps1",
+                    @"../AzureRM.Resources.ps1"
+                })
+                .WithNewRmModules(helper => new[]
+                {
+                    helper.RMProfileModule,
+                    helper.GetRMModulePath("Az.DataMigration.psd1")
+                })
+                .WithNewRecordMatcherArguments(
+                    userAgentsToIgnore: new Dictionary<string, string>
+                    {
+                        {"Microsoft.Azure.Management.Resources.ResourceManagementClient", "2016-02-01"}
+                    },
+                    resourceProviders: new Dictionary<string, string>
+                    {
+                        {"Microsoft.Resources", null},
+                        {"Microsoft.Features", null},
+                        {"Microsoft.Authorization", null},
+                        {"Microsoft.Compute", null}
+                    }
+                )
+                .Build();
+        }
+    }
+}