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 user = new User { Id = Random.Shared.NextInt64(1, long.MaxValue), Name = "roundtrip-user", NormalizedName = "ROUNDTRIP-USER" };
            await database.Users.AddAsync(user, TestContext.Current.CancellationToken);

            var track = new LocalTrack
            {
                Title = "Round Trip Track",
                NormalizedTitle = "ROUND TRIP TRACK",
                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("Round Trip Track", 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 user = new User { Id = Random.Shared.NextInt64(1, long.MaxValue), Name = "cascade-user", NormalizedName = "CASCADE-USER" };
            var album = new UserAlbum { Title = "Cascade Album", NormalizedTitle = "CASCADE ALBUM", 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) => new()
        {
            Title = "External",
            NormalizedTitle = "EXTERNAL",
            DurationSeconds = 100,
            Source = TrackSource.YouTube,
            ExternalId = externalId,
            Pictures = new TrackPictures(),
            Audio = new TrackAudio()
        };
    }
}