PkmnLib.NET/Plugins/PkmnLib.Plugin.Gen7.Tests/Scripts/Moves/CaptivateTests.cs

using PkmnLib.Dynamic.Models;
using PkmnLib.Plugin.Gen7.Scripts.Moves;
using PkmnLib.Static;
using PkmnLib.Static.Species;
using PkmnLib.Static.Utils;

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

/// <summary>
/// Tests for the <see cref="Captivate"/> move script.
/// Gen VII Bulbapedia behavior: "Captivate lowers the Special Attack stat of the target by two stages,
/// if it is the opposite gender of the user." Gender-unknown Pokémon and Pokémon with the Oblivious
/// Ability cannot be affected, and the move fails when used by a gender-unknown Pokémon.
/// </summary>
public class CaptivateTests
{
    private static (Captivate script, IExecutingMove move, IPokemon user, IPokemon target, IHitData hitData)
        CreateTestSetup(Gender userGender, Gender targetGender)
    {
        var script = new Captivate();
        var move = Substitute.For<IExecutingMove>();
        var user = Substitute.For<IPokemon>();
        user.Gender.Returns(userGender);
        move.User.Returns(user);

        var target = Substitute.For<IPokemon>();
        target.Gender.Returns(targetGender);
        var hitData = Substitute.For<IHitData>();
        move.GetHitData(target, 0).Returns(hitData);

        return (script, move, user, target, hitData);
    }

    /// <summary>
    /// Helper that checks whether a stat boost change was applied to the given Pokémon.
    /// </summary>
    private static bool ReceivedStatBoost(IPokemon pokemon, Statistic stat, sbyte amount) =>
        pokemon.ReceivedCalls().Any(c => c.GetMethodInfo().Name == "ChangeStatBoost" &&
                                         (Statistic)c.GetArguments()[0]! == stat &&
                                         (sbyte)c.GetArguments()[1]! == amount);

    /// <summary>
    /// Bulbapedia: "Captivate lowers the Special Attack stat of the target by two stages, if it is the
    /// opposite gender of the user."
    /// </summary>
    [Test, Arguments(Gender.Male, Gender.Female), Arguments(Gender.Female, Gender.Male)]
    public async Task OnSecondaryEffect_OppositeGender_LowersTargetSpecialAttackByTwoStages(Gender userGender,
        Gender targetGender)
    {
        // Arrange
        var (script, move, _, target, hitData) = CreateTestSetup(userGender, targetGender);

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

        // Assert
        await Assert.That(ReceivedStatBoost(target, Statistic.SpecialAttack, -2)).IsTrue();
        hitData.DidNotReceive().Fail();
    }

    /// <summary>
    /// Bulbapedia: Captivate only affects a target "if it is the opposite gender of the user" — against a
    /// same-gender target the move fails and no stat is lowered.
    /// </summary>
    [Test, Arguments(Gender.Male), Arguments(Gender.Female)]
    public async Task OnSecondaryEffect_SameGender_Fails(Gender gender)
    {
        // Arrange
        var (script, move, _, target, hitData) = CreateTestSetup(gender, gender);

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

        // Assert
        hitData.Received(1).Fail();
        await Assert.That(target.ReceivedCalls().Any(c => c.GetMethodInfo().Name == "ChangeStatBoost")).IsFalse();
    }

    /// <summary>
    /// Bulbapedia: "Pokémon that are gender-unknown [...] cannot be affected."
    /// </summary>
    [Test]
    public async Task OnSecondaryEffect_GenderlessTarget_Fails()
    {
        // Arrange
        var (script, move, _, target, hitData) = CreateTestSetup(Gender.Male, Gender.Genderless);

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

        // Assert
        hitData.Received(1).Fail();
        await Assert.That(target.ReceivedCalls().Any(c => c.GetMethodInfo().Name == "ChangeStatBoost")).IsFalse();
    }

    /// <summary>
    /// Bulbapedia: "those with the Oblivious ability cannot be affected."
    /// </summary>
    [Test]
    public async Task OnSecondaryEffect_TargetHasOblivious_Fails()
    {
        // Arrange
        var (script, move, _, target, hitData) = CreateTestSetup(Gender.Male, Gender.Female);
        var ability = Substitute.For<IAbility>();
        ability.Name.Returns(new StringKey("oblivious"));
        target.ActiveAbility.Returns(ability);

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

        // Assert
        hitData.Received(1).Fail();
        await Assert.That(target.ReceivedCalls().Any(c => c.GetMethodInfo().Name == "ChangeStatBoost")).IsFalse();
    }

    /// <summary>
    /// Bulbapedia: "The move fails automatically if used by a gender-unknown Pokémon." A genderless user
    /// against a gendered target must fail rather than lower the target's Special Attack.
    /// </summary>
    [Test]
    public async Task OnSecondaryEffect_GenderlessUser_Fails()
    {
        // Arrange
        var (script, move, _, target, hitData) = CreateTestSetup(Gender.Genderless, Gender.Female);

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

        // Assert
        hitData.Received(1).Fail();
        await Assert.That(target.ReceivedCalls().Any(c => c.GetMethodInfo().Name == "ChangeStatBoost")).IsFalse();
    }
}