def test_player_buys_property(self):
buy_property_phase = BuyProperty()
bsmt_phase = BSMT()
phases = {
'BuyProperty': buy_property_phase,
'BSMT': bsmt_phase,
}
board = Board()
agent = Agent()
agent.buy_property = MagicMock(return_value=True)
current_position = board.property_at(1)
current_position.cost = 100
player_1 = Player(1, amount=20, position=current_position, agent=agent)
player_2 = Player(2)
players = [player_1, player_2]
game_state = GameState(players, board)
game_phase = buy_property_phase
context = Context(phases, game_state, game_phase)
bsmt_phase.apply = self.fake_bsmt_cycle(100)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is bsmt_phase)
self.assertTrue(
new_context.state.current_player.position.owned_by is player_1)
self.assertEqual(new_context.state.current_player.amount, 20)
self.assertEqual(new_context.state.current_player._debt['bank'], 0)
def test_player_declines_buying_property(self):
buy_property_phase = BuyProperty()
bsmt_phase = BSMT()
auction_phase = Auction()
phases = {
'BuyProperty': buy_property_phase,
'BSMT': bsmt_phase,
'Auction': auction_phase,
}
board = Board()
agent = Agent()
agent.buy_property = MagicMock(return_value=False)
player_1 = Player(1,
amount=900,
position=board.property_at(1),
agent=agent)
player_2 = Player(2, position=board.property_at(0))
players = [player_1, player_2]
game_state = GameState(players, board)
game_phase = buy_property_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is auction_phase)
self.assertEqual(new_context.state.current_player.amount, 900)
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_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_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 setup(agent_1, agent_2):
dice = utils.Dice()
roll_mock = Mock()
roll_mock.side_effect = cycle([(1, 2), (2, 2)])
dice.roll = roll_mock
dice_roll_phase = DiceRoll(dice)
square_effect_phase = SquareEffect()
bsmt_phase = BSMT()
buy_property_phase = BuyProperty()
pay_rent_phase = PayRent()
auction_phase = Auction()
phases = {
'DiceRoll': dice_roll_phase,
'SquareEffect': square_effect_phase,
'BSMT': bsmt_phase,
'BuyProperty': buy_property_phase,
'PayRent': pay_rent_phase,
'Auction': auction_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(0), agent=agent_1)
player_2 = Player(2, position=board.property_at(0), agent=agent_2)
players = [player_1, player_2]
game_state = GameState(players, board)
start_phase = dice_roll_phase
context = Context(phases, game_state, start_phase)
return context
def test_special_property_community_chest(self):
square_effect_phase = SquareEffect()
special_property_phase = SpecialProperty()
phases = {
'SpecialProperty': special_property_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(2))
player_2 = Player(2, position=board.property_at(1))
players = [player_1, player_2]
game_state = GameState(players)
game_phase = square_effect_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is special_property_phase)
def test_apply(self):
dice = utils.Dice()
dice.roll = MagicMock(return_value=(1, 2))
dice_roll_phase = DiceRoll(dice)
square_effect_phase = SquareEffect()
phases = {
'DiceRoll': dice_roll_phase,
'SquareEffect': square_effect_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(0))
player_2 = Player(2, position=board.property_at(0))
players = [player_1, player_2]
game_state = GameState(players, board)
game_phase = dice_roll_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is square_effect_phase)
self.assertTrue(
new_context.state.current_player.position is board.property_at(3))
def test_unowned_property(self):
dice_roll_phase = DiceRoll()
square_effect_phase = SquareEffect()
buy_property_phase = BuyProperty()
pay_rent_phase = PayRent()
phases = {
'DiceRoll': dice_roll_phase,
'SquareEffect': square_effect_phase,
'BuyProperty': buy_property_phase,
'PayRent': pay_rent_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(1))
player_2 = Player(2, position=board.property_at(1))
players = [player_1, player_2]
game_state = GameState(players)
game_phase = square_effect_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(isinstance(new_context.phase, BuyProperty))
def test_change_player_on_regular_roll(self):
turn_end_phase = TurnEnd()
dice_roll_phase = DiceRoll()
phases = {
'DiceRoll': dice_roll_phase,
'TurnEnd': turn_end_phase,
}
board = Board()
regular_roll = ((1, 2), (2, 3))
player_1 = Player(1,
position=board.property_at(1),
previous_rolls=regular_roll)
player_2 = Player(2, position=board.property_at(0))
players = [player_1, player_2]
game_state = GameState(players, board)
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)
def test_self_owned_property(self):
bsmt_phase = BSMT()
square_effect_phase = SquareEffect()
buy_property_phase = BuyProperty()
pay_rent_phase = PayRent()
phases = {
'BSMT': bsmt_phase,
'SquareEffect': square_effect_phase,
'BuyProperty': buy_property_phase,
'PayRent': pay_rent_phase,
}
board = Board()
player_1 = Player(1, position=board.property_at(1))
player_2 = Player(2, position=board.property_at(1))
mediterranean_avenue = board.property_at(1)
mediterranean_avenue.own(player_1)
players = [player_1, player_2]
game_state = GameState(players)
game_phase = square_effect_phase
context = Context(phases, game_state, game_phase)
new_context, next_action = context.apply()
self.assertTrue(new_context.phase is bsmt_phase)