def test_mortgage_property_action(self):
bsmt_phase = BSMT()
mortgage_property_phase = MortgageProperty()
turn_end_phase = TurnEnd()
phases = {
'MortgageProperty': mortgage_property_phase,
'BSMT': bsmt_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
agent_1 = Agent()
agent_2 = Agent()
player_1 = Player(1, position=board.property_at(0), agent=agent_1)
player_2 = Player(2, position=board.property_at(1), agent=agent_2)
mediterranean_avenue = board.property_at(1)
mediterranean_avenue.build(2)
mediterranean_avenue.own(player_2)
agent_1.bsmt_decision = MagicMock(return_value=(None, None))
agent_2_mock = Mock()
agent_2_mock.side_effect = iter([(Action.MORTGAGE_PROPERTY,
mediterranean_avenue), (None, None)])
agent_2.bsmt_decision = agent_2_mock
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = bsmt_phase
context = Context(phases, game_state, game_phase)
mortgage_property_phase.apply = MagicMock(return_value=(context, None))
context.apply()
mortgage_property_phase.apply.assert_called_once()
def test_chance_3_unowned(self):
special_property_phase = SpecialProperty()
buy_property_phase = BuyProperty()
phases = {
'SpecialProperty': special_property_phase,
'BuyProperty': buy_property_phase
}
board = Board()
board.next_chance = MagicMock(return_value=3)
electric_company = board.property_at(12)
board.nearest_utility = MagicMock(return_value=electric_company)
player_1 = Player(1, amount=100, position=board.property_at(0))
player_2 = Player(2, amount=300, position=board.property_at(36))
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = special_property_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is buy_property_phase)
self.assertTrue(player_2.position is electric_company)
self.assertEqual(player_2.amount, 300)
self.assertEqual(player_1.amount, 100)
def test_sell_house_action(self):
bsmt_phase = BSMT()
sell_house_phase = SellHouse()
turn_end_phase = TurnEnd()
phases = {
'SellHouse': sell_house_phase,
'BSMT': bsmt_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
agent_1 = Agent()
agent_1.bsmt_decision = MagicMock(return_value=(None, None))
agent_2 = Agent()
sell_action = [1, 1]
agent_2_mock = Mock()
agent_2_mock.side_effect = iter([(Action.SELL_HOUSE, sell_action),
(None, None)])
agent_2.bsmt_decision = agent_2_mock
player_1 = Player(1, position=board.property_at(0), agent=agent_1)
player_2 = Player(2, position=board.property_at(1), agent=agent_2)
mediterranean_avenue = board.property_at(1)
mediterranean_avenue.build(2)
mediterranean_avenue.own(player_2)
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = bsmt_phase
context = Context(phases, game_state, game_phase)
sell_house_phase.apply = MagicMock(return_value=(context, None))
context.apply()
sell_house_phase.apply.assert_called_once()
def test_chance_2(self):
bsmt_phase = BSMT()
special_property_phase = SpecialProperty()
phases = {
'BSMT': bsmt_phase,
'SpecialProperty': special_property_phase,
}
board = Board()
board.next_chance = MagicMock(return_value=2)
board.passes_go = MagicMock(return_value=True)
player_1 = Player(1, amount=100, position=board.property_at(1))
player_2 = Player(2, amount=300, position=board.property_at(36))
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = special_property_phase
context = Context(phases, game_state, game_phase)
st_charles_place = board.property_at(11)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is bsmt_phase)
self.assertTrue(player_2.position is st_charles_place)
self.assertEqual(player_2.amount, 500)
self.assertEqual(player_1.amount, 100)
def test_chance_4_5_owned(self):
dice = utils.Dice()
dice.roll = MagicMock(return_value=(1, 2))
special_property_phase = SpecialProperty(dice)
bsmt_phase = BSMT()
phases = {
'SpecialProperty': special_property_phase,
'BSMT': bsmt_phase
}
board = Board()
board.next_chance = MagicMock(return_value=4)
reading_railroad = board.property_at(5)
board.nearest_railroad = MagicMock(return_value=reading_railroad)
player_1 = Player(1, amount=100, position=board.property_at(0))
player_2 = Player(2, amount=300, position=board.property_at(36))
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = special_property_phase
context = Context(phases, game_state, game_phase)
reading_railroad.own(player_1)
reading_railroad.rent = MagicMock(return_value=15)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is bsmt_phase)
self.assertTrue(player_2.position is reading_railroad)
self.assertEqual(player_2.amount, 300 - 10 * 15)
self.assertEqual(player_1.amount, 100 + 10 * 15)
def test_apply_player_bankrupt(self):
bsmt_phase = BSMT()
pay_rent_phase = PayRent()
turn_end_phase = TurnEnd()
phases = {
'BSMT': bsmt_phase,
'PayRent': pay_rent_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(0))
player_2 = Player(2, amount=10, position=board.property_at(1))
mediterranean_avenue = board.property_at(1)
mediterranean_avenue.own(player_1)
rent = 100
mediterranean_avenue.rent = MagicMock(return_value=rent)
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = pay_rent_phase
context = Context(phases, game_state, game_phase)
bsmt_phase.apply = self.fake_bsmt_cycle(rent - 20)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is turn_end_phase)
self.assertEqual(new_context.state.current_player.amount, 90)
def test_trade_property_action(self):
bsmt_phase = BSMT()
trade_property_phase = TradeProperty()
turn_end_phase = TurnEnd()
phases = {
'TradeProperty': trade_property_phase,
'BSMT': bsmt_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
agent_1 = Agent()
agent_2 = Agent()
player_1 = Player(1, position=board.property_at(0), agent=agent_1)
player_2 = Player(2, position=board.property_at(1), agent=agent_2)
mediterranean_avenue = board.property_at(1)
mediterranean_avenue.own(player_1)
baltic_avenue = board.property_at(3)
baltic_avenue.own(player_1)
reading_railroad = board.property_at(5)
reading_railroad.own(player_2)
oriental_avenue = board.property_at(6)
oriental_avenue.own(player_2)
agent_1_mock = Mock()
agent_1_mock.side_effect = iter([
(Action.TRADE_PROPERTY,
(200, [mediterranean_avenue,
baltic_avenue], 400, [reading_railroad, oriental_avenue])),
(None, None)
])
agent_1.bsmt_decision = agent_1_mock
agent_2.bsmt_decision = MagicMock(return_value=(None, None))
agent_2.respond_trade = MagicMock(return_value=(True, None))
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = bsmt_phase
context = Context(phases, game_state, game_phase)
trade_property_phase.apply = MagicMock(return_value=(context, None))
expected_trades = [{
'buyer': player_2,
'seller': player_1,
'price': 200,
'properties': [mediterranean_avenue, baltic_avenue]
}, {
'buyer': player_1,
'seller': player_2,
'price': 400,
'properties': [reading_railroad, oriental_avenue]
}]
context.apply()
trade_property_phase.apply.assert_called_once_with(
context, expected_trades)
def test_change_player_on_double_roll(self):
dice_roll_phase = DiceRoll()
turn_end_phase = TurnEnd()
phases = {
'DiceRoll': dice_roll_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
double_roll = ((2, 2), (2, 2))
player_1 = Player(1, position=board.property_at(1))
player_2 = Player(2,
position=board.property_at(0),
previous_rolls=double_roll)
players = [player_1, player_2]
game_state = GameState(players, board)
game_state.next_player()
game_phase = turn_end_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is dice_roll_phase)
self.assertTrue(new_context.state.current_player is player_2)