Musify/backend/Tests/Musify.Infrastructure.Tests/Persistence/DatabaseTests.cs

using Microsoft.EntityFrameworkCore;
using Musify.Domain.Entities;
using Musify.Domain.ValueObjects;
using Musify.Infrastructure.Tests.TestSupport;
using Xunit;

namespace Musify.Infrastructure.Tests.Persistence
{
    [Collection(InfrastructureCollection.Name)]
    public sealed class DatabaseTests(InfrastructureTestFixture fixture)
    {
        [Fact]
        public async Task SaveChangesAsync_LocalTrackWithOwnedTypes_RoundTripsThroughPostgres()
        {
            await using var database = fixture.CreateDatabase();

            var userName = "roundtrip-user";
            var user = new User { Id = Random.Shared.NextInt64(1, long.MaxValue), Name = userName, NormalizedName = userName.ToUpperInvariant() };
            await database.Users.AddAsync(user, TestContext.Current.CancellationToken);

            var trackTitle = "Round Trip Track";
            var track = new LocalTrack
            {
                Title = trackTitle,
                NormalizedTitle = trackTitle.ToUpperInvariant(),
                DurationSeconds = 123.4,
                OwnerUserId = user.Id,
                Owner = user,
                Pictures = new TrackPictures { OriginalName = "cover.webp", ProcessingStatus = ProcessingStatus.Pending },
                Audio = new TrackAudio { OriginalName = "song.mp3", TranscodeStatus = ProcessingStatus.Pending }
            };
            await database.LocalTracks.AddAsync(track, TestContext.Current.CancellationToken);
            await database.SaveChangesAsync(TestContext.Current.CancellationToken);

            await using var reloaded = fixture.CreateDatabase();
            var stored = await reloaded.Tracks.AsNoTracking().SingleAsync(t => t.Id == track.Id, TestContext.Current.CancellationToken);

            Assert.IsType<LocalTrack>(stored);
            Assert.Equal(trackTitle, stored.Title);
            Assert.Equal("cover.webp", stored.Pictures.OriginalName);
            Assert.Equal("song.mp3", stored.Audio.OriginalName);
            Assert.Equal(user.Id, ((LocalTrack)stored).OwnerUserId);
        }

        [Fact]
        public async Task ExternalTracks_SourceAndExternalIdUniqueIndex_RejectsDuplicates()
        {
            await using var database = fixture.CreateDatabase();

            var videoId = $"dup-{Guid.NewGuid():N}";
            await database.ExternalTracks.AddAsync(BuildExternalTrack(videoId), TestContext.Current.CancellationToken);
            await database.SaveChangesAsync(TestContext.Current.CancellationToken);

            await using var second = fixture.CreateDatabase();
            await second.ExternalTracks.AddAsync(BuildExternalTrack(videoId), TestContext.Current.CancellationToken);

            await Assert.ThrowsAsync<DbUpdateException>(() => second.SaveChangesAsync(TestContext.Current.CancellationToken));
        }

        [Fact]
        public async Task AlbumHasTracks_DeletingAlbum_CascadesToTheLinkRows()
        {
            await using var database = fixture.CreateDatabase();

            var userName = "cascade-user";
            var user = new User { Id = Random.Shared.NextInt64(1, long.MaxValue), Name = userName, NormalizedName = userName.ToUpperInvariant() };
            var albumTitle = "Cascade Album";
            var album = new UserAlbum { Title = albumTitle, NormalizedTitle = albumTitle.ToUpperInvariant(), OwnerUserId = user.Id };
            var track = BuildExternalTrack($"cascade-{Guid.NewGuid():N}");
            var link = new AlbumHasTrack { AlbumId = album.Id, TrackId = track.Id, TrackNumber = 1 };

            await database.AddRangeAsync(user, album, track, link);
            await database.SaveChangesAsync(TestContext.Current.CancellationToken);

            await using var deleter = fixture.CreateDatabase();
            var albumToDelete = await deleter.UserAlbums.SingleAsync(a => a.Id == album.Id, TestContext.Current.CancellationToken);
            deleter.UserAlbums.Remove(albumToDelete);
            await deleter.SaveChangesAsync(TestContext.Current.CancellationToken);

            await using var verifier = fixture.CreateDatabase();
            Assert.False(await verifier.AlbumHasTracks.AnyAsync(l => l.AlbumId == album.Id, TestContext.Current.CancellationToken));
            Assert.True(await verifier.ExternalTracks.AnyAsync(t => t.Id == track.Id, TestContext.Current.CancellationToken));
        }

        private static ExternalTrack BuildExternalTrack(string externalId)
        {
            var title = "External";
            return new ExternalTrack
            {
                Title = title,
                NormalizedTitle = title.ToUpperInvariant(),
                DurationSeconds = 100,
                Source = TrackSource.YouTube,
                ExternalId = externalId,
                Pictures = new TrackPictures(),
                Audio = new TrackAudio()
            };
        }
    }
}