Add boop when changing actions

[naja.git] / naja / tests / test_gameboard.py

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index 07c872c611916a821c1191e33950c13129915124..5dae954fafaae21a532b7a7d842b408f6d210101 100644 (file)
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -6,6 +6,20 @@ from naja import actions
 
 
 class TestGameBoard(TestCase):
+    def assert_state(self, state1, state2, exclude=(), player_exclude=()):
+        def filter_dict(source, exclude_keys):
+            return dict((k, v) for k, v in source.items()
+                        if k not in exclude_keys)
+
+        state1 = filter_dict(state1, exclude)
+        if 'player' in state1:
+            state1['player'] = filter_dict(state1['player'], player_exclude)
+        state2 = filter_dict(state2, exclude)
+        if 'player' in state2:
+            state2['player'] = filter_dict(state2['player'], player_exclude)
+
+        self.assertEqual(state1, state2)
+
     def test_export_new_board(self):
         board = GameBoard.new_game([{'actions': [{
                 'action_class': 'LoseHealthOrMSB',
@@ -46,10 +60,69 @@ class TestGameBoard(TestCase):
         self.assertEqual(board.health, 3)
         state_2 = board.export()
 
-        # Make sure nothing else has changed.
-        state_1.pop('health')
-        state_2.pop('health')
-        self.assertEqual(state_1, state_2)
+        self.assert_state(state_1, state_2, exclude=['health'])
+
+    def test_gain_health(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.health = 2
+        self.assertEqual(board.health, 2)
+        state_1 = board.export()
+
+        board.gain_health()
+        self.assertEqual(board.health, 3)
+        state_2 = board.export()
+
+        self.assert_state(state_1, state_2, exclude=['health'])
+
+    def test_gain_health_at_max(self):
+        board = GameBoard.new_game([{'actions': []}])
+        self.assertEqual(board.health, 4)
+        state_1 = board.export()
+
+        board.gain_health()
+        self.assertEqual(board.health, 4)
+        state_2 = board.export()
+
+        self.assert_state(state_1, state_2)
+
+    def generate_locations(self, override_dict=None):
+        locations_dict = dict(((x, y), '%s%s' % (x, y))
+                              for x in range(5) for y in range(5))
+        if override_dict:
+            locations_dict.update(override_dict)
+        return locations_dict
+
+    def test_shift_locations_north(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('NORTH')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (2, 0): '21', (2, 1): '23', (2, 3): '24', (2, 4): '20',
+        }))
+
+    def test_shift_locations_south(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('SOUTH')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (2, 0): '24', (2, 1): '20', (2, 3): '21', (2, 4): '23',
+        }))
+
+    def test_shift_locations_east(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('EAST')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 2): '42', (1, 2): '02', (3, 2): '12', (4, 2): '32',
+        }))
+
+    def test_shift_locations_west(self):
+        board = GameBoard.new_game([{'actions': []}])
+        board.board_locations = self.generate_locations()
+        board.shift_locations('WEST')
+        self.assertEqual(board.board_locations, self.generate_locations({
+            (0, 2): '12', (1, 2): '32', (3, 2): '42', (4, 2): '02',
+        }))
 
 
 class TestLocationCard(TestCase):