Mercurial > repos > IBBoard.WarFoundry.API.Tests
diff API/Objects/Requirement/AbstractUnitRequirementTest.cs @ 231:5e03b68dd214
Re #410: Create "N units per M models in parent unit" requirement
* Add required usage of context objects
* Add tests for new requirement
* Add some extras to test unit type setup that the new requirement needs
author | IBBoard <dev@ibboard.co.uk> |
---|---|
date | Fri, 27 Jul 2012 20:33:40 +0100 |
parents | f951595143a9 |
children | 99248d05ba9c |
line wrap: on
line diff
--- a/API/Objects/Requirement/AbstractUnitRequirementTest.cs Wed Jul 11 20:54:12 2012 +0100 +++ b/API/Objects/Requirement/AbstractUnitRequirementTest.cs Fri Jul 27 20:33:40 2012 +0100 @@ -5,6 +5,7 @@ using NUnit.Framework; using NUnit.Framework.SyntaxHelpers; using IBBoard.NUnit; +using System.Collections.Generic; namespace IBBoard.WarFoundry.API.Objects.Requirement { @@ -58,17 +59,23 @@ Assert.That(req.GetValidationMessage(army), Is.EqualTo(message)); } - protected static void AddUnitsOfTypeToArmy(int count, UnitType unitType, Army army) + protected static ICollection<Unit> AddUnitsOfTypeToArmy(int count, UnitType unitType, Army army) { + ICollection<Unit> units = new List<Unit>(); + for (int i = 0; i < count; i++) { - AddUnitOfTypeToArmy(unitType, army); + units.Add(AddUnitOfTypeToArmy(unitType, army)); } + + return units; } - protected static void AddUnitOfTypeToArmy(UnitType unitType, Army army) + protected static Unit AddUnitOfTypeToArmy(UnitType unitType, Army army) { - army.AddUnit(CreateUnitOfType(unitType, army)); + Unit unit = CreateUnitOfType(unitType, army); + army.AddUnit(unit); + return unit; } protected static Unit CreateUnitOfType(UnitType unitType, Army army)