Musify/backend/Tests/Musify.Infrastructure.Tests/TestSupport/InfrastructureTestFixture.cs

using System.Text.Json;
using Amazon.S3;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Musify.Infrastructure.Configuration;
using Musify.Infrastructure.Persistence;
using Testcontainers.PostgreSql;
using DotNet.Testcontainers.Builders;
using DotNet.Testcontainers.Containers;
using Xunit;

namespace Musify.Infrastructure.Tests.TestSupport
{
    public sealed class InfrastructureTestFixture : IAsyncLifetime
    {
        public const string S3AccessKey = "musify-test";
        public const string S3SecretKey = "musify-test-secret";
        public const string S3Bucket = "musify-test-bucket";

        private readonly PostgreSqlContainer postgres = new PostgreSqlBuilder("postgres:17-alpine")
            .WithDatabase("musify_test")
            .WithUsername("postgres")
            .WithPassword("postgres")
            .Build();

        private readonly IContainer seaweedFs = new ContainerBuilder("chrislusf/seaweedfs:latest")
            .WithEntrypoint("/bin/sh", "-c")
            .WithCommand(BuildSeaweedFsStartupScript())
            .WithPortBinding(9333, assignRandomHostPort: true)
            .WithPortBinding(8333, assignRandomHostPort: true)
            .WithWaitStrategy(Wait.ForUnixContainer()
                .UntilHttpRequestIsSucceeded(request => request.ForPort(9333).ForPath("/cluster/healthz")))
            .Build();

        public string ConnectionString => postgres.GetConnectionString();

        public string S3ServiceUrl => $"http://{seaweedFs.Hostname}:{seaweedFs.GetMappedPublicPort(8333)}";

        public async ValueTask InitializeAsync()
        {
            await Task.WhenAll(postgres.StartAsync(), seaweedFs.StartAsync());

            await using var database = new Database(new DatabaseConfiguration { ConnectionString = ConnectionString });
            await database.Database.MigrateAsync();

            using var s3 = CreateS3Client();
            await s3.PutBucketAsync(S3Bucket);
        }

        public async ValueTask DisposeAsync()
        {
            await postgres.DisposeAsync();
            await seaweedFs.DisposeAsync();
        }

        public Database CreateDatabase()
        {
            var configuration = new ConfigurationBuilder()
                .AddInMemoryCollection([new("Database:ConnectionString", ConnectionString)])
                .Build();

            var services = new ServiceCollection();
            services.AddDatabase(configuration);
            return services.BuildServiceProvider().GetRequiredService<Database>();
        }

        public IAmazonS3 CreateS3Client() =>
            new AmazonS3Client(S3AccessKey, S3SecretKey, new AmazonS3Config
            {
                ServiceURL = S3ServiceUrl,
                ForcePathStyle = true,
                UseHttp = true,
            });

        private static string BuildSeaweedFsStartupScript()
        {
            var identity = JsonSerializer.Serialize(new
            {
                identities = new[]
                {
                    new
                    {
                        name = "musify-test",
                        credentials = new[] { new { accessKey = S3AccessKey, secretKey = S3SecretKey } },
                        actions = new[] { "Admin", "Read", "Write" }
                    }
                }
            });

            return $"""
                cat > /tmp/s3.conf <<'JSON'
                {identity}
                JSON
                exec weed server -ip.bind=0.0.0.0 -dir=/data -master.volumeSizeLimitMB=1024 -volume.max=0 -filer -s3 -s3.config=/tmp/s3.conf
                """;
        }
    }

    [CollectionDefinition(Name)]
    public sealed class InfrastructureCollection : ICollectionFixture<InfrastructureTestFixture>
    {
        public const string Name = "Infrastructure";
    }
}