Mercurial > repos > IBBoard.WarFoundry.API.Tests
diff API/Objects/RaceTests.cs @ 224:49480e1a8f90
Re #359: Add "only contained" attribute to unit types
* Add tests to make sure contained units aren't returned in list for category
author | IBBoard <dev@ibboard.co.uk> |
---|---|
date | Sat, 14 Apr 2012 16:58:20 +0100 |
parents | 2ae80631ec9c |
children |
line wrap: on
line diff
--- 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 ()