view API/Util/UnitEquipmentUtilTest.cs @ 24:92db6f6a6b0b

Re #198: Add equipment slots * Consolidate unit tests and add first test for percentage limit retrieval * Add tests to highlight problem with max limit actually taking min limit
author IBBoard <dev@ibboard.co.uk>
date Tue, 27 Oct 2009 20:52:56 +0000
parents 54abb61b9db8
children d48e73488838
line wrap: on
line source

// This file (UnitEquipmentItemUtilTest.cs) is a part of the IBBoard.WarFoundry.API.Tests project and is copyright 2009 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.Limits;
using IBBoard.WarFoundry.API.Objects;
using IBBoard.WarFoundry.API.Objects.Mock;

namespace IBBoard.WarFoundry.API.Util
{
	[TestFixture()]
	public class UnitEquipmentUtilTest
	{
		[Test()]
		public void TestGetEquipmentCountsForFixedLimitItemOnDefaultSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem(3);
			UnitType unitType = equip.EquipmentForUnit;
			Unit unit = GetUnit (unitType);
			Assert.AreEqual(3, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(3, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForFixedLimitItemOnDefaultSlotWithDifferentMinLimit()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem(1,3);
			UnitType unitType = equip.EquipmentForUnit;
			Unit unit = GetUnit (unitType);
			Assert.AreEqual(3, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(1, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForFixedLimitItemOnNumberLimitedSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.MaxLimit = new NumericSizeConstrainedLimit(3);
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(2));
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(2, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(2, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForFixedLimitItemOnPercentageLimitedSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.MaxLimit = new NumericSizeConstrainedLimit(7);
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new SimpleRoundedPercentageLimit(50, false));
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(5, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(5, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForFixedLimitItemOnNumberLimitedSlotWithExistingEquipment()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.MaxLimit = new NumericSizeConstrainedLimit(3);
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(6));
			Unit unit =GetUnit(unitType);
			UnitEquipmentItem equip2 = new MockPercentageAmountUnitEquipmentItem(new MockEquipmentItem("equip2"), unitType);
			equip2.SlotName = "slot";
			unit.SetEquipmentAmount(equip2, 5);
			Assert.AreEqual(1, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(1, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForPercentageLimitItemOnDefaultSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			UnitType unitType = equip.EquipmentForUnit;
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(5, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(5, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForPercentageLimitItemOnNumberLimitedSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(2));
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(2, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(2, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForPercentageLimitItemOnPercentageLimitedSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new SimpleRoundedPercentageLimit(25, false));
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(2, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(2, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestGetEquipmentCountsForPercentageLimitItemOnNumberLimitedSlotWithExistingEquipment()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(6));
			Unit unit = GetUnit(unitType);
			UnitEquipmentItem equip2 = new MockPercentageAmountUnitEquipmentItem(new MockEquipmentItem("equip2"), unitType);
			equip2.SlotName = "slot";
			unit.SetEquipmentAmount(equip2, 5);
			Assert.AreEqual(1, UnitEquipmentUtil.GetMaxEquipmentCount(unit, equip));
			Assert.AreEqual(1, UnitEquipmentUtil.GetMinEquipmentCount(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithRatioLimitedEquipmentAndUnlimitedSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();			
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new UnlimitedLimit());
			Unit unit = GetUnit(unitType);
			Assert.IsTrue(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithRatioLimitedEquipmentAndRatioSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new SimpleRoundedPercentageLimit(50, true));
			Unit unit = GetUnit(unitType);
			Assert.IsTrue(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithRatioLimitedEquipmentAndAbsoluteSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(5));
			Unit unit = GetUnit(unitType);
			Assert.IsFalse(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithAbsoluteLimitedEquipmentAndAbsoluteSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new AbsoluteNumericLimit(5));
			Unit unit = GetUnit(unitType);
			Assert.IsFalse(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithAbsoluteLimitedEquipmentAndUnlimitedSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new UnlimitedLimit());
			Unit unit = GetUnit(unitType);
			Assert.IsFalse(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestIsEquipmentRatioLimitsWithAbsoluteLimitedEquipmentAndRatioSlot()
		{
			UnitEquipmentItem equip = new MockNumericAmountUnitEquipmentItem();
			equip.SlotName = "slot";
			UnitType unitType = equip.EquipmentForUnit;
			unitType.AddEquipmentSlot("slot", new SimpleRoundedPercentageLimit(50, true));
			Unit unit = GetUnit(unitType);
			Assert.IsFalse(UnitEquipmentUtil.IsEquipmentRatioLimited(unit, equip));
		}
		
		[Test()]
		public void TestGetMaxPercentageForPercentageLimitedEquipmentOnUnlimitedSlot()
		{
			UnitEquipmentItem equip = new MockPercentageAmountUnitEquipmentItem(75);
			UnitType unitType = equip.EquipmentForUnit;
			Unit unit = GetUnit(unitType);
			Assert.AreEqual(75, UnitEquipmentUtil.GetMaxEquipmentPercentage(unit, equip));
		}
		
		private Unit GetUnit (UnitType unitType)
		{
			return new Unit ("unit", "Test Unit", 10, unitType, new MockArmyCategory (unitType.MainCategory));
		}
	}
}