# HG changeset patch
# User IBBoard <dev@ibboard.co.uk>
# Date 1274211207 0
# Node ID 24a40c49b6aef8b6a16ac793707802420a022e1f
# Parent  ea8d30377a7216dde8fee25203b1d1b810ae5082
Re #270: Add multiple categories to API
 * Add more tests of behaviour

diff -r ea8d30377a72 -r 24a40c49b6ae API/Objects/UnitTypeTest.cs
--- a/API/Objects/UnitTypeTest.cs	Tue May 18 18:49:12 2010 +0000
+++ b/API/Objects/UnitTypeTest.cs	Tue May 18 19:33:27 2010 +0000
@@ -26,7 +26,22 @@
 			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
 			MockCategory mockCategory = new MockCategory();
 			unitType.AddCategory(mockCategory);
+			Assert.That(unitType.Categories.Length, Is.EqualTo(1));
 			Assert.That(unitType.Categories, Has.Member(mockCategory));
+			Assert.That(unitType.MainCategory, Is.EqualTo(mockCategory));			
+		}
+
+		[Test()]
+		public void TestAddingMultipleCategoriesSetsMainCategory()
+		{
+			UnitType unitType = new UnitType("unittype1", "Swordsmen", new MockRace());
+			MockCategory mockCategory = new MockCategory();
+			unitType.AddCategory(mockCategory);
+			Category mockCategory2 = new Category("id1", "CatName");
+			unitType.AddCategory(mockCategory2);
+			Assert.That(unitType.Categories.Length, Is.EqualTo(2));
+			Assert.That(unitType.Categories, Has.Member(mockCategory));
+			Assert.That(unitType.Categories, Has.Member(mockCategory2));
 			Assert.That(unitType.MainCategory, Is.EqualTo(mockCategory));
 		}
 	}