31 lines
921 B
Python
31 lines
921 B
Python
from unittest import TestCase
|
|
|
|
from taleweave.models.entity import Room, World
|
|
from taleweave.utils.search import find_room
|
|
|
|
|
|
class TestFindRoom(TestCase):
|
|
def test_existing_room(self):
|
|
world = World(name="Test World", rooms=[], theme="testing", order=[])
|
|
room = Room(
|
|
name="Test Room",
|
|
description="A test room.",
|
|
characters=[],
|
|
items=[],
|
|
portals=[],
|
|
)
|
|
world.rooms.append(room)
|
|
self.assertEqual(find_room(world, "Test Room"), room)
|
|
|
|
def test_missing_room(self):
|
|
world = World(name="Test World", rooms=[], theme="testing", order=[])
|
|
room = Room(
|
|
name="Test Room",
|
|
description="A test room.",
|
|
characters=[],
|
|
items=[],
|
|
portals=[],
|
|
)
|
|
world.rooms.append(room)
|
|
self.assertEqual(find_room(world, "Missing Room"), None)
|