def test__setUpVictoryCards(self):
victoryCards = 10
self.mocker.StubOutWithMock(dominion_rules, 'getGameSetupVictoryCardCount')
self.mocker.StubOutWithMock(Game, '_makePile')
self.mocker.StubOutWithMock(Game, '_combinePiles')
dominion_rules.getGameSetupVictoryCardCount(0).AndReturn(victoryCards)
Game._makePile(dominion_data.cards['estate'], victoryCards).AndReturn("estate pile")
Game._makePile(dominion_data.cards['duchy'], victoryCards).AndReturn("duchy pile")
Game._makePile(dominion_data.cards['province'], victoryCards).AndReturn("province pile")
Game._makePile(dominion_data.cards['estate'], 0).AndReturn("additional estates")
Game._combinePiles(["estate pile","additional estates"]).AndReturn("updated estate pile")
self.mocker.ReplayAll()
game = Game()
game._setUpVictoryCards()
self.assertEqual(game.SupplyArea.EstatePile, "updated estate pile")
self.assertEqual(game.SupplyArea.DuchyPile, "duchy pile")
self.assertEqual(game.SupplyArea.ProvincePile, "province pile")
self.mocker.VerifyAll()
def _setUpVictoryCards(self):
# 2) Add Estates, Duchies, and Provinces to the SupplyArea
numberOfVictoryCards = dominion_rules.getGameSetupVictoryCardCount(len(self.Players))
estateData = dominion_data.cards['estate']
duchyData = dominion_data.cards['duchy']
provinceData = dominion_data.cards['province']
self.SupplyArea.EstatePile = self._makePile(estateData, numberOfVictoryCards)
self.SupplyArea.DuchyPile = self._makePile(duchyData, numberOfVictoryCards)
self.SupplyArea.ProvincePile = self._makePile(provinceData, numberOfVictoryCards)
additionalEstateCount = len(self.Players) * dominion_rules.FIRST_DECK.ESTATE_CARDS
additionalEstates = self._makePile(estateData, additionalEstateCount)
self.SupplyArea.EstatePile = self._combinePiles([self.SupplyArea.EstatePile, additionalEstates])
