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         return Player(sum(1 << bit for bit in bits), None)
  12
  13     def make_board(self, player_bits=None, locations=None):
  14         if locations is None:
  15             locations = [{'actions': []}]
  16         board = GameBoard.new_game(locations)
  17         if player_bits is not None:
  18             board.player.bits.set_bits(player_bits)
  19         return board
  20
  21     def test_check_available(self):
  22         def check_available(action_bits, player_bits, expected_result):
  23             action = actions.LocationAction(action_bits)
  24             player = self.make_player(*player_bits)
  25             self.assertEqual(action.check_available(player), expected_result)
  26
  27         check_available(set(), [], True)
  28         check_available(set(), [BITS.MSB], True)
  29         check_available(set([BITS.MSB]), [], False)
  30         check_available(set([BITS.MSB]), [BITS.MSB], True)
  31
  32     def test_DoNothing(self):
  33         board = self.make_board()
  34         state_before = board.export()
  35         actions.DoNothing(set()).perform_action(board, None)
  36         state_after = board.export()
  37         self.assertEqual(state_before, state_after)
  38
  39     def test_LoseHealthOrMSB_MSB_clear(self):
  40         board = self.make_board()
  41         state_before = board.export()
  42         actions.LoseHeathOrMSB(set()).perform_action(board, None)
  43         state_after = board.export()
  44         self.assertEqual(state_after['health'], state_before['health'] - 1)
  45
  46         state_before.pop('health')
  47         state_after.pop('health')
  48         self.assertEqual(state_before, state_after)
  49
  50     def test_LoseHealthOrMSB_MSB_set(self):
  51         board = self.make_board(player_bits=[BITS.MSB])
  52         state_before = board.export()
  53         actions.LoseHeathOrMSB(set()).perform_action(board, None)
  54         state_after = board.export()
  55         self.assertEqual(board.player.bits.check_bit(BITS.MSB), False)
  56
  57         state_before['player'].pop('bits')
  58         state_after['player'].pop('bits')
  59         self.assertEqual(state_before, state_after)
  60
  61     def test_SetBits(self):
  62         board = self.make_board()
  63         state_before = board.export()
  64         location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
  65         actions.SetBits(set()).perform_action(board, location)
  66         state_after = board.export()
  67         self.assertEqual(
  68             board.player.bits.check_bits([BITS.MSB, BITS.NORTH]), True)
  69
  70         state_before['player'].pop('bits')
  71         state_after['player'].pop('bits')
  72         self.assertEqual(state_before, state_after)
  73
  74     def test_ToggleBits(self):
  75         board = self.make_board(player_bits=[BITS.NORTH])
  76         state_before = board.export()
  77         location = LocationCard(set([BITS.MSB, BITS.NORTH]), [])
  78         actions.ToggleBits(set()).perform_action(board, location)
  79         state_after = board.export()
  80         self.assertEqual(board.player.bits.check_bit(BITS.MSB), True)
  81         self.assertEqual(board.player.bits.check_bit(BITS.NORTH), False)
  82
  83         state_before['player'].pop('bits')
  84         state_after['player'].pop('bits')
  85         self.assertEqual(state_before, state_after)