from unittest import TestCase
+from naja.constants import BITS
+from naja.gameboard import GameBoard, LocationCard
from naja.player import Player
from naja import actions
class TestActions(TestCase):
def make_player(self, *bits):
- return Player(sum(1 << bit for bit in bits), None)
+ player_bits = 0
+ for bit in bits:
+ player_bits |= bit
+ return Player(player_bits, None)
- def test_DoNothing_check_available(self):
- player = self.make_player()
- action = actions.DoNothing(set())
- self.assertEqual(action.check_available(player), True)
+ def make_board(self, player_bits=None, locations=None):
+ if locations is None:
+ locations = [{'actions': []}]
+ board = GameBoard.new_game(locations)
+ if player_bits is not None:
+ board.player.bits.set_bits(player_bits)
+ return board
- def test_LoseHealthOrMSB_check_available(self):
- player = self.make_player()
- action = actions.LoseHeathOrMSB(set())
- self.assertEqual(action.check_available(player), True)
+ def test_check_available(self):
+ def check_available(action_bits, player_bits, expected_result):
+ action = actions.LocationAction(action_bits)
+ player = self.make_player(*player_bits)
+ self.assertEqual(action.check_available(player), expected_result)
+
+ check_available(set(), [], True)
+ check_available(set(), [BITS.MSB], True)
+ 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_DoNothing(self):
+ board = self.make_board()
+ state_before = board.export()
+ actions.DoNothing(set()).perform_action(board, None)
+ state_after = board.export()
+ self.assertEqual(state_before, state_after)
+
+ def test_LoseHealthOrMSB_MSB_clear(self):
+ board = self.make_board()
+ state_before = board.export()
+ actions.LoseHealthOrMSB(set()).perform_action(board, None)
+ state_after = board.export()
+ self.assertEqual(state_after['health'], state_before['health'] - 1)
+
+ state_before.pop('health')
+ state_after.pop('health')
+ self.assertEqual(state_before, state_after)
+
+ def test_LoseHealthOrMSB_MSB_set(self):
+ board = self.make_board(player_bits=[BITS.MSB])
+ state_before = board.export()
+ actions.LoseHealthOrMSB(set()).perform_action(board, None)
+ state_after = board.export()
+ self.assertEqual(board.player.bits.check_bit(BITS.MSB), False)
+
+ state_before['player'].pop('bits')
+ state_after['player'].pop('bits')
+ self.assertEqual(state_before, state_after)
+
+ def test_SetBits(self):
+ board = self.make_board()
+ state_before = board.export()
+ location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
+ actions.SetBits(set()).perform_action(board, location)
+ state_after = board.export()
+ self.assertEqual(
+ board.player.bits.check_bits([BITS.MSB, BITS.NORTH]), True)
+
+ state_before['player'].pop('bits')
+ state_after['player'].pop('bits')
+ self.assertEqual(state_before, state_after)
+
+ def test_ToggleBits(self):
+ board = self.make_board(player_bits=[BITS.NORTH])
+ state_before = board.export()
+ location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
+ actions.ToggleBits(set()).perform_action(board, location)
+ state_after = board.export()
+ self.assertEqual(board.player.bits.check_bit(BITS.MSB), True)
+ self.assertEqual(board.player.bits.check_bit(BITS.NORTH), False)
+
+ state_before['player'].pop('bits')
+ state_after['player'].pop('bits')
+ self.assertEqual(state_before, state_after)