@@ -100,10 +100,10 @@ class StateTest : public Test {
100100 path_planner.get (), 10 , 10 , 10 , 10 , 10 , 10 ));
101101
102102 // Player 2 has 1 villager
103- villagers[1 ].push_back (make_unique<Villager>(
104- 4 , PlayerId::PLAYER1, ActorType::VILLAGER, 100 , 100 ,
105- DoubleVec2D ( 10 , 10 ), gold_manager. get (), path_planner. get () , 10 , 10 ,
106- 10 , 10 , 10 , 10 ));
103+ villagers[1 ].push_back (
104+ make_unique<Villager>( 4 , PlayerId::PLAYER1, ActorType::VILLAGER,
105+ 100 , 100 , DoubleVec2D ( 10 , 10 ), gold_manager ,
106+ path_planner. get (), 10 , 10 , 10 , 10 , 10 , 10 ));
107107
108108 // Player2 has 1 soldier, say..
109109 soldiers[1 ].push_back (make_unique<Soldier>(
@@ -120,9 +120,6 @@ class StateTest : public Test {
120120};
121121
122122TEST_F (StateTest, MoveUnitTest) {
123- // Adding expectation for gold manager as state is updated
124- EXPECT_CALL (*gold_manager, Update ());
125-
126123 state->MoveUnit (PlayerId::PLAYER1, 1 , Vec2D (5 , 5 ));
127124 auto curr_villagers = state->GetVillagers ();
128125 auto curr_villager = curr_villagers[0 ].front ();
@@ -135,9 +132,6 @@ TEST_F(StateTest, MoveUnitTest) {
135132}
136133
137134TEST_F (StateTest, MineLocationTest) {
138- // Adding expectation for gold manager as state is updated
139- EXPECT_CALL (*gold_manager, Update ());
140-
141135 state->MineLocation (PlayerId::PLAYER1, 1 , Vec2D (25 , 25 ));
142136 auto curr_villagers = state->GetVillagers ();
143137 auto curr_villager = curr_villagers[0 ].front ();
@@ -151,9 +145,6 @@ TEST_F(StateTest, MineLocationTest) {
151145}
152146
153147TEST_F (StateTest, VillagerAttackEnemyTest) {
154- // Adding expectation for gold manager as state is updated
155- EXPECT_CALL (*gold_manager, Update ());
156-
157148 state->AttackActor (PlayerId::PLAYER1, 1 , 2 );
158149 auto curr_villagers = state->GetVillagers ();
159150 auto curr_villager = curr_villagers[0 ].front ();
@@ -170,9 +161,6 @@ TEST_F(StateTest, VillagerAttackEnemyTest) {
170161}
171162
172163TEST_F (StateTest, SoldierAttackEnemyTest) {
173- // Adding expectation for gold manager as state is updated
174- EXPECT_CALL (*gold_manager, Update ());
175-
176164 state->AttackActor (PlayerId::PLAYER2, 2 , 1 );
177165 auto curr_soldiers = state->GetSoldiers ();
178166 auto curr_soldier = curr_soldiers[1 ].front ();
@@ -195,17 +183,10 @@ TEST_F(StateTest, CreateFactoryTest) {
195183 auto curr_factories = state->GetFactories ();
196184 auto villager = curr_villagers[0 ].front ();
197185
198- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
199186 // Updating the state and soldier
200187 while (villager->GetState () != VillagerStateName::BUILD) {
201188 state->Update ();
202189 }
203- =======
204- // Adding expectation for gold manager as state is updated
205- EXPECT_CALL (*gold_manager, Update ());
206-
207- state->Update ();
208- >>>>>>> Add expectations for GoldManager mock
209190
210191 // Checking if the villager has transitioned into the BUILD state
211192 ASSERT_EQ (villager->GetState (), VillagerStateName::BUILD);
@@ -215,20 +196,8 @@ TEST_F(StateTest, CreateFactoryTest) {
215196 curr_factories = state->GetFactories ();
216197 auto factory = curr_factories[0 ].front ();
217198 ASSERT_EQ (curr_factories[0 ].size (), 1 );
218- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
219199 ASSERT_EQ (factory->GetState (), FactoryStateName::UNBUILT);
220-
221- <<<<<<< e22e3eaa527dea449d4e47edfe57c4f9204a9c48
222- auto gold_manager = villager->GetGoldManager ();
223- ASSERT_EQ (gold_manager->GetBalance (PlayerId::PLAYER1), 5000 - FACTORY_COST);
224- =======
225- // This is a gold manager test
226- // auto gold_manager = curr_villagers[0].front()->GetGoldManager();
227- // ASSERT_EQ(gold_manager->GetBalance(PlayerId::PLAYER1), 5000 - FACTORY_COST);
228- >>>>>>> Replace GoldManager with mock in all tests
229- =======
230200 ASSERT_EQ (curr_factories[0 ].front ()->GetState (), FactoryStateName::UNBUILT);
231- >>>>>>> Add expectations for GoldManager mock
232201}
233202
234203TEST_F (StateTest, BuildFactoryTest) {
@@ -239,43 +208,30 @@ TEST_F(StateTest, BuildFactoryTest) {
239208 auto curr_factories = state->GetFactories ();
240209 auto villager = curr_villagers[0 ].front ();
241210
242- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
243211 // Updating the villager's state
244212 while (villager->GetState () != VillagerStateName::BUILD) {
245213 state->Update ();
246214 }
247215
248- <<<<<<< 8d7402a77dd4b1910544206ee76d36a175b1ae5d
249216 ASSERT_EQ (villager->GetState (), VillagerStateName::BUILD);
250217
251218 // Getting the updated factories from state
252219 curr_factories = state->GetFactories ();
253220 auto factory = curr_factories[0 ].front ();
254- =======
255- EXPECT_CALL (*gold_manager, AssignGold ());
256- =======
257- EXPECT_CALL (*gold_manager, Update ());
258- >>>>>>> Add tests for GoldManager
259221
222+ // Updating state
260223 state->Update ();
261224
262225 // Adding expectation for gold manager as state is updated
263226
264227 ASSERT_EQ (curr_villagers[0 ].front ()->GetState (), VillagerStateName::BUILD);
265- >>>>>>> Add expectations for GoldManager mock
266228 ASSERT_EQ (curr_factories[0 ].size (), 1 );
267229 ASSERT_EQ (factory->GetState (), FactoryStateName::UNBUILT);
268230
269231 // Test BuildFactory for the same villager..
270232 state->BuildFactory (PlayerId::PLAYER1, 1 , factory->GetActorId ());
271233
272- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
273- // Updating state again
274- =======
275- // Adding expectation for gold manager as state is updated
276- EXPECT_CALL (*gold_manager, Update ());
277-
278- >>>>>>> Add expectations for GoldManager mock
234+ // Updating state
279235 state->Update ();
280236
281237 ASSERT_EQ (curr_villagers[0 ].front ()->GetState (), VillagerStateName::BUILD);
@@ -284,9 +240,7 @@ TEST_F(StateTest, BuildFactoryTest) {
284240 // Test BuildFactory for the other villager..
285241 state->BuildFactory (PlayerId::PLAYER1, 3 , factory->GetActorId ());
286242
287- // Adding expectation for gold manager as state is updated
288- EXPECT_CALL (*gold_manager, Update ());
289-
243+ // Updating state
290244 state->Update ();
291245
292246 ASSERT_EQ (curr_villagers[0 ][1 ]->GetState (), VillagerStateName::BUILD);
@@ -301,17 +255,13 @@ TEST_F(StateTest, FactoryProductionTest) {
301255 auto curr_factories = state->GetFactories ();
302256 auto villager = curr_villagers[0 ].front ();
303257
304- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
305258 // Updating the villager's state
306259 while (villager->GetState () != VillagerStateName::BUILD) {
307260 state->Update ();
308261 }
309- =======
310- // Adding expectation for gold manager as state is updated
311- EXPECT_CALL (*gold_manager, Update ());
312262
263+ // Updating state
313264 state->Update ();
314- >>>>>>> Add expectations for GoldManager mock
315265
316266 ASSERT_EQ (villager->GetState (), VillagerStateName::BUILD);
317267
@@ -327,9 +277,7 @@ TEST_F(StateTest, FactoryProductionTest) {
327277 state->SetFactoryProduction (PlayerId::PLAYER1, factory->GetActorId (),
328278 ActorType::SOLDIER);
329279
330- // Adding expectation for gold manager as state is updated
331- EXPECT_CALL (*gold_manager, Update ());
332-
280+ // Updating state
333281 state->Update ();
334282
335283 ASSERT_EQ (factory->GetProductionState (), ActorType::SOLDIER);
@@ -344,17 +292,13 @@ TEST_F(StateTest, FactoryDeathTest) {
344292 ASSERT_EQ (curr_factories[0 ].size (), 0 );
345293 auto villager = curr_villagers[0 ].front ();
346294
347- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
348295 // Updating the villager's state
349296 while (villager->GetState () != VillagerStateName::BUILD) {
350297 state->Update ();
351298 }
352- =======
353- // Adding expectation for gold manager as state is updated
354- EXPECT_CALL (*gold_manager, Update ());
355299
300+ // Updating state
356301 state->Update ();
357- >>>>>>> Add expectations for GoldManager mock
358302
359303 ASSERT_EQ (villager->GetState (), VillagerStateName::BUILD);
360304
@@ -364,7 +308,6 @@ TEST_F(StateTest, FactoryDeathTest) {
364308 ASSERT_EQ (curr_factories[0 ].size (), 1 );
365309 ASSERT_EQ (factory->GetState (), FactoryStateName::UNBUILT);
366310
367- <<<<<<< 72cb5844d7321e03c15e8a78f1b3e553a1a92ca5
368311 // Killing the factory
369312 factory->SetHp (0 );
370313 factory->Update ();
@@ -374,24 +317,16 @@ TEST_F(StateTest, FactoryDeathTest) {
374317 ASSERT_EQ (factory->GetState (), FactoryStateName::DEAD);
375318 ASSERT_EQ (factory->GetHp (), 0 );
376319
377- // Updating the state
378- =======
379320 // Kill factory
380- EXPECT_CALL (*gold_manager, Update ());
381-
382321 auto curr_factory = curr_factories[0 ].front ();
383322 curr_factory->SetHp (0 );
384- >>>>>>> Add expectations for GoldManager mock
385323 state->Update ();
386324
387325 auto new_facs = state->GetFactories ();
388326 ASSERT_EQ (new_facs[0 ].size (), 0 );
389327}
390328
391329TEST_F (StateTest, VillagerDeathTest) {
392- // Adding expectation for gold manager as state is updated
393- EXPECT_CALL (*gold_manager, Update ());
394-
395330 auto curr_villagers = state->GetVillagers ();
396331 ASSERT_EQ (curr_villagers[0 ].size (), 2 );
397332
@@ -404,9 +339,6 @@ TEST_F(StateTest, VillagerDeathTest) {
404339}
405340
406341TEST_F (StateTest, SoldierDeathTest) {
407- // Adding expectation for gold manager as state is updated
408- EXPECT_CALL (*gold_manager, Update ());
409-
410342 auto curr_soldiers = state->GetSoldiers ();
411343 ASSERT_EQ (curr_soldiers[1 ].size (), 1 );
412344
0 commit comments