using HSchool.Server.Game;

namespace HSchool.Server.Tests;

public class SwarmUiSettingsStoreTests
{
    [Fact]
    public void Resolve_UsesAgeRulePreset()
    {
        var config = SwarmUiConfigFile.CreateDefault();
        var profile = config.Resolve(8, PortraitKind.Avatar);

        Assert.Equal("child", profile.PresetId);
    }

    [Fact]
    public void Resolve_FallsBackToActivePresetForAdults()
    {
        var config = SwarmUiConfigFile.CreateDefault();
        var profile = config.Resolve(34, PortraitKind.Full);

        Assert.Equal("default", profile.PresetId);
    }

    [Fact]
    public void NormalizeAfterLoad_MigratesLegacyFlatFile()
    {
        var config = new SwarmUiConfigFile
        {
            Model = "legacy.safetensors",
            Steps = 12,
            Positive = "legacy prompt",
            Avatar = new SwarmUiKindPreset { Width = 640, Height = 640 },
        };

        config.NormalizeAfterLoad();

        Assert.Single(config.Presets);
        Assert.Equal("default", config.Presets[0].Id);
        Assert.Equal("legacy.safetensors", config.Presets[0].Model);
        Assert.Equal(12, config.Presets[0].Steps);
    }
}