// Copyright (C) Stichting Deltares 2016. All rights reserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// All names, logos, and references to "Deltares" are registered trademarks of
// Stichting Deltares and remain full property of Stichting Deltares at all times.
// All rights reserved.

using System.Collections.Generic;
using NUnit.Framework;
using Rhino.Mocks;
using Ringtoets.Common.Data.Calculation;

namespace Ringtoets.Common.Data.Test.Calculation
{
    [TestFixture]
    public class CalculationGroupExtensionsTest
    {
        [Test]
        public void GetCalculations_FromCalculationGroupWithoutChildren_ReturnEmpty()
        {
            // Setup
            CalculationGroup groupWithoutChildren = new CalculationGroup();

            // Call
            IEnumerable<ICalculation> result = groupWithoutChildren.GetCalculations();

            // Assert
            CollectionAssert.IsEmpty(result);
        }

        [Test]
        public void GetCalculations_FromCalculationGroupWithEmptyGroups_ReturnEmpty()
        {
            // Setup
            var rootGroup = new CalculationGroup();
            rootGroup.Children.Add(new CalculationGroup());
            rootGroup.Children.Add(new CalculationGroup());
            rootGroup.Children.Add(new CalculationGroup());

            // Call
            IEnumerable<ICalculation> result = rootGroup.GetCalculations();

            // Assert
            CollectionAssert.IsEmpty(result);
        }

        [Test]
        public void GetCalculations_FromCalculationGroupWithGroupsAndCalculations_ReturnAllCalculationsRecursiveslyInAnyOrder()
        {
            // Setup
            var mocks = new MockRepository();
            var calculation1 = mocks.StrictMock<ICalculation>();
            var calculation2 = mocks.StrictMock<ICalculation>();
            var calculation3 = mocks.StrictMock<ICalculation>();
            var calculation4 = mocks.StrictMock<ICalculation>();
            mocks.ReplayAll();

            var subsubGroup = new CalculationGroup();
            subsubGroup.Children.Add(calculation4);

            var subgroup1 = new CalculationGroup();
            subgroup1.Children.Add(calculation2);
            subgroup1.Children.Add(subsubGroup);

            var subgroup2 = new CalculationGroup();
            subgroup2.Children.Add(calculation3);

            var rootGroup = new CalculationGroup();
            rootGroup.Children.Add(subgroup1);
            rootGroup.Children.Add(calculation1);
            rootGroup.Children.Add(subgroup2);

            // Call
            IEnumerable<ICalculation> result = rootGroup.GetCalculations();

            // Assert
            var itemsThatShouldBeFound = new[]
            {
                calculation1,
                calculation2,
                calculation3,
                calculation4
            };
            CollectionAssert.AreEquivalent(itemsThatShouldBeFound, result);
        }

        [Test]
        public void ClearCalculationOutput_ForCalculationGroupWithGroupsAndCalculations_OutputOfRelevantCalculationsIsClearedAndObserversAreNotified()
        {
            // Setup
            var mocks = new MockRepository();
            var calculation1 = mocks.StrictMock<ICalculation>();
            var calculation2 = mocks.StrictMock<ICalculation>();
            var calculation3 = mocks.StrictMock<ICalculation>();
            var calculation4 = mocks.StrictMock<ICalculation>();

            calculation1.Expect(c => c.HasOutput).Return(true);
            calculation2.Expect(c => c.HasOutput).Return(true);
            calculation3.Expect(c => c.HasOutput).Return(false);
            calculation4.Expect(c => c.HasOutput).Return(false);
            calculation1.Expect(c => c.ClearOutput());
            calculation2.Expect(c => c.ClearOutput());
            calculation1.Expect(c => c.NotifyObservers());
            calculation2.Expect(c => c.NotifyObservers());

            mocks.ReplayAll();

            var subsubGroup = new CalculationGroup();
            subsubGroup.Children.Add(calculation4);

            var subgroup1 = new CalculationGroup();
            subgroup1.Children.Add(calculation2);
            subgroup1.Children.Add(subsubGroup);

            var subgroup2 = new CalculationGroup();
            subgroup2.Children.Add(calculation3);

            var rootGroup = new CalculationGroup();
            rootGroup.Children.Add(subgroup1);
            rootGroup.Children.Add(calculation1);
            rootGroup.Children.Add(subgroup2);

            // Call
            rootGroup.ClearCalculationOutput();

            // Assert
            mocks.VerifyAll();
        }

        [Test]
        public void HasOutput_ForCalculationGroupWithGroupsAndCalculationsWithoutOutput_ReturnsFalse()
        {
            // Setup
            var mocks = new MockRepository();
            var calculation1 = mocks.StrictMock<ICalculation>();
            var calculation2 = mocks.StrictMock<ICalculation>();
            var calculation3 = mocks.StrictMock<ICalculation>();
            var calculation4 = mocks.StrictMock<ICalculation>();

            calculation1.Expect(c => c.HasOutput).Return(false);
            calculation2.Expect(c => c.HasOutput).Return(false);
            calculation3.Expect(c => c.HasOutput).Return(false);
            calculation4.Expect(c => c.HasOutput).Return(false);

            mocks.ReplayAll();

            var subsubGroup = new CalculationGroup();
            subsubGroup.Children.Add(calculation4);

            var subgroup1 = new CalculationGroup();
            subgroup1.Children.Add(calculation2);
            subgroup1.Children.Add(subsubGroup);

            var subgroup2 = new CalculationGroup();
            subgroup2.Children.Add(calculation3);

            var rootGroup = new CalculationGroup();
            rootGroup.Children.Add(subgroup1);
            rootGroup.Children.Add(calculation1);
            rootGroup.Children.Add(subgroup2);

            // Call
            var hasOutput = rootGroup.HasOutput();

            // Assert
            Assert.IsFalse(hasOutput);
            mocks.VerifyAll();
        }

        [Test]
        public void HasOutput_ForCalculationGroupWithGroupsAndOneCalculationWithOutput_ReturnsTrue()
        {
            // Setup
            var mocks = new MockRepository();
            var calculation1 = mocks.StrictMock<ICalculation>();
            var calculation2 = mocks.StrictMock<ICalculation>();
            var calculation3 = mocks.StrictMock<ICalculation>();
            var calculation4 = mocks.StrictMock<ICalculation>();

            calculation1.Stub(c => c.HasOutput).Return(false);
            calculation2.Stub(c => c.HasOutput).Return(false);
            calculation3.Stub(c => c.HasOutput).Return(false);
            calculation4.Stub(c => c.HasOutput).Return(true);

            mocks.ReplayAll();

            var subsubGroup = new CalculationGroup();
            subsubGroup.Children.Add(calculation4);

            var subgroup1 = new CalculationGroup();
            subgroup1.Children.Add(calculation2);
            subgroup1.Children.Add(subsubGroup);

            var subgroup2 = new CalculationGroup();
            subgroup2.Children.Add(calculation3);

            var rootGroup = new CalculationGroup();
            rootGroup.Children.Add(subgroup1);
            rootGroup.Children.Add(calculation1);
            rootGroup.Children.Add(subgroup2);

            // Call
            var hasOutput = rootGroup.HasOutput();

            // Assert
            Assert.IsTrue(hasOutput);
            mocks.VerifyAll();
        }
    }
}