ModernUO/Projects/UOContent.Tests/Tests/Mobiles/AI/PetRetaliationTests.cs

using System;
using System.Collections.Generic;
using Server;
using Server.Mobiles;
using Xunit;

namespace UOContent.Tests.Mobiles.AI;

// Publish 51 (26 March 2008): a pet told to follow, come, stay or stop "will not attack
// anything, even if it is attacked". Guard and attack are unaffected.
[Collection("Sequential UOContent Tests")]
public class PetRetaliationTests : IDisposable
{
    private readonly List<Mobile> _created = new();

    public void Dispose()
    {
        foreach (var m in _created)
        {
            m?.Delete();
        }

        _created.Clear();
    }

    private sealed class StandDownPet : PetTestStub
    {
        public override bool StandsDownOnCommand => true;
    }

    private sealed class FightBackPet : PetTestStub
    {
        public override bool StandsDownOnCommand => false;
    }

    private (PlayerMobile master, T pet) Spawn<T>() where T : BaseCreature, new()
    {
        var master = new PlayerMobile(World.NewMobile);
        master.DefaultMobileInit();
        master.MoveToWorld(new Point3D(1000, 1000, 0), Map.Felucca);
        _created.Add(master);

        var pet = new T();
        pet.MoveToWorld(new Point3D(1001, 1000, 0), Map.Felucca);
        pet.SetControlMaster(master);
        pet.AIObject.AITimer?.Stop();
        _created.Add(pet);

        return (master, pet);
    }

    private void Order(BaseCreature pet, Mobile master, OrderType order)
    {
        if (order == OrderType.None) // Publish 51's "stop": stops, may wander, will not attack
        {
            pet.ControlTarget = master;
            pet.ControlOrder = OrderType.Follow;
            pet.ControlOrder = OrderType.Stop;
            return;
        }

        pet.ControlTarget = master;
        pet.ControlOrder = order;
    }

    private BaseCreature Attack(BaseCreature pet)
    {
        var attacker = new PetTestStub();
        attacker.MoveToWorld(new Point3D(1002, 1000, 0), pet.Map);
        _created.Add(attacker);
        pet.AIObject.AITimer?.Stop();

        attacker.Combatant = pet; // a mob starts attacking the pet
        return attacker;
    }

    [Theory]
    [InlineData(OrderType.Follow)]
    [InlineData(OrderType.Come)]
    [InlineData(OrderType.Stay)]
    [InlineData(OrderType.None)] // stopped
    public void StandDownOrder_IgnoresTheAttacker(OrderType order)
    {
        var (master, pet) = Spawn<StandDownPet>();
        Order(pet, master, order);
        var resting = pet.ControlOrder;

        Attack(pet);

        Assert.Equal(resting, pet.ControlOrder); // never converts to Attack
        Assert.Null(pet.Combatant);
        Assert.False(pet.Warmode);
    }

    [Theory]
    [InlineData(OrderType.Follow)]
    [InlineData(OrderType.Come)]
    [InlineData(OrderType.Stay)]
    [InlineData(OrderType.None)]
    public void WithoutStandDown_TheSameOrdersRetaliate(OrderType order)
    {
        var (master, pet) = Spawn<FightBackPet>();
        Order(pet, master, order);

        var attacker = Attack(pet);

        Assert.Equal(OrderType.Attack, pet.ControlOrder);
        Assert.Same(attacker, pet.Combatant);
    }

    // No damage callback may put a stand-down pet back into combat behind the policy's back.
    [Theory]
    [InlineData(Expansion.AOS, false)]
    [InlineData(Expansion.AOS, true)]
    [InlineData(Expansion.ML, false)]
    [InlineData(Expansion.ML, true)]
    public void StandDownPet_StaysDown_ThroughRepeatedDamage(Expansion era, bool spellDamage)
    {
        var previous = Core.Expansion;

        try
        {
            Core.Expansion = era;
            var (master, pet) = Spawn<StandDownPet>();
            Order(pet, master, OrderType.Follow);
            var attacker = Attack(pet);

            Assert.Equal(OrderType.Follow, pet.ControlOrder); // the initial aggression stood down

            for (var i = 0; i < 500 && pet.ControlOrder == OrderType.Follow; i++)
            {
                if (spellDamage)
                {
                    pet.OnDamagedBySpell(attacker, 1);
                }
                else
                {
                    pet.OnDamage(1, attacker, false);
                }
            }

            Assert.Equal(OrderType.Follow, pet.ControlOrder);
            Assert.Null(pet.Combatant);
        }
        finally
        {
            Core.Expansion = previous;
        }
    }

    // "Guard: the pet should guard as it does currently."
    [Fact]
    public void GuardingPet_StillFights_UnderStandDown()
    {
        var (master, pet) = Spawn<StandDownPet>();
        Order(pet, master, OrderType.Guard);

        var attacker = Attack(pet);

        Assert.Equal(OrderType.Guard, pet.ControlOrder);
        Assert.Same(attacker, pet.Combatant);
        Assert.True(pet.Warmode);
    }

    private sealed class UncommandablePet : PetTestStub
    {
        public override bool StandsDownOnCommand => true;
        public override bool Commandable => false;
    }

    // The publish speaks of commanded pets. A creature nobody can give an order to was never
    // told anything, yet it rests on None (wild, uncontrolled summon) or a system-issued Follow
    // (familiar, escortee) - the same orders a pet stands down on. It must still fight back.
    [Fact]
    public void WildCreature_Retaliates_UnderStandDown()
    {
        var creature = new StandDownPet();
        creature.MoveToWorld(new Point3D(1001, 1000, 0), Map.Felucca);
        creature.AIObject.AITimer?.Stop();
        _created.Add(creature);

        Assert.Equal(OrderType.None, creature.ControlOrder);

        var attacker = Attack(creature);

        Assert.Same(attacker, creature.Combatant);
        Assert.True(creature.Warmode);
    }

    [Fact] // energy vortex, blade spirits: Summoned with a SummonMaster, never Controlled
    public void UncontrolledSummon_Retaliates_UnderStandDown()
    {
        var caster = new PlayerMobile(World.NewMobile);
        caster.DefaultMobileInit();
        caster.MoveToWorld(new Point3D(1000, 1000, 0), Map.Felucca);
        _created.Add(caster);

        var summon = new StandDownPet();
        summon.Summoned = true;
        summon.SummonMaster = caster;
        summon.MoveToWorld(new Point3D(1001, 1000, 0), Map.Felucca);
        summon.AIObject.AITimer?.Stop();
        _created.Add(summon);

        Assert.False(summon.Controlled);
        Assert.Equal(OrderType.None, summon.ControlOrder);

        var attacker = Attack(summon);

        Assert.Same(attacker, summon.Combatant);
        Assert.True(summon.Warmode);
    }

    [Fact] // familiar, escortee: Controlled with a master, but not Commandable
    public void UncommandableCreature_Retaliates_UnderStandDown()
    {
        var (master, familiar) = Spawn<UncommandablePet>();
        familiar.ControlTarget = master;
        familiar.ControlOrder = OrderType.Follow; // system-issued, not a command

        var attacker = Attack(familiar);

        Assert.Same(attacker, familiar.Combatant);
        Assert.True(familiar.Warmode);
    }
}