naja/tests/test_actions.py

   1 from unittest import TestCase
   2
   3 from naja.constants import BITS
   4 from naja.gameboard import GameBoard, LocationCard
   5 from naja.player import Player
   6 from naja import actions
   7
   8
   9 class TestActions(TestCase):
  10     def make_player(self, *bits):
  11         player_bits = 0
  12         for bit in bits:
  13             player_bits |= bit
  14         return Player(player_bits, None)
  15
  16     def make_board(self, player_bits=None, locations=None):
  17         if locations is None:
  18             locations = [{'actions': []}]
  19         board = GameBoard.new_game(locations)
  20         if player_bits is not None:
  21             board.player.bits.set_bits(player_bits)
  22         return board
  23
  24     def test_check_available(self):
  25         def check_available(action_bits, player_bits, expected_result):
  26             action = actions.LocationAction(action_bits)
  27             player = self.make_player(*player_bits)
  28             self.assertEqual(action.check_available(player), expected_result)
  29
  30         check_available(set(), [], True)
  31         check_available(set(), [BITS.MSB], True)
  32         check_available(set([BITS.MSB]), [], False)
  33         check_available(set([BITS.MSB]), [BITS.MSB], True)
  34
  35     def test_bits_translation(self):
  36         action = actions.LocationAction(set([BITS.NORTH, 'MSB']))
  37         self.assertEqual(action.required_bits, set([BITS.NORTH, BITS.MSB]))
  38
  39     def test_DoNothing(self):
  40         board = self.make_board()
  41         state_before = board.export()
  42         actions.DoNothing(set()).perform_action(board, None)
  43         state_after = board.export()
  44         self.assertEqual(state_before, state_after)
  45
  46     def test_LoseHealthOrMSB_MSB_clear(self):
  47         board = self.make_board()
  48         state_before = board.export()
  49         actions.LoseHealthOrMSB(set()).perform_action(board, None)
  50         state_after = board.export()
  51         self.assertEqual(state_after['health'], state_before['health'] - 1)
  52
  53         state_before.pop('health')
  54         state_after.pop('health')
  55         self.assertEqual(state_before, state_after)
  56
  57     def test_LoseHealthOrMSB_MSB_set(self):
  58         board = self.make_board(player_bits=[BITS.MSB])
  59         state_before = board.export()
  60         actions.LoseHealthOrMSB(set()).perform_action(board, None)
  61         state_after = board.export()
  62         self.assertEqual(board.player.bits.check_bit(BITS.MSB), False)
  63
  64         state_before['player'].pop('bits')
  65         state_after['player'].pop('bits')
  66         self.assertEqual(state_before, state_after)
  67
  68     def test_SetBits(self):
  69         board = self.make_board()
  70         state_before = board.export()
  71         location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
  72         actions.SetBits(set()).perform_action(board, location)
  73         state_after = board.export()
  74         self.assertEqual(
  75             board.player.bits.check_bits([BITS.MSB, BITS.NORTH]), True)
  76
  77         state_before['player'].pop('bits')
  78         state_after['player'].pop('bits')
  79         self.assertEqual(state_before, state_after)
  80
  81     def test_ToggleBits(self):
  82         board = self.make_board(player_bits=[BITS.NORTH])
  83         state_before = board.export()
  84         location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
  85         actions.ToggleBits(set()).perform_action(board, location)
  86         state_after = board.export()
  87         self.assertEqual(board.player.bits.check_bit(BITS.MSB), True)
  88         self.assertEqual(board.player.bits.check_bit(BITS.NORTH), False)
  89
  90         state_before['player'].pop('bits')
  91         state_after['player'].pop('bits')
  92         self.assertEqual(state_before, state_after)