YachtDice/Assets/Scripts/Tests/Editor/InventoryModelTests.cs

using NUnit.Framework;
using YachtDice.Inventory;
using YachtDice.Modifiers.Definition;
using YachtDice.Modifiers.Runtime;

namespace YachtDice.Tests
{
    public class InventoryModelTests
    {
        private ModifierRegistry _registry;
        private InventoryModel _inventory;

        [SetUp]
        public void SetUp()
        {
            _registry = new ModifierRegistry(3);
            _inventory = new InventoryModel(_registry);
        }

        private ModifierDefinition CreateTestDef(string id = "test",
            bool hasLimitedUses = false, int maxUses = 0)
        {
            return ModifierDefinition.CreateForTest(id, null,
                hasLimitedUses: hasLimitedUses, maxUses: maxUses);
        }

        [Test]
        public void AddModifier_IncreasesCount()
        {
            _inventory.AddModifier(CreateTestDef());

            Assert.AreEqual(1, _inventory.OwnedModifiers.Count);
        }

        [Test]
        public void TryActivate_SucceedsWithinSlotLimit()
        {
            _inventory.AddModifier(CreateTestDef("a"));
            var mod = _inventory.OwnedModifiers[0];

            bool result = _inventory.TryActivate(mod);

            Assert.IsTrue(result);
            Assert.IsTrue(mod.IsActive);
            Assert.AreEqual(1, _inventory.ActiveCount);
        }

        [Test]
        public void TryActivate_FailsWhenSlotsFull()
        {
            for (int i = 0; i < 3; i++)
            {
                _inventory.AddModifier(CreateTestDef($"m{i}"));
                _inventory.TryActivate(_inventory.OwnedModifiers[i]);
            }

            _inventory.AddModifier(CreateTestDef("extra"));
            var extra = _inventory.OwnedModifiers[3];
            bool result = _inventory.TryActivate(extra);

            Assert.IsFalse(result);
            Assert.IsFalse(extra.IsActive);
            Assert.AreEqual(3, _inventory.ActiveCount);
        }

        [Test]
        public void Deactivate_FreesSlot()
        {
            _inventory.AddModifier(CreateTestDef());
            var mod = _inventory.OwnedModifiers[0];
            _inventory.TryActivate(mod);

            _inventory.Deactivate(mod);

            Assert.IsFalse(mod.IsActive);
            Assert.AreEqual(0, _inventory.ActiveCount);
        }

        [Test]
        public void RemoveModifier_DeactivatesAndRemoves()
        {
            _inventory.AddModifier(CreateTestDef());
            var mod = _inventory.OwnedModifiers[0];
            _inventory.TryActivate(mod);

            _inventory.RemoveModifier(mod);

            Assert.AreEqual(0, _inventory.OwnedModifiers.Count);
            Assert.AreEqual(0, _inventory.ActiveCount);
        }

        [Test]
        public void ConsumeUseOnActive_DecrementsUses()
        {
            _inventory.AddModifier(CreateTestDef("ltd", hasLimitedUses: true, maxUses: 3));
            var mod = _inventory.OwnedModifiers[0];
            _inventory.TryActivate(mod);

            _inventory.ConsumeUseOnActive();

            Assert.AreEqual(2, mod.RemainingUses);
        }

        [Test]
        public void ConsumeUseOnActive_RemovesExpired()
        {
            _inventory.AddModifier(CreateTestDef("ltd", hasLimitedUses: true, maxUses: 1));
            var mod = _inventory.OwnedModifiers[0];
            _inventory.TryActivate(mod);

            _inventory.ConsumeUseOnActive();

            Assert.AreEqual(0, _inventory.OwnedModifiers.Count);
        }

        [Test]
        public void ConsumeUseOnActive_IgnoresPermanent()
        {
            _inventory.AddModifier(CreateTestDef("perm"));
            var mod = _inventory.OwnedModifiers[0];
            _inventory.TryActivate(mod);

            _inventory.ConsumeUseOnActive();

            Assert.AreEqual(1, _inventory.OwnedModifiers.Count);
            Assert.IsTrue(mod.IsActive);
        }

        [Test]
        public void GetActiveModifierDefinitions_ReturnsOnlyActive()
        {
            _inventory.AddModifier(CreateTestDef("a"));
            _inventory.AddModifier(CreateTestDef("b"));
            _inventory.TryActivate(_inventory.OwnedModifiers[0]);

            var active = _inventory.GetActiveModifierDefinitions();

            Assert.AreEqual(1, active.Count);
        }

        [Test]
        public void SetMaxActiveSlots_AllowsExpansion()
        {
            _inventory.SetMaxActiveSlots(10);

            Assert.AreEqual(10, _inventory.MaxActiveSlots);
        }

        [Test]
        public void OnActiveModifiersChanged_FiredOnActivate()
        {
            bool fired = false;
            _inventory.OnActiveModifiersChanged += _ => fired = true;
            _inventory.AddModifier(CreateTestDef());

            _inventory.TryActivate(_inventory.OwnedModifiers[0]);

            Assert.IsTrue(fired);
        }
    }
}