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

using NUnit.Framework;
using UnityEngine;
using YachtDice.Categories;
using YachtDice.Modifiers.Core;
using YachtDice.Modifiers.Definition;
using YachtDice.Modifiers.Effects;
using YachtDice.Modifiers.Runtime;

namespace YachtDice.Tests
{
    public class ModifierEffectTests
    {
        private CategoryDefinition _testCategory;

        [SetUp]
        public void SetUp()
        {
            _testCategory = SumAllCategory.CreateForTest("chance", "Шанс");
        }

        [TearDown]
        public void TearDown()
        {
            Object.DestroyImmediate(_testCategory);
        }

        private ModifierInstance CreateInstance(string id = "test")
        {
            var def = ModifierDefinition.CreateForTest(id, null);
            return new ModifierInstance(def);
        }

        private ModifierContext CreateContext(int baseScore, int[] dice)
        {
            return new ModifierContext
            {
                BaseScore = baseScore,
                DiceValues = dice,
                Category = _testCategory,
            };
        }

        // ── AddPerDiceEffect ─────────────────────────────────────────

        /// <summary>
        /// Эффект добавляет бонус за каждую кость с целевым значением.
        /// </summary>
        [Test]
        public void AddPerDiceEffect_CountsMatchingDice()
        {
            var effect = AddPerDiceEffect.CreateForTest(10, targetDiceValue: 1);
            var ctx = CreateContext(5, new[] { 1, 1, 3, 4, 1 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(30, ctx.FlatBonus); // 10 * 3 matching dice
        }

        /// <summary>
        /// При целевом значении 0 бонус начисляется за все кости.
        /// </summary>
        [Test]
        public void AddPerDiceEffect_ZeroTarget_CountsAllDice()
        {
            var effect = AddPerDiceEffect.CreateForTest(2, targetDiceValue: 0);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(10, ctx.FlatBonus); // 2 * 5 dice
        }

        /// <summary>
        /// При отсутствии совпадений по значению бонус остается нулевым.
        /// </summary>
        [Test]
        public void AddPerDiceEffect_NoMatches_ZeroBonus()
        {
            var effect = AddPerDiceEffect.CreateForTest(10, targetDiceValue: 6);
            var ctx = CreateContext(5, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(0, ctx.FlatBonus);
        }

        /// <summary>
        /// Бонус AddPerDice масштабируется количеством стаков модификатора.
        /// </summary>
        [Test]
        public void AddPerDiceEffect_ScalesWithStacks()
        {
            var effect = AddPerDiceEffect.CreateForTest(10, targetDiceValue: 1);
            var ctx = CreateContext(5, new[] { 1, 1, 3, 4, 1 });
            var inst = CreateInstance();
            inst.Stacks = 2;

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(60, ctx.FlatBonus); // 10 * 3 * 2 stacks
        }

        // ── AddFlatScoreEffect ──────────────────────────────────────

        /// <summary>
        /// Эффект AddFlatScore добавляет фиксированный бонус к очкам.
        /// </summary>
        [Test]
        public void AddFlatScoreEffect_AddsFlat()
        {
            var effect = AddFlatScoreEffect.CreateForTest(15);
            var ctx = CreateContext(25, new[] { 3, 3, 2, 2, 2 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(15, ctx.FlatBonus);
        }

        /// <summary>
        /// Фиксированный бонус AddFlatScore масштабируется стаками.
        /// </summary>
        [Test]
        public void AddFlatScoreEffect_ScalesWithStacks()
        {
            var effect = AddFlatScoreEffect.CreateForTest(15);
            var ctx = CreateContext(25, new[] { 3, 3, 2, 2, 2 });
            var inst = CreateInstance();
            inst.Stacks = 3;

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(45, ctx.FlatBonus); // 15 * 3 stacks
        }

        // ── MultiplyPerDiceEffect ────────────────────────────────────

        /// <summary>
        /// MultiplyPerDice увеличивает множитель за каждое совпадение по кости.
        /// </summary>
        [Test]
        public void MultiplyPerDiceEffect_MultipliesPerMatch()
        {
            var effect = MultiplyPerDiceEffect.CreateForTest(2f, targetDiceValue: 6);
            var ctx = CreateContext(18, new[] { 6, 6, 6, 1, 2 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(8f, ctx.Multiplier, 0.001f); // 1 * 2 * 2 * 2 = 8
        }

        /// <summary>
        /// Без совпадений MultiplyPerDice не изменяет множитель.
        /// </summary>
        [Test]
        public void MultiplyPerDiceEffect_NoMatches_MultiplierUnchanged()
        {
            var effect = MultiplyPerDiceEffect.CreateForTest(3f, targetDiceValue: 6);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(1f, ctx.Multiplier);
        }

        // ── MultiplyScoreEffect ─────────────────────────────────────

        /// <summary>
        /// MultiplyScore применяет заданный множитель один раз.
        /// </summary>
        [Test]
        public void MultiplyScoreEffect_MultipliesOnce()
        {
            var effect = MultiplyScoreEffect.CreateForTest(1.5f);
            var ctx = CreateContext(50, new[] { 6, 6, 6, 6, 6 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(1.5f, ctx.Multiplier, 0.001f);
        }

        /// <summary>
        /// MultiplyScore масштабирует множитель в зависимости от стаков.
        /// </summary>
        [Test]
        public void MultiplyScoreEffect_ScalesWithStacks()
        {
            var effect = MultiplyScoreEffect.CreateForTest(2f);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();
            inst.Stacks = 3;

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            // Pow(2, 3) = 8
            Assert.AreEqual(8f, ctx.Multiplier, 0.001f);
        }

        // ── PostMultiplyEffect ──────────────────────────────────────

        /// <summary>
        /// PostMultiply изменяет пост-множитель контекста.
        /// </summary>
        [Test]
        public void PostMultiplyEffect_MultipliesPostMultiplier()
        {
            var effect = PostMultiplyEffect.CreateForTest(2f);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(2f, ctx.PostMultiplier, 0.001f);
        }

        // ── AddCurrencyEffect ───────────────────────────────────────

        /// <summary>
        /// AddCurrency увеличивает дельту валюты в контексте.
        /// </summary>
        [Test]
        public void AddCurrencyEffect_AddsToCurrencyDelta()
        {
            var effect = AddCurrencyEffect.CreateForTest(25);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(25, ctx.CurrencyDelta);
        }

        /// <summary>
        /// Эффект добавления валюты масштабируется стаками.
        /// </summary>
        [Test]
        public void AddCurrencyEffect_ScalesWithStacks()
        {
            var effect = AddCurrencyEffect.CreateForTest(25);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance();
            inst.Stacks = 2;

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(50, ctx.CurrencyDelta); // 25 * 2 stacks
        }

        // ── ConsumeChargeEffect ─────────────────────────────────────

        /// <summary>
        /// ConsumeCharge уменьшает число оставшихся использований ограниченного модификатора.
        /// </summary>
        [Test]
        public void ConsumeChargeEffect_DecrementsRemainingUses()
        {
            var effect = ConsumeChargeEffect.CreateForTest(1);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var def = ModifierDefinition.CreateForTest("limited", null,
                hasLimitedUses: true, maxUses: 3);
            var inst = new ModifierInstance(def);

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

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

        /// <summary>
        /// Постоянный модификатор игнорирует расход зарядов.
        /// </summary>
        [Test]
        public void ConsumeChargeEffect_IgnoresPermanent()
        {
            var effect = ConsumeChargeEffect.CreateForTest(1);
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            var inst = CreateInstance(); // permanent (no limited uses)

            effect.Apply(ctx, inst).GetAwaiter().GetResult();

            Assert.AreEqual(-1, inst.RemainingUses); // unchanged
        }

        // ── FinalScore Integration ──────────────────────────────────

        /// <summary>
        /// Финальный счет корректно учитывает базу, бонус и множители.
        /// </summary>
        [Test]
        public void FinalScore_CombinesBaseAndFlatAndMultiplier()
        {
            var ctx = CreateContext(10, new[] { 1, 2, 3, 4, 5 });
            ctx.FlatBonus = 5;
            ctx.Multiplier = 2f;
            ctx.PostMultiplier = 1.5f;

            // FinalScore = floor((10 + 5) * 2 * 1.5) = floor(45) = 45
            Assert.AreEqual(45, ctx.FinalScore);
        }
    }
}