//  This file (UnitTypeTest.cs) is a part of the IBBoard.WarFoundry.API.Tests project and is copyright 2010 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 IBBoard.WarFoundry.API.Objects.Mock;
using NUnit.Framework;
using NUnit.Framework.Constraints;
using IBBoard.WarFoundry.API.Objects.Requirement;
using System.Reflection;
using IBBoard.NUnit;
using IBBoard.WarFoundry.API.Factories;

namespace IBBoard.WarFoundry.API.Objects
{
	[TestFixture()]
	public class UnitTypeTest : AbstractEqualityTest<UnitType>
	{
		[Test()]
		public void TestSettingMainCategorySetsCategoriesCollection()
		{
			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
			MockCategory mockCategory = MockCategory.GetCategory();
			unitType.MainCategory = mockCategory;
			Assert.That(unitType.Categories, Has.Member(mockCategory));
		}
		
		[Test()]
		public void TestAddingFirstCategorySetsMainCategory()
		{
			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
			MockCategory mockCategory = MockCategory.GetCategory();
			unitType.AddCategory(mockCategory);
			Assert.That(unitType.Categories.Length, Is.EqualTo(1));
			Assert.That(unitType.Categories, Has.Member(mockCategory));
			Assert.That(unitType.MainCategory, Is.EqualTo(mockCategory));			
		}

		[Test()]
		public void TestAddingMultipleCategoriesSetsMainCategory()
		{
			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
			MockCategory mockCategory = MockCategory.GetCategory();
			unitType.AddCategory(mockCategory);
			Category mockCategory2 = new Category("id1", "CatName");
			unitType.AddCategory(mockCategory2);
			Assert.That(unitType.Categories.Length, Is.EqualTo(2));
			Assert.That(unitType.Categories, Has.Member(mockCategory));
			Assert.That(unitType.Categories, Has.Member(mockCategory2));
			Assert.That(unitType.MainCategory, Is.EqualTo(mockCategory));
		}

		[Test]
		public void TestAddingRequirementsToUnitType()
		{
			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
			UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType);
			Assert.That(unitType.GetRequirements(), Has.Length.EqualTo(0));
			unitType.AddRequirement(req);
			Assert.That(unitType.GetRequirements(), Has.Length.EqualTo(1));
		}

		[Test]
		public void TestGetRequirementsGetsUnitCountRequirements()
		{
			MockRace mockRace = new MockRace();
			UnitType unitType = new UnitType("unittype1", "Swordsmen", mockRace);
			unitType.MinNumber = 1;
			unitType.MaxNumber = 2;
			Assert.That(unitType.GetRequirements(), Has.Length.EqualTo(2));
			RaceRequiresNoMoreThanNUnitsRequirement maxLimit = new RaceRequiresNoMoreThanNUnitsRequirement(mockRace);
			maxLimit.AddUnitTypeRequirement(unitType, 2);
			Assert.That(unitType.GetRequirements(), Has.Member(maxLimit));
			RaceRequiresAtLeastNUnitsRequirement minLimit = new RaceRequiresAtLeastNUnitsRequirement(mockRace);
			minLimit.AddUnitTypeRequirement(unitType, 1);
			Assert.That(unitType.GetRequirements(), Has.Member(minLimit));
		}

		public override UnitType GetObject()
		{
			return new UnitType("id1", "Type 1", new MockRace());
		}

		public override UnitType GetSameObject()
		{
			return new UnitType("id1", "Type 1", new MockRace());
		}

		public override UnitType GetDifferentObject()
		{
			DummyWarFoundryFactory factory = new DummyWarFoundryFactory();
			GameSystem gameSystem = new GameSystem("system", "system", factory);
			Race race = new Race("race", "race", gameSystem, factory);
			return new UnitType("id2", "Type 2", race);
		}
	}
}