AvailabilityStatusListCommandTests.cs•7 kB
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
using System.CommandLine;
using System.CommandLine.Parsing;
using System.Text.Json;
using AzureMcp.Core.Models;
using AzureMcp.Core.Models.Command;
using AzureMcp.Core.Options;
using AzureMcp.ResourceHealth.Commands.AvailabilityStatus;
using AzureMcp.ResourceHealth.Services;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using NSubstitute;
using NSubstitute.ExceptionExtensions;
using Xunit;
using AvailabilityStatusModel = AzureMcp.ResourceHealth.Models.AvailabilityStatus;
namespace AzureMcp.ResourceHealth.UnitTests.AvailabilityStatus;
public class AvailabilityStatusListCommandTests
{
private readonly IServiceProvider _serviceProvider;
private readonly IResourceHealthService _resourceHealthService;
private readonly ILogger<AvailabilityStatusListCommand> _logger;
public AvailabilityStatusListCommandTests()
{
_resourceHealthService = Substitute.For<IResourceHealthService>();
_logger = Substitute.For<ILogger<AvailabilityStatusListCommand>>();
var collection = new ServiceCollection();
collection.AddSingleton(_resourceHealthService);
_serviceProvider = collection.BuildServiceProvider();
}
[Fact]
public async Task ExecuteAsync_ReturnsAvailabilityStatuses_WhenResourcesExist()
{
var subscriptionId = "12345678-1234-1234-1234-123456789012";
var expectedStatuses = new List<AvailabilityStatusModel>
{
new()
{
ResourceId = "/subscriptions/12345678-1234-1234-1234-123456789012/resourceGroups/rg1/providers/Microsoft.Compute/virtualMachines/vm1",
AvailabilityState = "Available",
Summary = "Resource is healthy",
DetailedStatus = "Virtual machine is running normally"
},
new()
{
ResourceId = "/subscriptions/12345678-1234-1234-1234-123456789012/resourceGroups/rg2/providers/Microsoft.Storage/storageAccounts/storage1",
AvailabilityState = "Available",
Summary = "Resource is healthy",
DetailedStatus = "Storage account is accessible"
}
};
_resourceHealthService.ListAvailabilityStatusesAsync(subscriptionId, null, Arg.Any<string?>(), Arg.Any<RetryPolicyOptions>())
.Returns(expectedStatuses);
var command = new AvailabilityStatusListCommand(_logger);
var args = command.GetCommand().Parse(["--subscription", subscriptionId]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.Equal(200, response.Status);
Assert.Equal("Success", response.Message);
Assert.NotNull(response.Results);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<AvailabilityStatusListResult>(json, new JsonSerializerOptions()
{
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
PropertyNameCaseInsensitive = true
});
Assert.NotNull(result);
Assert.Equal(2, result.Statuses.Count);
Assert.Equal("Available", result.Statuses[0].AvailabilityState);
Assert.Equal("Available", result.Statuses[1].AvailabilityState);
}
[Fact]
public async Task ExecuteAsync_ReturnsFilteredAvailabilityStatuses_WhenResourceGroupProvided()
{
var subscriptionId = "12345678-1234-1234-1234-123456789012";
var resourceGroup = "test-rg";
var expectedStatuses = new List<AvailabilityStatusModel>
{
new()
{
ResourceId = "/subscriptions/12345678-1234-1234-1234-123456789012/resourceGroups/test-rg/providers/Microsoft.Compute/virtualMachines/vm1",
AvailabilityState = "Available",
Summary = "Resource is healthy",
DetailedStatus = "Virtual machine is running normally"
}
};
_resourceHealthService.ListAvailabilityStatusesAsync(subscriptionId, resourceGroup, Arg.Any<string?>(), Arg.Any<RetryPolicyOptions>())
.Returns(expectedStatuses);
var command = new AvailabilityStatusListCommand(_logger);
var args = command.GetCommand().Parse(["--subscription", subscriptionId, "--resource-group", resourceGroup]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.Equal(200, response.Status);
Assert.Equal("Success", response.Message);
Assert.NotNull(response.Results);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<AvailabilityStatusListResult>(json, new JsonSerializerOptions()
{
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
PropertyNameCaseInsensitive = true
});
Assert.NotNull(result);
Assert.Single(result.Statuses);
Assert.Contains("test-rg", result.Statuses[0].ResourceId);
}
[Fact]
public async Task ExecuteAsync_HandlesException()
{
var subscriptionId = "12345678-1234-1234-1234-123456789012";
var expectedError = "Test error. To mitigate this issue, please refer to the troubleshooting guidelines here at https://aka.ms/azmcp/troubleshooting.";
_resourceHealthService.ListAvailabilityStatusesAsync(subscriptionId, null, Arg.Any<string?>(), Arg.Any<RetryPolicyOptions>())
.ThrowsAsync(new Exception("Test error"));
var command = new AvailabilityStatusListCommand(_logger);
var parser = new Parser(command.GetCommand());
var args = parser.Parse(["--subscription", subscriptionId]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.Equal(500, response.Status);
Assert.Equal(expectedError, response.Message);
}
[Theory]
[InlineData("--subscription")]
public async Task ExecuteAsync_ReturnsError_WhenRequiredParameterIsMissing(string missingParameter)
{
var command = new AvailabilityStatusListCommand(_logger);
var args = command.GetCommand().Parse(
[
missingParameter == "--subscription" ? "" : "--subscription", "12345678-1234-1234-1234-123456789012",
]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.Equal(400, response.Status);
Assert.Equal($"Missing Required options: {missingParameter}", response.Message);
}
private record AvailabilityStatusListResult(IReadOnlyList<AvailabilityStatusModel> Statuses);
}