112
|
1 // This file (UnitRequiresAtLeastNUnitsRequirement.cs) is a part of the IBBoard.WarFoundry.API.Tests project and is copyright 2011 IBBoard
|
|
2 //
|
|
3 // The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.
|
|
4 using System;
|
|
5 using NUnit.Framework;
|
|
6 using IBBoard.WarFoundry.API.Objects.Mock;
|
|
7 using NUnit.Framework.SyntaxHelpers;
|
|
8
|
|
9 namespace IBBoard.WarFoundry.API.Objects.Requirement
|
|
10 {
|
|
11 [TestFixture()]
|
|
12 public class UnitRequiresAtLeastNUnitsRequirementTest
|
|
13 {
|
|
14 private MockRace mockRace;
|
|
15 private UnitType unitType1;
|
|
16 private UnitType unitType2;
|
|
17 private UnitType unitType3;
|
|
18
|
|
19 [TestFixtureSetUp()]
|
|
20 public void SetupRace()
|
|
21 {
|
|
22 mockRace = new MockRace();
|
|
23 unitType1 = new MockUnitType("type1", "Unit Type 1");
|
|
24 mockRace.AddUnitType(unitType1);
|
|
25 unitType2 = new MockUnitType("type2", "Unit Type 2");
|
|
26 mockRace.AddUnitType(unitType2);
|
|
27 unitType3 = new MockUnitType("type3", "Unit Type 3");
|
|
28 mockRace.AddUnitType(unitType3);
|
|
29 }
|
|
30
|
|
31 [Test()]
|
|
32 public void TestAddingUnitWithNoUnitsAndOneUnitTypeRequired()
|
|
33 {
|
|
34 Army army = new Army(mockRace, "Test", 1000);
|
|
35 Unit unit = CreateUnitOfType(unitType1, army);
|
|
36 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
37 req.AddUnitTypeRequirement(unitType2);
|
|
38 Assert.That(req.AllowsAdding(unit, army), Is.False);
|
|
39 }
|
|
40
|
|
41 [Test()]
|
|
42 public void TestAddingUnitWithOneUnitAndOneUnitTypeRequired()
|
|
43 {
|
|
44 Army army = new Army(mockRace, "Test", 1000);
|
|
45 AddUnitOfTypeToArmy(unitType2, army);
|
|
46 Unit unit = CreateUnitOfType(unitType1, army);
|
|
47 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
48 req.AddUnitTypeRequirement(unitType2);
|
|
49 Assert.That(req.AllowsAdding(unit, army), Is.True);
|
|
50 }
|
|
51
|
|
52 [Test()]
|
|
53 public void TestAddingUnitTypeWithNoUnitsAndOneUnitTypeRequired()
|
|
54 {
|
|
55 Army army = new Army(mockRace, "Test", 1000);
|
|
56 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
57 req.AddUnitTypeRequirement(unitType2);
|
|
58 Assert.That(req.AllowsAdding(unitType1, army), Is.False);
|
|
59 }
|
|
60
|
|
61 [Test()]
|
|
62 public void TestAddingUnitTypeWithOneUnitAndOneUnitTypeRequired()
|
|
63 {
|
|
64 Army army = new Army(mockRace, "Test", 1000);
|
|
65 AddUnitOfTypeToArmy(unitType2, army);
|
|
66 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
67 req.AddUnitTypeRequirement(unitType2);
|
|
68 Assert.That(req.AllowsAdding(unitType1, army), Is.True);
|
|
69 }
|
|
70
|
|
71 [Test()]
|
|
72 public void TestAddingUnitTypeWithNoUnitsAndTwoUnitTypesRequired()
|
|
73 {
|
|
74 Army army = new Army(mockRace, "Test", 1000);
|
|
75 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
76 req.AddUnitTypeRequirement(unitType2);
|
|
77 req.AddUnitTypeRequirement(unitType3);
|
|
78 Assert.That(req.AllowsAdding(unitType1, army), Is.False);
|
|
79 }
|
|
80
|
|
81 [Test()]
|
|
82 public void TestAddingUnitTypeWithOneUnitAndTwoUnitTypesRequired()
|
|
83 {
|
|
84 Army army = new Army(mockRace, "Test", 1000);
|
|
85 AddUnitOfTypeToArmy(unitType2, army);
|
|
86 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
87 req.AddUnitTypeRequirement(unitType2);
|
|
88 req.AddUnitTypeRequirement(unitType3);
|
|
89 Assert.That(req.AllowsAdding(unitType1, army), Is.False);
|
|
90 }
|
|
91
|
|
92 [Test()]
|
|
93 public void TestAddingUnitTypeWithBothUnitsAndTwoUnitTypesRequired()
|
|
94 {
|
|
95 Army army = new Army(mockRace, "Test", 1000);
|
|
96 AddUnitOfTypeToArmy(unitType2, army);
|
|
97 AddUnitOfTypeToArmy(unitType3, army);
|
|
98 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
99 req.AddUnitTypeRequirement(unitType2);
|
|
100 req.AddUnitTypeRequirement(unitType3);
|
|
101 Assert.That(req.AllowsAdding(unitType1, army), Is.True);
|
|
102 }
|
|
103
|
|
104 [Test()]
|
|
105 public void TestAddingUnitTypeWithOneUnitAndTwoOfOneUnitTypeRequired()
|
|
106 {
|
|
107 Army army = new Army(mockRace, "Test", 1000);
|
|
108 AddUnitOfTypeToArmy(unitType2, army);
|
|
109 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
110 req.AddUnitTypeRequirement(unitType2, 2);
|
|
111 Assert.That(req.AllowsAdding(unitType1, army), Is.False);
|
|
112 }
|
|
113
|
|
114 [Test()]
|
|
115 public void TestAddingUnitTypeWithTwoUnitsAndTwoOfOneUnitTypeRequired()
|
|
116 {
|
|
117 Army army = new Army(mockRace, "Test", 1000);
|
|
118 AddUnitOfTypeToArmy(unitType2, army);
|
|
119 AddUnitOfTypeToArmy(unitType2, army);
|
|
120 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
121 req.AddUnitTypeRequirement(unitType2, 2);
|
|
122 Assert.That(req.AllowsAdding(unitType1, army), Is.True);
|
|
123 }
|
|
124
|
|
125 [Test()]
|
|
126 public void TestAddingUnitTypeRequirementWithNoNumberDefaultsToHavingOne()
|
|
127 {
|
|
128 Army army = new Army(mockRace, "Test", 1000);
|
|
129 AddUnitOfTypeToArmy(unitType2, army);
|
|
130 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
131 req.AddUnitTypeRequirement(unitType2);
|
|
132 Assert.That(req.AllowsAdding(unitType1, army), Is.True);
|
|
133 }
|
|
134
|
|
135 [Test()]
|
|
136 public void TestValidityOfArmyWithNoUnitsAndOneUnitTypeRequired()
|
|
137 {
|
|
138 Army army = new Army(mockRace, "Test", 1000);
|
|
139 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
140 req.AddUnitTypeRequirement(unitType2);
|
|
141 Assert.That(req.ValidatesArmy(army), Is.False);
|
|
142 }
|
|
143
|
|
144 [Test()]
|
|
145 public void TestValidityOfArmyWithOneUnitAndOneUnitTypeRequired()
|
|
146 {
|
|
147 Army army = new Army(mockRace, "Test", 1000);
|
|
148 AddUnitOfTypeToArmy(unitType2, army);
|
|
149 UnitRequiresAtLeastNUnitsRequirement req = new UnitRequiresAtLeastNUnitsRequirement(unitType1);
|
|
150 req.AddUnitTypeRequirement(unitType2);
|
|
151 Assert.That(req.ValidatesArmy(army), Is.True);
|
|
152 }
|
|
153 private static void AddUnitOfTypeToArmy(UnitType unitType, Army army)
|
|
154 {
|
|
155 army.AddUnit(CreateUnitOfType(unitType, army));
|
|
156 }
|
|
157 private static Unit CreateUnitOfType(UnitType unitType, Army army)
|
|
158 {
|
|
159 return new Unit(unitType, army.GetCategory(unitType.MainCategory));
|
|
160 }
|
|
161 }
|
|
162 }
|
|
163
|