More unit tests, fixes

Plugins/PkmnLib.Plugin.Gen7.Tests/Scripts/Moves/BlockTests.cs

@@ -0,0 +1,119 @@
+using PkmnLib.Dynamic.Models;
+using PkmnLib.Dynamic.Models.Choices;
+using PkmnLib.Dynamic.ScriptHandling;
+using PkmnLib.Dynamic.ScriptHandling.Registry;
+using PkmnLib.Plugin.Gen7.Scripts.Moves;
+using PkmnLib.Plugin.Gen7.Scripts.Pokemon;
+
+namespace PkmnLib.Plugin.Gen7.Tests.Scripts.Moves;
+
+/// <summary>
+/// Tests for the <see cref="Block"/> move script.
+/// Gen VII Bulbapedia behavior: "Block prevents the target from switching out or fleeing (including via
+/// Teleport)." From Generation VI onward: "Block no longer affects Ghost-type Pokémon."
+/// </summary>
+public class BlockTests
+{
+    private static (Block script, IExecutingMove move, IPokemon target, ScriptSet targetVolatile) CreateTestSetup()
+    {
+        var script = new Block();
+        var move = Substitute.For<IExecutingMove>();
+        var user = Substitute.For<IPokemon>();
+        move.User.Returns(user);
+
+        var target = Substitute.For<IPokemon>();
+        // Use a real script set so the volatile script added by Block can be inspected afterwards.
+        var targetVolatile = new ScriptSet(target);
+        target.Volatile.Returns(targetVolatile);
+        target.GetScripts().Returns(_ => new ScriptIterator(new List<IEnumerable<ScriptContainer>>()));
+        move.GetHitData(target, 0).Returns(Substitute.For<IHitData>());
+
+        return (script, move, target, targetVolatile);
+    }
+
+    /// <summary>
+    /// Bulbapedia: "Block prevents the target from switching out or fleeing (including via Teleport)."
+    /// Hitting the target attaches the <see cref="BlockEffect"/> volatile script to it.
+    /// </summary>
+    [Test]
+    public async Task OnSecondaryEffect_Hit_AddsBlockEffectToTarget()
+    {
+        // Arrange
+        var (script, move, target, targetVolatile) = CreateTestSetup();
+
+        // Act
+        script.OnSecondaryEffect(move, target, 0);
+
+        // Assert
+        await Assert.That(targetVolatile.Contains(ScriptUtils.ResolveName<BlockEffect>())).IsTrue();
+    }
+
+    /// <summary>
+    /// Bulbapedia: "Block prevents the target from switching out".
+    /// The <see cref="BlockEffect"/> applied by the move prevents the target's switch choices through
+    /// <see cref="BlockEffect.PreventSelfSwitch"/>.
+    /// </summary>
+    [Test]
+    public async Task OnSecondaryEffect_AppliedEffect_PreventsTargetFromSwitchingOut()
+    {
+        // Arrange
+        var (script, move, target, targetVolatile) = CreateTestSetup();
+        script.OnSecondaryEffect(move, target, 0);
+        await Assert.That(targetVolatile.TryGet<BlockEffect>(out var effect)).IsTrue();
+
+        // Act
+        var prevent = false;
+        effect!.PreventSelfSwitch(Substitute.For<ISwitchChoice>(), ref prevent);
+
+        // Assert
+        await Assert.That(prevent).IsTrue();
+    }
+
+    /// <summary>
+    /// Bulbapedia: "Block prevents the target from [...] fleeing (including via Teleport)."
+    /// The <see cref="BlockEffect"/> applied by the move prevents the target's flee choices through
+    /// <see cref="BlockEffect.PreventSelfRunAway"/>.
+    /// </summary>
+    [Test]
+    public async Task OnSecondaryEffect_AppliedEffect_PreventsTargetFromFleeing()
+    {
+        // Arrange
+        var (script, move, target, targetVolatile) = CreateTestSetup();
+        script.OnSecondaryEffect(move, target, 0);
+        await Assert.That(targetVolatile.TryGet<BlockEffect>(out var effect)).IsTrue();
+
+        // Act
+        var prevent = false;
+        effect!.PreventSelfRunAway(Substitute.For<IFleeChoice>(), ref prevent);
+
+        // Assert
+        await Assert.That(prevent).IsTrue();
+    }
+
+    /// <summary>
+    /// Bulbapedia (Generation VI onward): "Block no longer affects Ghost-type Pokémon."
+    /// Using Block on a Ghost-type target should not trap it, so no <see cref="BlockEffect"/> may be added.
+    /// </summary>
+    [Test]
+    public async Task OnSecondaryEffect_GhostTypeTarget_DoesNotTrapTarget()
+    {
+        // Arrange
+        var (script, move, target, targetVolatile) = CreateTestSetup();
+        var library = LibraryHelpers.LoadLibrary();
+        await Assert.That(library.StaticLibrary.Types.TryGetTypeIdentifier("ghost", out var ghostType)).IsTrue();
+        target.Types.Returns(new[] { ghostType });
+
+        var battle = Substitute.For<IBattle>();
+        battle.Library.Returns(library);
+        var battleData = Substitute.For<IPokemonBattleData>();
+        battleData.Battle.Returns(battle);
+        move.User.BattleData.Returns(battleData);
+        target.BattleData.Returns(battleData);
+
+        // Act
+        script.OnSecondaryEffect(move, target, 0);
+
+        // Assert
+        await Assert.That(targetVolatile.Contains(ScriptUtils.ResolveName<BlockEffect>())).IsFalse();
+    }
+}