from unittest import TestCase
-from naja.constants import BITS
+from naja.constants import BITS, MOVES
from naja.gameboard import GameBoard, LocationCard
from naja.player import Player
from naja import actions
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)