Testing adventure dungeon

evennia/contrib/tutorials/evadventure/tests/test_dungeon.py

@@ -0,0 +1,97 @@
+"""
+Test Dungeon orchestrator / procedurally generated dungeon rooms.
+
+"""
+
+from unittest.mock import MagicMock
+
+from evennia.utils.create import create_object
+from evennia.utils.test_resources import BaseEvenniaTest
+from evennia.utils.utils import inherits_from
+
+from .. import dungeon
+from ..rooms import EvAdventureDungeonRoom
+from .mixins import EvAdventureMixin
+
+
+class TestDungeon(EvAdventureMixin, BaseEvenniaTest):
+    """
+    Test with a starting room and a character moving through the dungeon,
+    generating more and more rooms as they go.
+
+    """
+
+    def setUp(self):
+        """
+        Create a start room with exits leading away from it
+
+        """
+        super().setUp()
+        droomclass = dungeon.EvAdventureDungeonRoomStart
+        droomclass.recycle_time = 0  # disable the tick
+
+        self.start_room = create_object(droomclass, key="bottom of well")
+
+        self.assertEqual(
+            self.start_room.scripts.get("evadventure_startroom_resetter")[0].interval, -1
+        )
+        self.start_north = create_object(
+            dungeon.EvAdventureStartRoomExit,
+            key="north",
+            location=self.start_room,
+            destination=self.start_room,
+        )
+        self.start_north
+        self.start_south = create_object(
+            dungeon.EvAdventureStartRoomExit,
+            key="south",
+            location=self.start_room,
+            destination=self.start_room,
+        )
+        self.character.location = self.start_room
+
+    def _move_character(self, direction):
+        old_location = self.character.location
+        for exi in old_location.exits:
+            if exi.key == direction:
+                # by setting target to old-location we trigger the
+                # special behavior of this Exit type
+                exi.at_traverse(self.character, old_location)
+                break
+        return self.character.location
+
+    def test_start_room(self):
+        """
+        Test move through one of the start room exits.
+
+        """
+        # begin in start room
+        self.assertEqual(self.character.location, self.start_room)
+
+        # first go north, this should generate a new room
+        new_room_north = self._move_character("north")
+        self.assertNotEqual(self.start_room, new_room_north)
+        self.assertTrue(inherits_from(new_room_north, EvAdventureDungeonRoom))
+
+        # check if Orchestrator was created
+        orchestrator = self.start_north.scripts.get(dungeon.EvAdventureDungeonOrchestrator)
+        self.assertTrue(bool(orchestrator))
+        self.assertTrue(orchestrator.key.startswith("dungeon_orchestrator_north_"))
+
+    def test_different_start_directions(self):
+        # first go north, this should generate a new room
+        new_room_north = self._move_character("north")
+        self.assertNotEqual(self.start_room, new_room_north)
+
+        # back to start room
+        start_room = self._move_character("south")
+        self.assertEqual(self.start_room, start_room)
+
+        # next go south, this should generate a new room
+        new_room_south = self._move_character("south")
+        self.assertNotEqual(self.start_room, new_room_south)
+        self.assertNotEqual(new_room_north, new_room_south)
+
+        # back to start room again
+        start_room = self._move_character("north")
+        self.assertEqual(self.start_room, start_room)