Skip to content

Commit 37f2858

Browse files
Update KeyVaultCommandTests and StorageCommandTests
1 parent 2a78190 commit 37f2858

2 files changed

Lines changed: 22 additions & 0 deletions

File tree

tools/Azure.Mcp.Tools.KeyVault/tests/Azure.Mcp.Tools.KeyVault.Tests/KeyVaultCommandTests.cs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,17 @@ public async Task Should_create_certificate()
217217
[Fact]
218218
public async Task Should_import_certificate()
219219
{
220+
if (string.Equals(Environment.GetEnvironmentVariable("MCP_TEST_TRANSPORT"), "http", StringComparison.OrdinalIgnoreCase))
221+
{
222+
var unavailableResult = await Client.CallToolAsync(
223+
"keyvault_certificate_import",
224+
new Dictionary<string, object?>(),
225+
cancellationToken: TestContext.Current.CancellationToken);
226+
Assert.True(unavailableResult.IsError);
227+
Assert.Contains("not found", McpTestUtilities.GetFirstText(unavailableResult.Content), StringComparison.OrdinalIgnoreCase);
228+
return;
229+
}
230+
220231
var fakePassword = _importCertificateAssets.Password;
221232
var tempPath = _importCertificateAssets.CreateTempCopy();
222233

tools/Azure.Mcp.Tools.Storage/tests/Azure.Mcp.Tools.Storage.Tests/StorageCommandTests.cs

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,17 @@ public async Task Should_get_blob_details()
264264
[Fact]
265265
public async Task Should_upload_blob()
266266
{
267+
if (string.Equals(Environment.GetEnvironmentVariable("MCP_TEST_TRANSPORT"), "http", StringComparison.OrdinalIgnoreCase))
268+
{
269+
var unavailableResult = await Client.CallToolAsync(
270+
"storage_blob_upload",
271+
new Dictionary<string, object?>(),
272+
cancellationToken: TestContext.Current.CancellationToken);
273+
Assert.True(unavailableResult.IsError);
274+
Assert.Contains("not found", McpTestUtilities.GetFirstText(unavailableResult.Content), StringComparison.OrdinalIgnoreCase);
275+
return;
276+
}
277+
267278
// Create a temporary file to upload
268279
var tempFileName = RegisterOrRetrieveVariable("blobName", $"test-upload-{DateTime.UtcNow.Ticks}.txt");
269280
var tempFilePath = Path.Combine(Path.GetTempPath(), tempFileName);

0 commit comments

Comments
 (0)