using Craftimizer.Simulator.Actions;

namespace Craftimizer.Test.Solver;

[TestClass]
public class ActionSetTests
{
    [TestMethod]
    public void TestAcceptedActions()
    {
        var actions = ActionSet.AcceptedActions;
        var lut = ActionSet.AcceptedActionsLUT;

        Assert.IsTrue(actions.Length <= 32);
        foreach (var i in Enum.GetValues<ActionType>())
        {
            var idx = lut[(byte)i];
            if (idx != -1)
                Assert.AreEqual(i, actions[idx]);
        }
    }

    [TestMethod]
    public void TestSize()
    {
        var set = new ActionSet();
        Assert.IsTrue(set.IsEmpty);
        Assert.AreEqual(0, set.Count);

        set.AddAction(ActionType.BasicSynthesis);
        set.AddAction(ActionType.WasteNot2);

        Assert.AreEqual(2, set.Count);
        Assert.IsFalse(set.IsEmpty);

        set.RemoveAction(ActionType.BasicSynthesis);
        set.RemoveAction(ActionType.WasteNot2);

        Assert.IsTrue(set.IsEmpty);
        Assert.AreEqual(0, set.Count);
    }

    [TestMethod]
    public void TestAddRemove()
    {
        var set = new ActionSet();

        Assert.IsTrue(set.AddAction(ActionType.BasicSynthesis));
        Assert.IsFalse(set.AddAction(ActionType.BasicSynthesis));

        Assert.IsTrue(set.RemoveAction(ActionType.BasicSynthesis));
        Assert.IsFalse(set.RemoveAction(ActionType.BasicSynthesis));

        Assert.IsTrue(set.AddAction(ActionType.BasicSynthesis));
        Assert.IsTrue(set.AddAction(ActionType.WasteNot2));

        Assert.IsTrue(set.RemoveAction(ActionType.BasicSynthesis));
        Assert.IsTrue(set.RemoveAction(ActionType.WasteNot2));
    }

    [TestMethod]
    public void TestHasAction()
    {
        var set = new ActionSet();

        set.AddAction(ActionType.BasicSynthesis);

        Assert.IsTrue(set.HasAction(ActionType.BasicSynthesis));
        Assert.IsFalse(set.HasAction(ActionType.WasteNot2));

        set.AddAction(ActionType.WasteNot2);
        Assert.IsTrue(set.HasAction(ActionType.BasicSynthesis));
        Assert.IsTrue(set.HasAction(ActionType.WasteNot2));

        set.RemoveAction(ActionType.BasicSynthesis);
        Assert.IsFalse(set.HasAction(ActionType.BasicSynthesis));
        Assert.IsTrue(set.HasAction(ActionType.WasteNot2));
    }

    [TestMethod]
    public void TestElementAt()
    {
        var set = new ActionSet();

        set.AddAction(ActionType.BasicSynthesis);
        set.AddAction(ActionType.ByregotsBlessing);
        set.AddAction(ActionType.DelicateSynthesis);
        set.AddAction(ActionType.FocusedTouch);

        Assert.AreEqual(4, set.Count);

        Assert.AreEqual(ActionType.DelicateSynthesis, set.ElementAt(0));
        Assert.AreEqual(ActionType.FocusedTouch, set.ElementAt(1));
        Assert.AreEqual(ActionType.ByregotsBlessing, set.ElementAt(2));
        Assert.AreEqual(ActionType.BasicSynthesis, set.ElementAt(3));

        set.RemoveAction(ActionType.FocusedTouch);

        Assert.AreEqual(3, set.Count);

        Assert.AreEqual(ActionType.DelicateSynthesis, set.ElementAt(0));
        Assert.AreEqual(ActionType.ByregotsBlessing, set.ElementAt(1));
        Assert.AreEqual(ActionType.BasicSynthesis, set.ElementAt(2));
    }

    [TestMethod]
    public void TestRandomIndex()
    {
#if IS_DETERMINISTIC
        Assert.Inconclusive("Craftimizer is currently built for determinism; all random actions are not actually random.");
#endif

        var actions = new[]
        {
            ActionType.BasicTouch,
            ActionType.BasicSynthesis,
            ActionType.GreatStrides,
            ActionType.TrainedFinesse,
        };

        var set = new ActionSet();
        foreach(var action in actions)
            set.AddAction(action);

        var counts = new Dictionary<ActionType, int>();
        var rng = new Random(0);
        for (var i = 0; i < 100; i++)
        {
            var action = set.SelectRandom(rng);

            CollectionAssert.Contains(actions, action);

            counts[action] = counts.GetValueOrDefault(action) + 1;
        }

        foreach (var action in actions)
            Assert.IsTrue(counts.GetValueOrDefault(action) > 0);
    }
}