using PkmnLib.Dynamic.Models;
using PkmnLib.Dynamic.Models.Choices;
using PkmnLib.Dynamic.ScriptHandling;
using PkmnLib.Plugin.Gen7.Scripts.MoveVolatile;
using PkmnLib.Plugin.Gen7.Scripts.Side;
using PkmnLib.Static.Utils;

namespace PkmnLib.Plugin.Gen7.Tests.Scripts.MoveVolatile;

/// <summary>
/// Tests for the <see cref="FireWaterPledgeMove"/> volatile script, which implements the combined move of
/// Fire Pledge and Water Pledge, and the <see cref="RainbowEffect"/> it creates.
/// Bulbapedia (Fire Pledge): "If Fire Pledge and Water Pledge are used in the same turn, the ally moving
/// second will use Water Pledge with a power of 150, and create a rainbow on the user's side of the field
/// for four turns."
/// </summary>
public class FireWaterPledgeMoveTests
{
    /// <summary>
    /// Creates a fully mocked test setup. The user's battle side gets a real <see cref="ScriptSet"/> as its
    /// volatile script set, so the rainbow created by the combined move can be inspected and driven.
    /// </summary>
    private static (FireWaterPledgeMove script, IExecutingMove move, IBattlePokemon user, IBattleSide side, IScriptSet
        sideVolatile) CreateTestSetup()
    {
        var script = new FireWaterPledgeMove();
        var move = Substitute.For<IExecutingMove>();

        var side = Substitute.For<IBattleSide>();
        side.GetScripts().Returns(_ => new ScriptIterator(new List<IEnumerable<ScriptContainer>>()));
        var sideVolatile = new ScriptSet(side);
        side.VolatileScripts.Returns(sideVolatile);

        var user = Substitute.For<IBattlePokemon>();
        move.User.Returns(user);
        user.BattleSide.Returns(side);

        return (script, move, user, side, sideVolatile);
    }

    /// <summary>
    /// Bulbapedia: "the ally moving second will use Water Pledge" when Fire Pledge and Water Pledge are
    /// combined. The marked choice's move is replaced with Water Pledge.
    /// </summary>
    [Test]
    public async Task ChangeMove_CombinedMove_ExecutesWaterPledge()
    {
        // Arrange
        var (script, _, _, _, _) = CreateTestSetup();
        var choice = Substitute.For<IMoveChoice>();
        StringKey moveName = "fire_pledge";

        // Act
        script.ChangeMove(choice, ref moveName);

        // Assert
        await Assert.That(moveName).IsEqualTo(new StringKey("water_pledge"));
    }

    /// <summary>
    /// Bulbapedia: "the ally moving second will use Water Pledge with a power of 150".
    /// </summary>
    [Test]
    public async Task ChangeBasePower_CombinedMove_PowerIs150()
    {
        // Arrange
        var (script, move, _, _, _) = CreateTestSetup();
        var target = Substitute.For<IBattlePokemon>();
        ushort basePower = 80;

        // Act
        script.ChangeBasePower(move, target, 0, ref basePower);

        // Assert
        await Assert.That(basePower).IsEqualTo((ushort)150);
    }

    /// <summary>
    /// Bulbapedia: the combined move will "create a rainbow on the user's side of the field for four turns."
    /// The <see cref="RainbowEffect"/> is added to the user's side.
    /// </summary>
    [Test]
    public async Task OnSecondaryEffect_CombinedMoveHits_AddsRainbowToUsersSide()
    {
        // Arrange
        var (script, move, _, _, sideVolatile) = CreateTestSetup();
        var target = Substitute.For<IBattlePokemon>();

        // Act
        script.OnSecondaryEffect(move, target, 0);

        // Assert
        await Assert.That(sideVolatile.Get<RainbowEffect>()).IsNotNull();
    }

    /// <summary>
    /// Bulbapedia: "The rainbow doubles the probability of additional effects taking place for moves used
    /// by that side of the field".
    /// </summary>
    [Test, Arguments(10f, 20f), Arguments(30f, 60f), Arguments(50f, 100f)]
    public async Task RainbowEffect_ChangeEffectChance_DoublesTheChance(float chance, float expected)
    {
        // Arrange
        var (script, move, _, _, sideVolatile) = CreateTestSetup();
        var target = Substitute.For<IBattlePokemon>();
        script.OnSecondaryEffect(move, target, 0);
        var rainbow = sideVolatile.Get<RainbowEffect>()!;

        // Act
        rainbow.ChangeEffectChance(move, target, 0, ref chance);

        // Assert
        await Assert.That(chance).IsEqualTo(expected);
    }

    /// <summary>
    /// Bulbapedia: the rainbow is created "on the user's side of the field for four turns." After the
    /// fourth end-of-turn since its creation the rainbow must be gone.
    /// </summary>
    [Test]
    public async Task RainbowEffect_AfterFourEndOfTurns_RainbowHasFaded()
    {
        // Arrange
        var (script, move, _, side, sideVolatile) = CreateTestSetup();
        var target = Substitute.For<IBattlePokemon>();
        script.OnSecondaryEffect(move, target, 0);
        var rainbow = sideVolatile.Get<RainbowEffect>()!;
        var battle = Substitute.For<IBattle>();

        // Act - four end-of-turn ticks, counting the turn the rainbow was created
        for (var i = 0; i < 4; i++)
            rainbow.OnEndTurn(side, battle);

        // Assert
        await Assert.That(sideVolatile.Contains("rainbow_effect")).IsFalse();
    }

    /// <summary>
    /// Technical test: the rainbow does not stay around forever; it removes itself from the side once its
    /// counter runs out (currently after the fifth end-of-turn).
    /// </summary>
    [Test]
    public async Task RainbowEffect_AfterFiveEndOfTurns_HasRemovedItself()
    {
        // Arrange
        var (script, move, _, side, sideVolatile) = CreateTestSetup();
        var target = Substitute.For<IBattlePokemon>();
        script.OnSecondaryEffect(move, target, 0);
        var rainbow = sideVolatile.Get<RainbowEffect>()!;
        var battle = Substitute.For<IBattle>();

        // Act
        for (var i = 0; i < 5; i++)
            rainbow.OnEndTurn(side, battle);

        // Assert
        await Assert.That(sideVolatile.Contains("rainbow_effect")).IsFalse();
    }
}