using MultiversalDiplomacy.Adjudicate.Decision;
using MultiversalDiplomacy.Model;

using NUnit.Framework;

using static MultiversalDiplomacyTests.Adjudicator;

namespace MultiversalDiplomacyTests;

public class MovementAdjudicatorTest
{
    [Test]
    public void Validation_ValidHold()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").Holds().GetReference(out var order);

        setup.ValidateOrders();

        Assert.That(order, Is.Valid, "Unexpected validation result");
        Assert.That(order.Replacement, Is.Null, "Unexpected order replacement");
    }

    [Test]
    public void Validation_ValidMove()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").MovesTo("Tyr").GetReference(out var order);

        setup.ValidateOrders();

        Assert.That(order, Is.Valid, "Unexpected validation result");
        Assert.That(order.Replacement, Is.Null, "Unexpected order replacement");
    }

    [Test]
    public void Validation_ValidConvoy()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Fleet("Nth").Convoys.Army("Hol").To("Lon").GetReference(out var order);

        setup.ValidateOrders();

        Assert.That(order, Is.Valid, "Unexpected validation result");
        Assert.That(order.Replacement, Is.Null, "Unexpected order replacement");
    }

    [Test]
    public void Validation_ValidSupportHold()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").Supports.Army("Kie").Hold().GetReference(out var order);

        setup.ValidateOrders();

        Assert.That(order, Is.Valid, "Unexpected validation result");
        Assert.That(order.Replacement, Is.Null, "Unexpected order replacement");
    }

    [Test]
    public void Validation_ValidSupportMove()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").Supports.Army("Kie").MoveTo("Ber").GetReference(out var order);

        setup.ValidateOrders();

        Assert.That(order, Is.Valid, "Unexpected validation result");
        Assert.That(order.Replacement, Is.Null, "Unexpected order replacement");
    }

    [Test]
    public void Adjudication_Hold()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").Holds().GetReference(out var order);

        setup.ValidateOrders();
        setup.AdjudicateOrders(MovementPhase);

        var adjMun = order.Adjudications;
        Assert.That(adjMun.All(adj => adj.Resolved), Is.True);
        Assert.That(adjMun.OfType<IsDislodged>().Count(), Is.EqualTo(1));

        IsDislodged isDislodged = adjMun.OfType<IsDislodged>().Single();
        Assert.That(isDislodged.Order, Is.EqualTo(order.Order));
        Assert.That(isDislodged.Outcome, Is.False);
        Assert.That(isDislodged.Incoming, Is.Empty);
    }

    [Test]
    public void Adjudication_Move()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").MovesTo("Tyr").GetReference(out var order);

        setup.ValidateOrders();
        setup.AdjudicateOrders();

        var adjMun = order.Adjudications;
        Assert.That(adjMun.All(adj => adj.Resolved), Is.True);
        Assert.That(adjMun.OfType<IsDislodged>().Count(), Is.EqualTo(1));
        Assert.That(adjMun.OfType<DoesMove>().Count(), Is.EqualTo(1));

        IsDislodged dislodged = adjMun.OfType<IsDislodged>().Single();
        Assert.That(dislodged.Order, Is.EqualTo(order.Order));
        Assert.That(dislodged.Outcome, Is.False);

        DoesMove moves = adjMun.OfType<DoesMove>().Single();
        Assert.That(moves.Order, Is.EqualTo(order.Order));
        Assert.That(moves.Outcome, Is.True);
        Assert.That(moves.Competing, Is.Empty);
        Assert.That(moves.OpposingMove, Is.Null);
    }

    [Test]
    public void Adjudication_Support()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").MovesTo("Tyr").GetReference(out var move)
            .Army("Boh").Supports.Army("Mun").MoveTo("Tyr").GetReference(out var support);

        setup.ValidateOrders();
        setup.AdjudicateOrders();

        var adjBoh = support.Adjudications;
        Assert.That(adjBoh.All(adj => adj.Resolved), Is.True);
        Assert.That(adjBoh.OfType<IsDislodged>().Count(), Is.EqualTo(1));
        Assert.That(adjBoh.OfType<GivesSupport>().Count(), Is.EqualTo(1));

        IsDislodged dislodgeBoh = adjBoh.OfType<IsDislodged>().Single();
        Assert.That(dislodgeBoh.Order, Is.EqualTo(support.Order));
        Assert.That(dislodgeBoh.Outcome, Is.False);

        GivesSupport supportBoh = adjBoh.OfType<GivesSupport>().Single();
        Assert.That(supportBoh.Order, Is.EqualTo(support.Order));
        Assert.That(supportBoh.Outcome, Is.True);

        var adjMun = move.Adjudications;
        Assert.That(adjMun.All(adj => adj.Resolved), Is.True);
        Assert.That(adjMun.OfType<AttackStrength>().Count(), Is.EqualTo(1));

        AttackStrength attackMun = adjMun.OfType<AttackStrength>().Single();
        Assert.That(attackMun.Order, Is.EqualTo(move.Order));
        Assert.That(attackMun.MinValue, Is.EqualTo(2));
        Assert.That(attackMun.MaxValue, Is.EqualTo(2));
    }

    [Test]
    public void Update_SingleHold()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup["Germany"]
            .Army("Mun").Holds().GetReference(out var mun);

        setup.ValidateOrders();
        Assert.That(mun, Is.Valid);

        setup.AdjudicateOrders();
        Assert.That(mun, Is.NotDislodged);

        World updated = setup.UpdateWorld();

        // Confirm the unit was created
        Assert.That(updated.Units.Count, Is.EqualTo(2));
        Unit second = updated.Units.OrderBy(u => -u.Season.Turn).First();
        Assert.That(second.Location, Is.EqualTo(mun.Order.Unit.Location));
        Assert.That(second.Season.Timeline, Is.EqualTo(mun.Order.Unit.Season.Timeline));

        // Confirm that the unit's season exists
        CollectionAssert.Contains(updated.Timelines.Pasts.Keys, second.Season.Key, "Season was not added");
        CollectionAssert.DoesNotContain(updated.Timelines.Pasts.Values, second.Season.Key, "Season should not have a future");

    }

    [Test]
    public void Update_DoubleHold()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup[("a", 0)]
            .GetReference(out Season s1)
            ["Germany"]
                .Army("Mun").Holds().GetReference(out var mun1);

        Assert.That(mun1.Order.Unit.Season, Is.EqualTo(s1));
        setup.ValidateOrders();
        Assert.That(mun1, Is.Valid);
        setup.AdjudicateOrders();
        Assert.That(mun1, Is.NotDislodged);
        World updated = setup.UpdateWorld();

        // Confirm the future was created
        Season s2 = new(s1.Timeline, s1.Turn + 1);
        Assert.That(updated.Timelines.Pasts[s2.Key], Is.EqualTo(s1));
        Assert.That(updated.Timelines.GetFutures(s2), Is.Empty);
        Assert.That(s2.Timeline, Is.EqualTo(s1.Timeline));
        Assert.That(s2.Turn, Is.EqualTo(s1.Turn + 1));

        // Confirm the unit was created in the future
        Unit u2 = updated.GetUnitAt("Mun", s2);
        Assert.That(updated.Units.Count, Is.EqualTo(2));
        Assert.That(u2.Key, Is.Not.EqualTo(mun1.Order.Unit.Key));
        Assert.That(u2.Season, Is.EqualTo(s2));

        setup[("a", 1)]
            ["Germany"]
                .Army("Mun").Holds().GetReference(out var mun2);

        // Validate the second set of orders
        var validations = setup.ValidateOrders();
        Assert.That(validations.Count, Is.EqualTo(1));
        Assert.That(mun2, Is.Valid);

        // Adjudicate the second set of orders
        var adjudications = setup.AdjudicateOrders();
        Assert.That(mun2, Is.NotDislodged);

        // Update the world again
        updated = setup.UpdateWorld();
        Season s3 = new(s2.Timeline, s2.Turn + 1);
        Unit u3 = updated.GetUnitAt("Mun", s3);
    }

    [Test]
    public void Update_DoubleMove()
    {
        TestCaseBuilder setup = new(World.WithStandardMap(), MovementPhase);
        setup[("a", 0)]
            .GetReference(out Season s1)
            ["Germany"]
                .Army("Mun").MovesTo("Tyr").GetReference(out var mun1);

        Assert.That(mun1.Order.Unit.Season, Is.EqualTo(s1));
        setup.ValidateOrders();
        Assert.That(mun1, Is.Valid);
        setup.AdjudicateOrders();
        Assert.That(mun1, Is.Victorious);
        World updated = setup.UpdateWorld();

        // Confirm the future was created
        Season s2 = new(s1.Timeline, s1.Turn + 1);
        Assert.That(updated.Timelines.Pasts[s2.Key], Is.EqualTo(s1));
        Assert.That(updated.Timelines.GetFutures(s2), Is.Empty);
        Assert.That(s2.Timeline, Is.EqualTo(s1.Timeline));
        Assert.That(s2.Turn, Is.EqualTo(s1.Turn + 1));

        // Confirm the unit was created in the future
        Unit u2 = updated.GetUnitAt("Tyr", s2);
        Assert.That(updated.Units.Count, Is.EqualTo(2));
        Assert.That(u2.Key, Is.Not.EqualTo(mun1.Order.Unit.Key));
        Assert.That(u2.Season, Is.EqualTo(s2));

        setup[("a", 1)]
            ["Germany"]
                .Army("Tyr").MovesTo("Mun").GetReference(out var tyr2);

        // Validate the second set of orders
        var validations = setup.ValidateOrders();
        Assert.That(validations.Count, Is.EqualTo(1));
        Assert.That(tyr2, Is.Valid);

        // Adjudicate the second set of orders
        var adjudications = setup.AdjudicateOrders();
        Assert.That(tyr2, Is.Victorious);

        // Update the world again
        updated = setup.UpdateWorld();
        Season s3 = new(s2.Timeline, s2.Turn + 1);
        Unit u3 = updated.GetUnitAt("Mun", s3);
    }
}