Action to gain health.

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

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index 9d39fe85a4e34a2fcb7a5ad3290ce4104ad5ed7f..bbc51284cfcb9847b6cfe8e8e194d482f126c802 100644 (file)
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -6,8 +6,25 @@ 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': []}])
+        board = GameBoard.new_game([{'actions': [{
+                'action_class': 'LoseHealthOrMSB',
+                'required_bits': [],
+        }]}])
         exported_state = board.export()
         board_locations = exported_state.pop('board_locations')
         self.assertEqual(exported_state, {
@@ -15,7 +32,10 @@ class TestGameBoard(TestCase):
             'health': 4,
             'wins_required': 4,
             'wins': 0,
-            'locations': [{'actions': []}],
+            'locations': [{'actions': [{
+                'action_class': 'LoseHealthOrMSB',
+                'required_bits': [],
+            }]}],
             'player': board.player.export(),
         })
         self.assertEqual(
@@ -24,7 +44,11 @@ class TestGameBoard(TestCase):
         for location_state in board_locations.values():
             self.assertEqual(
                 sorted(location_state.keys()), ['actions', 'bitwise_operand'])
-            self.assertEqual(location_state['actions'], [])
+            self.assertEqual(location_state['actions'], [{
+                'action_class': 'LoseHealthOrMSB',
+                'required_bits': [],
+                'data': {},
+            }])
             self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3)
 
     def test_lose_health(self):
@@ -36,10 +60,30 @@ 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)
 
 
 class TestLocationCard(TestCase):
@@ -61,7 +105,9 @@ class TestLocationCard(TestCase):
 
     def test_new_location_no_actions(self):
         location = LocationCard.new_location({'actions': []})
-        self.assertEqual(location.actions, [])
+        [action] = location.actions
+        self.assertEqual(type(action), actions.DoNothing)
+        self.assertEqual(action.required_bits, set())
 
     def test_new_location_one_action(self):
         location = LocationCard.new_location({'actions': [