// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
using System.CommandLine;
using System.Text.Json;
using AzureMcp.Core.Models.Command;
using AzureMcp.Core.Options;
using AzureMcp.FunctionApp.Commands;
using AzureMcp.FunctionApp.Commands.FunctionApp;
using AzureMcp.FunctionApp.Models;
using AzureMcp.FunctionApp.Services;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using NSubstitute;
using Xunit;
namespace AzureMcp.FunctionApp.UnitTests.FunctionApp;
public sealed class FunctionAppGetCommandTests
{
private readonly IServiceProvider _serviceProvider;
private readonly IFunctionAppService _service;
private readonly ILogger<FunctionAppGetCommand> _logger;
private readonly FunctionAppGetCommand _command;
public FunctionAppGetCommandTests()
{
_service = Substitute.For<IFunctionAppService>();
_logger = Substitute.For<ILogger<FunctionAppGetCommand>>();
var collection = new ServiceCollection();
collection.AddSingleton(_service);
_serviceProvider = collection.BuildServiceProvider();
_command = new(_logger);
}
[Fact]
public void Constructor_InitializesCommandCorrectly()
{
var command = _command.GetCommand();
Assert.Equal("get", command.Name);
Assert.NotNull(command.Description);
Assert.NotEmpty(command.Description);
}
[Theory]
[InlineData("--subscription sub123 --resource-group rg1 --function-app app1", true)]
[InlineData("--subscription sub123 --resource-group rg1", false)]
[InlineData("--subscription sub123 --function-app app1", false)]
[InlineData("--resource-group rg1 --function-app app1", false)]
public async Task ExecuteAsync_ValidatesInputCorrectly(string args, bool shouldSucceed)
{
if (shouldSucceed)
{
_service.GetFunctionApp(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string?>(), Arg.Any<RetryPolicyOptions?>())
.Returns(new FunctionAppInfo("app1", "rg1", "eastus", "plan1", "Running", "app1.azurewebsites.net", null));
}
var context = new CommandContext(_serviceProvider);
var parseResult = _command.GetCommand().Parse(args);
var response = await _command.ExecuteAsync(context, parseResult);
Assert.Equal(shouldSucceed ? 200 : 400, response.Status);
}
[Fact]
public async Task ExecuteAsync_ReturnsFunctionApp()
{
var expected = new FunctionAppInfo("app1", "rg1", "eastus", "plan1", "Running", "app1.azurewebsites.net", null);
_service.GetFunctionApp(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string?>(), Arg.Any<RetryPolicyOptions?>())
.Returns(expected);
var context = new CommandContext(_serviceProvider);
var parseResult = _command.GetCommand().Parse("--subscription sub123 --resource-group rg1 --function-app app1");
var response = await _command.ExecuteAsync(context, parseResult);
Assert.Equal(200, response.Status);
Assert.NotNull(response.Results);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<FunctionAppGetCommand.FunctionAppGetCommandResult>(json, FunctionAppJsonContext.Default.FunctionAppGetCommandResult);
Assert.NotNull(result);
Assert.Equal(expected.Name, result.FunctionApp.Name);
Assert.Equal(expected.ResourceGroupName, result.FunctionApp.ResourceGroupName);
}
[Fact]
public async Task ExecuteAsync_ReturnsNullWhenNotFound()
{
_service.GetFunctionApp(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string?>(), Arg.Any<RetryPolicyOptions?>())
.Returns((FunctionAppInfo?)null);
var context = new CommandContext(_serviceProvider);
var parseResult = _command.GetCommand().Parse("--subscription sub123 --resource-group rg1 --function-app app1");
var response = await _command.ExecuteAsync(context, parseResult);
Assert.Equal(200, response.Status);
Assert.Null(response.Results);
}
[Fact]
public async Task ExecuteAsync_HandlesServiceErrors()
{
_service.GetFunctionApp(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string?>(), Arg.Any<RetryPolicyOptions?>())
.Returns(Task.FromException<FunctionAppInfo?>(new Exception("Test error")));
var context = new CommandContext(_serviceProvider);
var parseResult = _command.GetCommand().Parse("--subscription sub123 --resource-group rg1 --function-app app1");
var response = await _command.ExecuteAsync(context, parseResult);
Assert.Equal(500, response.Status);
Assert.Contains("Test error", response.Message);
Assert.Contains("troubleshooting", response.Message);
}
}
