KeyValueUnlockCommandTests.cs•4.84 kB
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
using System.CommandLine.Parsing;
using System.Text.Json;
using AzureMcp.AppConfig.Commands.KeyValue;
using AzureMcp.AppConfig.Services;
using AzureMcp.Core.Models.Command;
using AzureMcp.Core.Options;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using NSubstitute;
using NSubstitute.ExceptionExtensions;
using Xunit;
using static AzureMcp.AppConfig.Commands.KeyValue.KeyValueUnlockCommand;
namespace AzureMcp.AppConfig.UnitTests.KeyValue;
[Trait("Area", "AppConfig")]
public class KeyValueUnlockCommandTests
{
private readonly IServiceProvider _serviceProvider;
private readonly IAppConfigService _appConfigService;
private readonly ILogger<KeyValueUnlockCommand> _logger;
private readonly KeyValueUnlockCommand _command;
private readonly CommandContext _context;
private readonly Parser _parser;
public KeyValueUnlockCommandTests()
{
_appConfigService = Substitute.For<IAppConfigService>();
_logger = Substitute.For<ILogger<KeyValueUnlockCommand>>();
_command = new(_logger);
_parser = new(_command.GetCommand());
_serviceProvider = new ServiceCollection()
.AddSingleton(_appConfigService)
.BuildServiceProvider();
_context = new(_serviceProvider);
}
[Fact]
public async Task ExecuteAsync_UnlocksKeyValue_WhenValidParametersProvided()
{
// Arrange
var args = _parser.Parse([
"--subscription", "sub123",
"--account", "account1",
"--key", "my-key"
]);
// Act
var response = await _command.ExecuteAsync(_context, args);
// Assert
Assert.Equal(200, response.Status);
await _appConfigService.Received(1).UnlockKeyValue(
"account1",
"my-key",
"sub123",
null,
Arg.Any<RetryPolicyOptions>(),
null);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<KeyValueUnlockResult>(json, new JsonSerializerOptions
{
PropertyNamingPolicy = JsonNamingPolicy.CamelCase
});
Assert.NotNull(result);
Assert.Equal("my-key", result.Key);
}
[Fact]
public async Task ExecuteAsync_UnlocksKeyValueWithLabel_WhenLabelProvided()
{
// Arrange
var args = _parser.Parse([
"--subscription", "sub123",
"--account", "account1",
"--key", "my-key",
"--label", "prod"
]);
// Act
var response = await _command.ExecuteAsync(_context, args);
// Assert
Assert.Equal(200, response.Status);
await _appConfigService.Received(1).UnlockKeyValue(
"account1",
"my-key",
"sub123",
null,
Arg.Any<RetryPolicyOptions>(),
"prod");
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<KeyValueUnlockResult>(json, new JsonSerializerOptions
{
PropertyNamingPolicy = JsonNamingPolicy.CamelCase
});
Assert.NotNull(result);
Assert.Equal("my-key", result.Key);
Assert.Equal("prod", result.Label);
}
[Fact]
public async Task ExecuteAsync_Returns500_WhenServiceThrowsException()
{
// Arrange
_appConfigService.UnlockKeyValue(
Arg.Any<string>(),
Arg.Any<string>(),
Arg.Any<string>(),
Arg.Any<string>(),
Arg.Any<RetryPolicyOptions>(),
Arg.Any<string>())
.ThrowsAsync(new Exception("Failed to unlock key-value"));
var args = _parser.Parse([
"--subscription", "sub123",
"--account", "account1",
"--key", "my-key"
]);
// Act
var response = await _command.ExecuteAsync(_context, args);
// Assert
Assert.Equal(500, response.Status);
Assert.Contains("Failed to unlock key-value", response.Message);
}
[Theory]
[InlineData("--account", "account1", "--key", "my-key")] // Missing subscription
[InlineData("--subscription", "sub123", "--key", "my-key")] // Missing account
[InlineData("--subscription", "sub123", "--account", "account1")] // Missing key
public async Task ExecuteAsync_Returns400_WhenRequiredParametersAreMissing(params string[] args)
{
// Arrange
var parseResult = _parser.Parse(args);
// Act
var response = await _command.ExecuteAsync(_context, parseResult);
// Assert
Assert.Equal(400, response.Status);
Assert.Contains("required", response.Message.ToLower());
}
}