Musify/backend/Tests/Musify.Application.Tests/Albums/RequestAlbumPictureUploadCommandHandlerTests.cs

using ErrorOr;
using Microsoft.EntityFrameworkCore;
using Musify.Application.Albums;
using Musify.Application.Configuration;
using Musify.Application.Contracts;
using Musify.Application.Services;
using Musify.Application.Tests.TestSupport;
using NSubstitute;
using Xunit;

namespace Musify.Application.Tests.Albums
{
    public sealed class RequestAlbumPictureUploadCommandHandlerTests : HandlerTestBase
    {
        private readonly IStorageService storageService = Substitute.For<IStorageService>();

        private RequestAlbumPictureUploadCommandHandler CreateHandler(UploadIntentConfiguration? uploadIntentConfig = null)
        {
            storageService
                .GetUploadUrlAsync(Arg.Any<string>(), Arg.Any<string>(), Arg.Any<string>(), Arg.Any<TimeSpan>(), Arg.Any<bool>(), Arg.Any<CancellationToken>())
                .Returns("https://storage.musify.test/presigned-upload");

            var config = uploadIntentConfig ?? TestConfigurations.UploadIntent();

            return new RequestAlbumPictureUploadCommandHandler(
                new UploadIntentIssuer(
                    Database,
                    storageService,
                    new UploadIntentValidator(Database, storageService, config),
                    TestConfigurations.Storage(),
                    config,
                    NoOpLogger<UploadIntentIssuer>()),
                TestConfigurations.Album());
        }

        [Fact]
        public async Task Handle_UserExists_IssuesUploadUrlAndCreatesIntent()
        {
            var user = TestEntities.User();
            await SeedAsync(user);

            var command = new RequestAlbumPictureUploadCommand(user.Id, "webp", "image/webp");

            var result = await CreateHandler().Handle(command, TestContext.Current.CancellationToken);

            Assert.False(result.IsError);
            Assert.Equal("https://storage.musify.test/presigned-upload", result.Value.UploadUrl);
            Assert.Single(await Database.UploadIntents.ToListAsync(TestContext.Current.CancellationToken));
        }

        [Fact]
        public async Task Handle_UserMissing_ReturnsNotFound()
        {
            var command = new RequestAlbumPictureUploadCommand(404, "webp", "image/webp");

            var result = await CreateHandler().Handle(command, TestContext.Current.CancellationToken);

            Assert.Equal(ErrorType.NotFound, result.FirstError.Type);
        }

        [Fact]
        public async Task Handle_UserOverByteQuota_ReturnsValidationErrorAndCreatesNoIntent()
        {
            var config = TestConfigurations.UploadIntent();
            config.MaxActiveUploadBytesPerUser = 100;

            var user = TestEntities.User();
            await SeedAsync(user);

            var command = new RequestAlbumPictureUploadCommand(user.Id, "webp", "image/webp", ExpectedSizeBytes: 1_000_000);

            var result = await CreateHandler(config).Handle(command, TestContext.Current.CancellationToken);

            Assert.Equal(ErrorType.Validation, result.FirstError.Type);
            Assert.Empty(await Database.UploadIntents.ToListAsync(TestContext.Current.CancellationToken));
        }
    }
}