X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_gameboard.py;h=bd1a16c1c080c3ea95b5a075ed8f1cc2f6345ef2;hb=7ffa20720668ad9e8701d0b5f98a2e3161132423;hp=a895e18b32f484a27976fded4ad8397d4da28fa0;hpb=eccb327f1260b9aa981755840e097d5b23af05a9;p=naja.git

diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py
index a895e18..bd1a16c 100644
--- a/naja/tests/test_gameboard.py
+++ b/naja/tests/test_gameboard.py
@@ -1,7 +1,36 @@
 from unittest import TestCase
 
 from naja.constants import BITS
-from naja.gameboard import LocationCard
+from naja.gameboard import GameBoard, LocationCard
+from naja import actions
+
+
+class TestGameBoard(TestCase):
+    def test_export_new_board(self):
+        board = GameBoard.new_game(4, 4, [])
+        self.assertEqual(board.export(), {
+            'max_health': 4,
+            'health': 4,
+            'wins_required': 4,
+            'wins': 0,
+            'locations': [],
+            'board_locations': {},
+            'player': board.player.export(),
+        })
+
+    def test_lose_health(self):
+        board = GameBoard.new_game(4, 4, [])
+        self.assertEqual(board.health, 4)
+        state_1 = board.export()
+
+        board.lose_health()
+        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)
 
 
 class TestLocationCard(TestCase):
@@ -20,3 +49,15 @@ class TestLocationCard(TestCase):
         self.assertTrue(sizes[3] > 0)
         self.assertTrue(sizes[2] > sizes[3])
         self.assertEqual(bits, set(BITS.values()))
+
+    def test_new_location_no_actions(self):
+        location = LocationCard.new_location({'actions': []})
+        self.assertEqual(location.actions, [])
+
+    def test_new_location_one_action(self):
+        location = LocationCard.new_location({'actions': [
+            {'required_bits': [], 'action_class': 'DoNothing'},
+        ]})
+        [action] = location.actions
+        self.assertEqual(type(action), actions.DoNothing)
+        self.assertEqual(action.required_bits, set())