X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_actions.py;h=5cf4c76011958267a324087a7a5eae9706fe6b19;hb=cfe014c380c8a5be0062f5bfcd37cc053dd6d56f;hp=02c07964342df9f4d06ee6c7cca4427d49e62edd;hpb=2a950900b3b7ae9514e5351c3a07903611b9153e;p=naja.git

diff --git a/naja/tests/test_actions.py b/naja/tests/test_actions.py
index 02c0796..5cf4c76 100644
--- a/naja/tests/test_actions.py
+++ b/naja/tests/test_actions.py
@@ -50,6 +50,33 @@ class TestActions(TestCase):
         check_available(set([BITS.MSB]), [], False)
         check_available(set([BITS.MSB]), [BITS.MSB], True)
 
+    def test_get_text(self):
+        class LocationAction1(actions.LocationAction):
+            TEXT = "foo"
+        action1 = LocationAction1([])
+        self.assertEqual(action1.get_text(), "foo")
+
+        class LocationAction2(actions.LocationAction):
+            TEXT = "foo %(bar)s"
+        action2 = LocationAction2([], bar="baz")
+        self.assertEqual(action2.get_text(), "foo baz")
+
+    def test_get_text_row_column(self):
+        class DirectionAction(actions.LocationAction):
+            TEXT = "foo %(direction)s %(rowcol)s"
+
+        action_north = DirectionAction([], direction='NORTH')
+        self.assertEqual(action_north.get_text(), "foo NORTH column")
+
+        action_south = DirectionAction([], direction='SOUTH')
+        self.assertEqual(action_south.get_text(), "foo SOUTH column")
+
+        action_east = DirectionAction([], direction='EAST')
+        self.assertEqual(action_east.get_text(), "foo EAST row")
+
+        action_west = DirectionAction([], direction='WEST')
+        self.assertEqual(action_west.get_text(), "foo WEST row")
+
     def test_bits_translation(self):
         action = actions.LocationAction(set([BITS.NORTH, 'MSB']))
         self.assertEqual(action.required_bits, set([BITS.NORTH, BITS.MSB]))
@@ -100,11 +127,11 @@ class TestActions(TestCase):
     def test_LoseHealthOrMSBAndSetBits_MSB_clear(self):
         board = self.make_board(player_bits=[])
         state_before = board.export()
-        card = LocationCard(set([BITS.CYAN, BITS.NORTH]), [])
+        card = LocationCard(set([BITS.BLUE, BITS.NORTH]), [])
         actions.LoseHealthOrMSBAndSetBits(set()).perform_action(board, card)
         state_after = board.export()
         self.assertEqual(state_after['health'], state_before['health'] - 1)
-        self.assert_player_bits(board, BITS.CYAN, BITS.NORTH)
+        self.assert_player_bits(board, BITS.BLUE, BITS.NORTH)
         self.assert_state(
             state_before, state_after, exclude=['health'],
             player_exclude=['bits'])
@@ -112,10 +139,10 @@ class TestActions(TestCase):
     def test_LoseHealthOrMSBAndSetBits_MSB_set(self):
         board = self.make_board(player_bits=[BITS.MSB])
         state_before = board.export()
-        card = LocationCard(set([BITS.CYAN, BITS.NORTH]), [])
+        card = LocationCard(set([BITS.BLUE, BITS.NORTH]), [])
         actions.LoseHealthOrMSBAndSetBits(set()).perform_action(board, card)
         state_after = board.export()
-        self.assert_player_bits(board, BITS.CYAN, BITS.NORTH)
+        self.assert_player_bits(board, BITS.BLUE, BITS.NORTH)
         self.assert_state(state_before, state_after, player_exclude=['bits'])
 
     def test_LoseHealthOrMSBAndSetBits_MSB_set_and_on_card(self):
@@ -129,7 +156,7 @@ class TestActions(TestCase):
 
     def test_AcquireWinToken(self):
         board = self.make_board(
-            player_bits=[BITS.CYAN, BITS.MAGENTA, BITS.YELLOW])
+            player_bits=[BITS.RED, BITS.GREEN, BITS.BLUE])
         state_before = board.export()
         actions.AcquireWinToken(set()).perform_action(board, None)
         state_after = board.export()
@@ -143,7 +170,7 @@ class TestActions(TestCase):
         board = self.make_board(player_bits=[BITS.NORTH])
         board.lose_health()
         state_before = board.export()
-        card = LocationCard(set([BITS.CYAN, BITS.NORTH]), [])
+        card = LocationCard(set([BITS.BLUE, BITS.NORTH]), [])
         actions.GainHealthAndClearBitsOrMSB(set()).perform_action(board, card)
         state_after = board.export()
         self.assertEqual(state_after['health'], state_before['health'] + 1)
@@ -156,7 +183,7 @@ class TestActions(TestCase):
         board = self.make_board(player_bits=[BITS.MSB, BITS.NORTH])
         board.lose_health()
         state_before = board.export()
-        card = LocationCard(set([BITS.CYAN, BITS.NORTH]), [])
+        card = LocationCard(set([BITS.BLUE, BITS.NORTH]), [])
         actions.GainHealthAndClearBitsOrMSB(set()).perform_action(board, card)
         state_after = board.export()
         self.assertEqual(state_after['health'], state_before['health'] + 1)