# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1302364795 0
# Node ID 7ca536ccd27a2790af709cf92dad97690b4bc9db
# Parent  779ad76c26445d9395cc9b580563f0faa1585866
Re #27:
 * Start testing adding requirements to unit types

diff -r 779ad76c2644 -r 7ca536ccd27a API/Objects/UnitTypeTest.cs
--- a/API/Objects/UnitTypeTest.cs	Sat Apr 09 15:49:09 2011 +0000
+++ b/API/Objects/UnitTypeTest.cs	Sat Apr 09 15:59:55 2011 +0000
@@ -5,6 +5,7 @@
 using NUnit.Framework;
 using NUnit.Framework.Constraints;
 using NUnit.Framework.SyntaxHelpers;
+using IBBoard.WarFoundry.API.Objects.Requirement;
 
 namespace IBBoard.WarFoundry.API.Objects
 {
@@ -44,6 +45,16 @@
 			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(0));
+			unitType.AddRequirement(req);
+			Assert.That(unitType.GetRequirements(), Has.Length(1));
+		}
 	}
 }