refactored test base to handle playback mode (Azure#33283)

azure-sdk · Jan 4, 2023 · abed7d9 · abed7d9
1 parent b6d858c
commit abed7d9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sdk/chaos/Azure.ResourceManager.Chaos/tests/ChaosManagementTestBase.cs b/sdk/chaos/Azure.ResourceManager.Chaos/tests/ChaosManagementTestBase.cs
@@ -70,6 +70,10 @@ protected async Task Initialize()
         /// </summary>
         public int CreateVmssId()
         {
+            if (this.Mode == RecordedTestMode.Playback)
+            {
+                return this.IsAsync ? 5 : 4;
+            }
             var framework = RuntimeInformation.FrameworkDescription;
             if (framework.IndexOf(TestConstants.DotNetFrameworkName, StringComparison.OrdinalIgnoreCase) != -1)
             {