1 from unittest import TestCase
3 from naja.constants import BITS
4 from naja.gameboard import GameBoard, LocationCard
5 from naja.player import Player
6 from naja import actions
9 class TestActions(TestCase):
10 def make_player(self, *bits):
11 return Player(sum(1 << bit for bit in bits), None)
13 def make_board(self, player_bits=None, locations=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)
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)
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)
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)
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)
46 state_before.pop('health')
47 state_after.pop('health')
48 self.assertEqual(state_before, state_after)
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)
57 state_before['player'].pop('bits')
58 state_after['player'].pop('bits')
59 self.assertEqual(state_before, state_after)
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()
68 board.player.bits.check_bits([BITS.MSB, BITS.NORTH]), True)
70 state_before['player'].pop('bits')
71 state_after['player'].pop('bits')
72 self.assertEqual(state_before, state_after)
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)
83 state_before['player'].pop('bits')
84 state_after['player'].pop('bits')
85 self.assertEqual(state_before, state_after)