Update Managed Lustre test recordings (#2376)

Author: alzimmermsft

core/Microsoft.Mcp.Core/tests/Microsoft.Mcp.Tests/Client/RecordedCommandTestsBase.cs

@@ -483,4 +483,14 @@ public string GetSessionFilePath(string displayName)
         var fullPath = Path.Combine(dir, fileName).Replace('\\', '/');
         return fullPath;
     }
+
+    /// <summary>
+    /// Determines the polling interval to use for long-running operations. During live testing (Live or Record) the
+    /// poll interval will use liveMilliseconds for the interval. During playback testing a static 1 millisecond poll
+    /// interval is used.
+    /// </summary>
+    /// <param name="liveMilliseconds">Polling interval in milliseconds for live tests.</param>
+    /// <returns>The polling interval TimeSpan.</returns>
+    public TimeSpan PollInterval(long liveMilliseconds)
+        => TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(1) : TimeSpan.FromMilliseconds(liveMilliseconds);
 }

tools/Azure.Mcp.Tools.ManagedLustre/tests/Azure.Mcp.Tools.ManagedLustre.LiveTests/ManagedLustreCommandTests.cs

@@ -12,7 +12,8 @@
 
 namespace Azure.Mcp.Tools.ManagedLustre.LiveTests;
 
-public partial class ManagedLustreCommandTests(ITestOutputHelper output, TestProxyFixture fixture, LiveServerFixture liveServerFixture) : RecordedCommandTestsBase(output, fixture, liveServerFixture)
+public partial class ManagedLustreCommandTests(ITestOutputHelper output, TestProxyFixture fixture, LiveServerFixture liveServerFixture)
+    : RecordedCommandTestsBase(output, fixture, liveServerFixture)
 {
     private static readonly string[] _sanitizedHeaders =
     [
@@ -232,7 +233,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
 
         // Wait for filesystem to be available before creating jobs
         var maxWaitTime = TimeSpan.FromMinutes(30);
-        var pollInterval = TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(30);
+        var pollInterval = PollInterval(30_000);
         var startTime = DateTime.UtcNow;
         var isAvailable = false;
 
@@ -271,7 +272,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
         Assert.True(isAvailable, $"Filesystem '{fsName}' did not reach 'Succeeded' provisioning state within {maxWaitTime.TotalMinutes} minutes.");
 
         // Wait for filesystem to stabilize before creating jobs
-        await Task.Delay(TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(15), TestContext.Current.CancellationToken);
+        await Task.Delay(PollInterval(15_000), TestContext.Current.CancellationToken);
 
         // Test autoimport job lifecycle
         var autoimportJobNameStr = $"autoimport-{fsName}";
@@ -336,7 +337,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
         Assert.Contains(autoimportJobNameStr, autoimportJobText);
 
         // Wait 15 seconds to cancel auto import job
-        await Task.Delay(TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(15), TestContext.Current.CancellationToken);
+        await Task.Delay(PollInterval(15_000), TestContext.Current.CancellationToken);
 
         // Cancel autoimport job
         var autoimportCancelResult = await CallToolAsync(
@@ -373,7 +374,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
         Assert.Equal("Deleted", autoimportDeleteStatus.GetString());
 
         // Wait for filesystem to stabilize after deleting import job and before creating export job
-        await Task.Delay(TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(15), TestContext.Current.CancellationToken);
+        await Task.Delay(PollInterval(15_000), TestContext.Current.CancellationToken);
 
         // Test autoexport job lifecycle.
         var autoexportJobNameStr = $"autoexport-{fsName}";
@@ -436,7 +437,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
         Assert.Contains(autoexportJobNameStr, autoexportJobText);
 
         // Wait 15 seconds to cancel auto export job.
-        await Task.Delay(TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(15), TestContext.Current.CancellationToken);
+        await Task.Delay(PollInterval(15_000), TestContext.Current.CancellationToken);
         // Cancel autoexport job
         var autoexportCancelResult = await CallToolAsync(
             "managedlustre_fs_blob_autoexport_cancel",
@@ -472,7 +473,7 @@ public async Task Should_create_azure_managed_lustre_with_storage_and_cmk()
         Assert.Equal("Deleted", autoexportDeleteStatus.GetString());
 
         // Wait for filesystem to stabilize after deleting autoexport job and before creating import job
-        await Task.Delay(TestMode == TestMode.Playback ? TimeSpan.FromMilliseconds(100) : TimeSpan.FromSeconds(15), TestContext.Current.CancellationToken);
+        await Task.Delay(PollInterval(15_000), TestContext.Current.CancellationToken);
 
         // Test import job lifecycle
         var importJobNameStr = $"import-{fsName}";

tools/Azure.Mcp.Tools.ManagedLustre/tests/Azure.Mcp.Tools.ManagedLustre.LiveTests/assets.json

@@ -2,5 +2,5 @@
     "AssetsRepo": "Azure/azure-sdk-assets",
     "AssetsRepoPrefixPath": "",
     "TagPrefix": "Azure.Mcp.Tools.ManagedLustre.LiveTests",
-    "Tag": "Azure.Mcp.Tools.ManagedLustre.LiveTests_11ea42d368"
+    "Tag": "Azure.Mcp.Tools.ManagedLustre.LiveTests_2f6d918693"
 }