# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1301173375 0
# Node ID 1c32b8a1600ea252d7bdeb36913bb3c3416c5d5d
# Parent  3319955829908117ab235bf8250148a0936193c7
Re #27: Define unit requirements
 * Add checking for passing requirements

diff -r 331995582990 -r 1c32b8a1600e api/Objects/UnitRequiresAtLeastNUnitsRequirement.cs
--- a/api/Objects/UnitRequiresAtLeastNUnitsRequirement.cs	Sat Mar 26 17:03:00 2011 +0000
+++ b/api/Objects/UnitRequiresAtLeastNUnitsRequirement.cs	Sat Mar 26 21:02:55 2011 +0000
@@ -21,7 +21,18 @@
 
 		public bool CanAddToArmy(Unit unit, Army army)
 		{
-			return false;
+			bool canAdd = true;
+
+			foreach (UnitType type in requiredTypes)
+			{
+				if (army.GetUnitTypeCount(type) < 1)
+				{
+					canAdd = false;
+					break;
+				}
+			}
+
+			return canAdd;
 		}
 	}
 }