from unittest import TestCase
-from naja.constants import BITS
+from naja.constants import BITS, MOVES
from naja.gameboard import GameBoard, LocationCard
+from naja.options import parse_args
from naja.player import Player
from naja import actions
class TestActions(TestCase):
+ def setUp(self):
+ parse_args([])
+
def make_player(self, *bits):
player_bits = 0
for bit in bits:
def make_board(self, player_bits=None, locations=None):
if locations is None:
locations = [{'actions': []}]
- board = GameBoard.new_game(locations)
+ board = GameBoard.new_game({'cards': locations})
if player_bits is not None:
board.player.bits.bits = 0
board.player.bits.set_bits(player_bits)
TEXT = "foo %(direction)s %(rowcol)s"
action_north = DirectionAction([], direction='NORTH')
- self.assertEqual(action_north.get_text(), "foo NORTH column")
+ self.assertEqual(action_north.get_text(), "foo {NORTH} column")
action_south = DirectionAction([], direction='SOUTH')
- self.assertEqual(action_south.get_text(), "foo SOUTH column")
+ self.assertEqual(action_south.get_text(), "foo {SOUTH} column")
action_east = DirectionAction([], direction='EAST')
- self.assertEqual(action_east.get_text(), "foo EAST row")
+ self.assertEqual(action_east.get_text(), "foo {EAST} row")
action_west = DirectionAction([], direction='WEST')
- self.assertEqual(action_west.get_text(), "foo WEST row")
-
- def test_bits_translation(self):
- action = actions.LocationAction(set([BITS.NORTH, 'MSB']))
- self.assertEqual(action.required_bits, set([BITS.NORTH, BITS.MSB]))
+ self.assertEqual(action_west.get_text(), "foo {WEST} row")
def test_DoNothing(self):
board = self.make_board()
self.assert_state(
state_before, state_after, exclude=['health'],
player_exclude=['bits'])
+
+ def test_AllowKnightMove(self):
+ board = self.make_board(player_bits=[BITS.RED, BITS.BLUE])
+ actions.AllowChessMove(
+ set([BITS.RED, BITS.BLUE]), chesspiece="KNIGHT"
+ ).perform_action(board, None)
+ self.assertEqual(board.player.movement_mode, MOVES.KNIGHT)
+
+ def test_AllowBishopMove(self):
+ board = self.make_board(player_bits=[BITS.RED, BITS.BLUE])
+ actions.AllowChessMove(
+ set([BITS.RED, BITS.BLUE]), chesspiece="BISHOP"
+ ).perform_action(board, None)
+ self.assertEqual(board.player.movement_mode, MOVES.BISHOP)
+
+ def test_AllowCastleMove(self):
+ board = self.make_board(player_bits=[BITS.RED, BITS.BLUE])
+ actions.AllowChessMove(
+ set([BITS.RED, BITS.BLUE]), chesspiece="CASTLE"
+ ).perform_action(board, None)
+ self.assertEqual(board.player.movement_mode, MOVES.CASTLE)