# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1334419100 -3600
# Node ID 49480e1a8f90b4b291b9f153c40c40a592b98985
# Parent  fd51527f6d64190112f55f6e6e60ae790a24cd42
Re #359: Add "only contained" attribute to unit types
 * Add tests to make sure contained units aren't returned in list for category

diff -r fd51527f6d64 -r 49480e1a8f90 API/Objects/RaceTests.cs
--- a/API/Objects/RaceTests.cs	Sat Mar 31 20:17:44 2012 +0100
+++ b/API/Objects/RaceTests.cs	Sat Apr 14 16:58:20 2012 +0100
@@ -62,6 +62,19 @@
 			Assert.That(raceRequirements, Has.Member(unitRequirement1));
 			Assert.That(raceRequirements, Has.Member(unitRequirement2));
 			Assert.That(raceRequirements, Has.Member(unitRequirement3));
+		}
+
+		[Test()]
+		public void TestContainedOnlyUnitsAreNotReturned()
+		{
+			MockRace race = new MockRace();
+			MockUnitType unitType1 = new MockUnitType("type1", "Type 1");
+			race.AddUnitType(unitType1);
+			MockContainedUnitType unitType2 = new MockContainedUnitType("type2", "Type 2");
+			race.AddUnitType(unitType2);
+			unitType1.AddContainedUnitType(unitType2);
+			Assert.That(race.GetUnitTypes(MockCategory.GetCategory()), Has.Length(1));
+			Assert.That(race.GetUnitTypes(MockCategory.GetCategory()), Has.Member(unitType1));
 		}
 
 		public override Race GetObject ()
diff -r fd51527f6d64 -r 49480e1a8f90 IBBoard.WarFoundry.API.Tests.csproj
--- a/IBBoard.WarFoundry.API.Tests.csproj	Sat Mar 31 20:17:44 2012 +0100
+++ b/IBBoard.WarFoundry.API.Tests.csproj	Sat Apr 14 16:58:20 2012 +0100
@@ -121,6 +121,7 @@
     <Compile Include="API\Objects\Requirement\RaceRequiresNoMoreThanNUnitsRequirementTest.cs" />
     <Compile Include="API\Objects\Requirement\AbstractUnitTypeUnitRequirementTest.cs" />
     <Compile Include="API\Objects\Requirement\AbstractRaceUnitRequirementTest.cs" />
+    <Compile Include="API\Objects\Mock\MockContainedUnitType.cs" />
   </ItemGroup>
   <ItemGroup>
     <None Include="app.config" />