X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_actions.py;h=b233df76adc709c73ac0ea31570f9b20cc40cc84;hb=98f1e5d415b33e746850ec64c0cc804d38bd1bc3;hp=02c07964342df9f4d06ee6c7cca4427d49e62edd;hpb=2a950900b3b7ae9514e5351c3a07903611b9153e;p=naja.git diff --git a/naja/tests/test_actions.py b/naja/tests/test_actions.py index 02c0796..b233df7 100644 --- a/naja/tests/test_actions.py +++ b/naja/tests/test_actions.py @@ -1,12 +1,16 @@ 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: @@ -15,13 +19,16 @@ class TestActions(TestCase): def make_board(self, player_bits=None, locations=None): if locations is None: - locations = [{'actions': []}] - board = GameBoard.new_game(locations) + locations = [{'card_name': 'card', 'actions': []}] + board = GameBoard.new_game({'cards': locations}) if player_bits is not None: board.player.bits.bits = 0 board.player.bits.set_bits(player_bits) return board + def make_location_card(self, bits=(), name='card', actions=()): + return LocationCard(name, set(bits), list(actions), None) + def assert_player_bits(self, board, *bits): self.assertEqual(sum(1 << bit for bit in bits), board.player.bits.bits) @@ -50,9 +57,32 @@ class TestActions(TestCase): check_available(set([BITS.MSB]), [], False) check_available(set([BITS.MSB]), [BITS.MSB], True) - def test_bits_translation(self): - action = actions.LocationAction(set([BITS.NORTH, 'MSB'])) - self.assertEqual(action.required_bits, set([BITS.NORTH, BITS.MSB])) + def test_get_text(self): + class LocationAction1(actions.LocationAction): + TEXT = "foo" + action1 = LocationAction1([]) + self.assertEqual(action1.get_text(), "foo") + + class LocationAction2(actions.LocationAction): + TEXT = "foo %(bar)s" + action2 = LocationAction2([], bar="baz") + self.assertEqual(action2.get_text(), "foo baz") + + def test_get_text_row_column(self): + class DirectionAction(actions.LocationAction): + TEXT = "foo %(direction)s %(rowcol)s" + + action_north = DirectionAction([], direction='NORTH') + self.assertEqual(action_north.get_text(), "foo {NORTH} column") + + action_south = DirectionAction([], direction='SOUTH') + self.assertEqual(action_south.get_text(), "foo {SOUTH} column") + + action_east = DirectionAction([], direction='EAST') + 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_DoNothing(self): board = self.make_board() @@ -80,7 +110,7 @@ class TestActions(TestCase): def test_SetBits(self): board = self.make_board() state_before = board.export() - card = LocationCard(set([BITS.MSB, BITS.NORTH]), []) + card = self.make_location_card([BITS.MSB, BITS.NORTH]) actions.SetBits(set()).perform_action(board, card) state_after = board.export() self.assertEqual( @@ -90,7 +120,7 @@ class TestActions(TestCase): def test_ToggleBits(self): board = self.make_board(player_bits=[BITS.NORTH]) state_before = board.export() - card = LocationCard(set([BITS.MSB, BITS.NORTH]), []) + card = self.make_location_card([BITS.MSB, BITS.NORTH]) actions.ToggleBits(set()).perform_action(board, card) state_after = board.export() self.assertEqual(board.player.bits.check_bit(BITS.MSB), True) @@ -100,11 +130,11 @@ class TestActions(TestCase): def test_LoseHealthOrMSBAndSetBits_MSB_clear(self): board = self.make_board(player_bits=[]) state_before = board.export() - card = LocationCard(set([BITS.CYAN, BITS.NORTH]), []) + card = self.make_location_card([BITS.BLUE, BITS.NORTH]) actions.LoseHealthOrMSBAndSetBits(set()).perform_action(board, card) state_after = board.export() self.assertEqual(state_after['health'], state_before['health'] - 1) - self.assert_player_bits(board, BITS.CYAN, BITS.NORTH) + self.assert_player_bits(board, BITS.BLUE, BITS.NORTH) self.assert_state( state_before, state_after, exclude=['health'], player_exclude=['bits']) @@ -112,16 +142,16 @@ class TestActions(TestCase): def test_LoseHealthOrMSBAndSetBits_MSB_set(self): board = self.make_board(player_bits=[BITS.MSB]) state_before = board.export() - card = LocationCard(set([BITS.CYAN, BITS.NORTH]), []) + card = self.make_location_card([BITS.BLUE, BITS.NORTH]) actions.LoseHealthOrMSBAndSetBits(set()).perform_action(board, card) state_after = board.export() - self.assert_player_bits(board, BITS.CYAN, BITS.NORTH) + self.assert_player_bits(board, BITS.BLUE, BITS.NORTH) self.assert_state(state_before, state_after, player_exclude=['bits']) def test_LoseHealthOrMSBAndSetBits_MSB_set_and_on_card(self): board = self.make_board(player_bits=[BITS.MSB]) state_before = board.export() - card = LocationCard(set([BITS.MSB, BITS.NORTH]), []) + card = self.make_location_card([BITS.MSB, BITS.NORTH]) actions.LoseHealthOrMSBAndSetBits(set()).perform_action(board, card) state_after = board.export() self.assert_player_bits(board, BITS.MSB, BITS.NORTH) @@ -129,7 +159,7 @@ class TestActions(TestCase): def test_AcquireWinToken(self): board = self.make_board( - player_bits=[BITS.CYAN, BITS.MAGENTA, BITS.YELLOW]) + player_bits=[BITS.RED, BITS.GREEN, BITS.BLUE]) state_before = board.export() actions.AcquireWinToken(set()).perform_action(board, None) state_after = board.export() @@ -143,7 +173,7 @@ class TestActions(TestCase): board = self.make_board(player_bits=[BITS.NORTH]) board.lose_health() state_before = board.export() - card = LocationCard(set([BITS.CYAN, BITS.NORTH]), []) + card = self.make_location_card([BITS.BLUE, BITS.NORTH]) actions.GainHealthAndClearBitsOrMSB(set()).perform_action(board, card) state_after = board.export() self.assertEqual(state_after['health'], state_before['health'] + 1) @@ -156,7 +186,7 @@ class TestActions(TestCase): board = self.make_board(player_bits=[BITS.MSB, BITS.NORTH]) board.lose_health() state_before = board.export() - card = LocationCard(set([BITS.CYAN, BITS.NORTH]), []) + card = self.make_location_card([BITS.BLUE, BITS.NORTH]) actions.GainHealthAndClearBitsOrMSB(set()).perform_action(board, card) state_after = board.export() self.assertEqual(state_after['health'], state_before['health'] + 1) @@ -164,3 +194,24 @@ class TestActions(TestCase): 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)