Test merchants

squirrelbattle/tests/game_test.py

@@ -7,8 +7,8 @@ import unittest
 from ..bootstrap import Bootstrap
 from ..display.display import Display
 from ..display.display_manager import DisplayManager
-from ..entities.friendly import Sunflower, Merchant
-from ..entities.items import Bomb
+from ..entities.friendly import Merchant, Sunflower
+from ..entities.items import Bomb, Sword
 from ..entities.player import Player
 from ..enums import DisplayActions
 from ..game import Game, KeyValues, GameMode
@@ -35,8 +35,11 @@ class TestGame(unittest.TestCase):
         """
         bomb = Bomb()
         self.game.map.add_entity(bomb)
-        # Add an item in the inventory to check that it is well loaded
+        sword = Sword()
+        self.game.map.add_entity(sword)
+        # Add items in the inventory to check that it is well loaded
         bomb.hold(self.game.player)
+        sword.hold(self.game.player)
 
         # Ensure that merchants can be saved
         merchant = Merchant()
@@ -125,6 +128,9 @@ class TestGame(unittest.TestCase):
         self.assertEqual(KeyValues.translate_key(
             self.game.settings.KEY_INVENTORY, self.game.settings),
             KeyValues.INVENTORY)
+        self.assertEqual(KeyValues.translate_key(
+            self.game.settings.KEY_CHAT, self.game.settings),
+            KeyValues.CHAT)
         self.assertEqual(KeyValues.translate_key(
             self.game.settings.KEY_USE, self.game.settings),
             KeyValues.USE)
@@ -288,7 +294,7 @@ class TestGame(unittest.TestCase):
         self.assertEqual(self.game.settings.KEY_LEFT_PRIMARY, 'a')
 
         # Navigate to "texture pack"
-        for ignored in range(9):
+        for ignored in range(10):
             self.game.handle_key_pressed(KeyValues.DOWN)
 
         # Change texture pack
@@ -433,8 +439,16 @@ class TestGame(unittest.TestCase):
         sunflower.move(2, 6)
         self.game.map.add_entity(sunflower)
 
+        # Does nothing
+        self.assertIsNone(self.game.handle_friendly_entity_chat(KeyValues.UP))
+
         # Talk to sunflower... or not
-        self.game.handle_key_pressed(KeyValues.T)
+        self.game.handle_key_pressed(KeyValues.CHAT)
+        self.assertTrue(self.game.waiting_for_friendly_key)
+        # Wrong key
+        self.game.handle_key_pressed(KeyValues.EQUIP)
+        self.assertFalse(self.game.waiting_for_friendly_key)
+        self.game.handle_key_pressed(KeyValues.CHAT)
         self.assertTrue(self.game.waiting_for_friendly_key)
         self.game.handle_key_pressed(KeyValues.UP)
         self.assertFalse(self.game.waiting_for_friendly_key)
@@ -442,7 +456,7 @@ class TestGame(unittest.TestCase):
         self.assertFalse(len(self.game.logs.messages) > 1)
 
         # Talk to sunflower
-        self.game.handle_key_pressed(KeyValues.T)
+        self.game.handle_key_pressed(KeyValues.CHAT)
         self.assertTrue(self.game.waiting_for_friendly_key)
         self.game.handle_key_pressed(KeyValues.DOWN)
         self.assertFalse(self.game.waiting_for_friendly_key)
@@ -451,3 +465,55 @@ class TestGame(unittest.TestCase):
         # Ensure that the message is a good message
         self.assertIn(self.game.logs.messages[1][21:],
                       Sunflower.dialogue_option)
+
+        # Test all directions to detect the friendly entity
+        self.game.player.move(3, 6)
+        self.game.handle_key_pressed(KeyValues.CHAT)
+        self.game.handle_key_pressed(KeyValues.UP)
+        self.assertEqual(len(self.game.logs.messages), 3)
+        self.game.player.move(2, 7)
+        self.game.handle_key_pressed(KeyValues.CHAT)
+        self.game.handle_key_pressed(KeyValues.LEFT)
+        self.assertEqual(len(self.game.logs.messages), 4)
+        self.game.player.move(2, 5)
+        self.game.handle_key_pressed(KeyValues.CHAT)
+        self.game.handle_key_pressed(KeyValues.RIGHT)
+        self.assertEqual(len(self.game.logs.messages), 5)
+
+    def test_talk_to_merchant(self) -> None:
+        """
+        Interact with merchants
+        """
+        self.game.state = GameMode.PLAY
+
+        merchant = Merchant()
+        merchant.move(2, 6)
+        self.game.map.add_entity(merchant)
+
+        # Does nothing
+        self.assertIsNone(self.game.handle_friendly_entity_chat(KeyValues.UP))
+
+        # Talk to merchant
+        self.game.handle_key_pressed(KeyValues.CHAT)
+        self.assertTrue(self.game.waiting_for_friendly_key)
+        self.game.handle_key_pressed(KeyValues.DOWN)
+        self.assertFalse(self.game.waiting_for_friendly_key)
+        self.assertEqual(self.game.state, GameMode.STORE)
+        self.assertTrue(self.game.logs.messages)
+
+        # Navigate in the menu
+        self.game.handle_key_pressed(KeyValues.DOWN)
+        self.game.handle_key_pressed(KeyValues.DOWN)
+        self.game.handle_key_pressed(KeyValues.UP)
+        self.assertEqual(self.game.store_menu.position, 1)
+
+        # Buy the second item
+        item = self.game.store_menu.validate()
+        self.assertIn(item, merchant.inventory)
+        self.game.handle_key_pressed(KeyValues.ENTER)
+        self.assertIn(item, self.game.player.inventory)
+        self.assertNotIn(item, merchant.inventory)
+
+        # Exit the menu
+        self.game.handle_key_pressed(KeyValues.SPACE)
+        self.assertEqual(self.game.state, GameMode.PLAY)