TestRunCreateCommandTests.cs•6.64 kB
// 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.LoadTesting.Commands.LoadTestRun;
using AzureMcp.LoadTesting.Models.LoadTestRun;
using AzureMcp.LoadTesting.Services;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using NSubstitute;
using Xunit;
namespace AzureMcp.LoadTesting.UnitTests;
public class TestRunCreateCommandTests
{
private readonly IServiceProvider _serviceProvider;
private readonly ILoadTestingService _service;
private readonly ILogger<TestRunCreateCommand> _logger;
private readonly TestRunCreateCommand _command;
public TestRunCreateCommandTests()
{
_service = Substitute.For<ILoadTestingService>();
_logger = Substitute.For<ILogger<TestRunCreateCommand>>();
var collection = new ServiceCollection();
collection.AddSingleton(_service);
_serviceProvider = collection.BuildServiceProvider();
_command = new(_logger);
}
[Fact]
public void Constructor_InitializesCommandCorrectly()
{
var command = _command.GetCommand();
Assert.Equal("create", command.Name);
Assert.NotNull(command.Description);
Assert.NotEmpty(command.Description);
}
[Fact]
public async Task ExecuteAsync_CreatesLoadTestRun()
{
var expected = new TestRun { TestId = "testId1", TestRunId = "testRunId1", DisplayName = "displayName" };
_service.CreateOrUpdateLoadTestRunAsync(
Arg.Is("sub123"), Arg.Is("testResourceName"), Arg.Is("testId1"), Arg.Is("run1"), Arg.Is((string?)null), Arg.Is("resourceGroup123"), Arg.Is("tenant123"), Arg.Is("displayName"), Arg.Is((string?)null), Arg.Is(false), Arg.Any<RetryPolicyOptions>())
.Returns(expected);
var command = new TestRunCreateCommand(_logger);
var args = command.GetCommand().Parse([
"--subscription", "sub123",
"--resource-group", "resourceGroup123",
"--test-resource-name", "testResourceName",
"--testrun-id", "run1",
"--tenant", "tenant123",
"--test-id", "testId1",
"--display-name", "displayName"
]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.NotNull(response.Results);
Assert.Equal(200, response.Status);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<TestRunCreateCommandResult>(json);
Assert.NotNull(result);
Assert.Equal(expected.TestId, result.TestRun.TestId);
Assert.Equal(expected.TestRunId, result.TestRun.TestRunId);
Assert.Equal(expected.DisplayName, result.TestRun.DisplayName);
}
[Fact]
public async Task ExecuteAsync_RerunLoadTestRun()
{
var expected = new TestRun { TestId = "testId1", TestRunId = "testRunId1" };
_service.CreateOrUpdateLoadTestRunAsync(
Arg.Is("sub123"), Arg.Is("testResourceName"), Arg.Is("testId1"), Arg.Is("run1"), Arg.Is("oldId1"), Arg.Is("resourceGroup123"), Arg.Is("tenant123"), Arg.Is((string?)null), Arg.Is((string?)null), Arg.Is(false), Arg.Any<RetryPolicyOptions>())
.Returns(expected);
var command = new TestRunCreateCommand(_logger);
var args = command.GetCommand().Parse([
"--subscription", "sub123",
"--resource-group", "resourceGroup123",
"--test-resource-name", "testResourceName",
"--testrun-id", "run1",
"--tenant", "tenant123",
"--test-id", "testId1",
"--old-testrun-id", "oldId1"
]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.NotNull(response);
Assert.NotNull(response.Results);
Assert.Equal(200, response.Status);
var json = JsonSerializer.Serialize(response.Results);
var result = JsonSerializer.Deserialize<TestRunCreateCommandResult>(json);
Assert.NotNull(result);
Assert.Equal(expected.TestId, result.TestRun.TestId);
Assert.Equal(expected.TestRunId, result.TestRun.TestRunId);
}
[Fact]
public async Task ExecuteAsync_HandlesBadRequestErrors()
{
var expected = new TestRun();
_service.CreateOrUpdateLoadTestRunAsync(
Arg.Is("sub123"), Arg.Is("testResourceName"), Arg.Is("testId1"), Arg.Is("run1"), Arg.Is((string?)null), Arg.Is("resourceGroup123"), Arg.Is("tenant123"), Arg.Is((string?)null), Arg.Is((string?)null), Arg.Is(false), Arg.Any<RetryPolicyOptions>())
.Returns(expected);
var command = new TestRunCreateCommand(_logger);
var args = command.GetCommand().Parse([
"--subscription", "sub123",
"--resource-group", "resourceGroup123",
"--test-resource-name", "testResourceName",
"--tenant", "tenant123",
"--testrun-id", "run1"
]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.Equal(400, response.Status);
}
[Fact]
public async Task ExecuteAsync_HandlesServiceErrors()
{
_service.CreateOrUpdateLoadTestRunAsync(
Arg.Is("sub123"), Arg.Is("testResourceName"), Arg.Is("testId1"), Arg.Is("run1"), Arg.Is((string?)null), Arg.Is("resourceGroup123"), Arg.Is("tenant123"), Arg.Is((string?)null), Arg.Is((string?)null), Arg.Is(false), Arg.Any<RetryPolicyOptions>())
.Returns(Task.FromException<TestRun>(new Exception("Test error")));
var command = new TestRunCreateCommand(_logger);
var args = command.GetCommand().Parse([
"--subscription", "sub123",
"--resource-group", "resourceGroup123",
"--test-resource-name", "testResourceName",
"--testrun-id", "run1",
"--tenant", "tenant123",
"--test-id", "testId1"
]);
var context = new CommandContext(_serviceProvider);
var response = await command.ExecuteAsync(context, args);
Assert.Equal(500, response.Status);
Assert.Contains("Test error", response.Message);
Assert.Contains("troubleshooting", response.Message);
}
private class TestRunCreateCommandResult
{
public TestRun TestRun { get; set; } = new();
}
}