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])