// This file (UnitRequiresNoMoreThanNUnitsRequirementFactoryTest.cs) is a part of the IBBoard.WarFoundry.API.Tests project and is copyright 2011 IBBoard
// 
// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.
using System;
using NUnit.Framework;
using IBBoard.WarFoundry.API.Objects.Mock;
using NUnit.Framework.SyntaxHelpers;
using IBBoard.WarFoundry.API.Objects;
using IBBoard.WarFoundry.API.Objects.Requirement;
using IBBoard.WarFoundry.API.Factories.Mock;

namespace IBBoard.WarFoundry.API.Factories.Requirement
{
	[TestFixture()]
	public class UnitRequiresNoMoreThanNUnitsRequirementFactoryTest
	{
		private static Race race;
		private static string typeID1 = "unit1";
		private static string typeID2 = "unit2";
		private static UnitType unitType1;
		private static UnitType unitType2;
		private static MockRaceFactory raceFactory = new MockRaceFactory();

		[TestFixtureSetUp()]
		public void Setup()
		{
			race = new MockRace();
			unitType1 = new UnitType(typeID1, "Unit 1", race);
			race.AddUnitType(unitType1);
			unitType2 = new UnitType(typeID2, "Unit 2", race);
			race.AddUnitType(unitType2);
		}

		[Test()]
		public void TestCreatesRequirement()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			Assert.That(factory.CreateRequirement(unitType2, typeID1, raceFactory), Is.Not.Null);
		}

		[Test()]
		public void TestCreatesCorrectRequirement()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1);
			IRequirement req = factory.CreateRequirement(unitType, typeID1, raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		public void TestCreatesCorrectRequirementForTwoIDs()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1);
			expectedReq.AddUnitTypeRequirement(unitType2);
			IRequirement req = factory.CreateRequirement(unitType, typeID1 + "|" + typeID2, raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		public void TestCreatesCorrectRequirementWithNumber()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1, 2);
			IRequirement req = factory.CreateRequirement(unitType, typeID1 + ":2", raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		public void TestCreatesCorrectRequirementForTwoIDsWithTwoNumbers()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1, 2);
			expectedReq.AddUnitTypeRequirement(unitType2, 3);
			IRequirement req = factory.CreateRequirement(unitType, typeID1 + ":2|" + typeID2 + ":3", raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		public void TestCreatesCorrectRequirementForTwoIDsWithOneNumber()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1, 2);
			expectedReq.AddUnitTypeRequirement(unitType2);
			IRequirement req = factory.CreateRequirement(unitType, typeID1 + ":2|" + typeID2, raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		public void TestCreatesCorrectRequirementForTwoIDsWithOtherNumber()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			UnitRequiresNoMoreThanNOfUnitTypeRequirement expectedReq = new UnitRequiresNoMoreThanNOfUnitTypeRequirement(unitType);
			expectedReq.AddUnitTypeRequirement(unitType1);
			expectedReq.AddUnitTypeRequirement(unitType2, 3);
			IRequirement req = factory.CreateRequirement(unitType, typeID1 + "|" + typeID2 + ":3", raceFactory);
			Assert.That(req, Is.EqualTo(expectedReq));
		}

		[Test()]
		[ExpectedException(typeof(InvalidRequirementException), ExpectedMessage="Invalid unit type 'fibble' for 'Requires at least N units' requirement")]
		public void TestExceptionsForInvalidID()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			factory.CreateRequirement(unitType, "fibble", raceFactory);
		}

		[Test()]
		[ExpectedException(typeof(InvalidRequirementException), ExpectedMessage="Invalid amount 'a' for unit type 'unit1' for 'Requires at least N units' requirement")]
		public void TestExceptionsForInvalidNumber()
		{
			UnitRequiresNoMoreThanNUnitsRequirementFactory factory = new UnitRequiresNoMoreThanNUnitsRequirementFactory();
			UnitType unitType = new UnitType("testUnit", "Test Unit", race);
			factory.CreateRequirement(unitType, typeID1+":a", raceFactory);
		}
	}
}