Annikka/evennia
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use global mocks to assert called callbacks.

Browse source
This commit is contained in:
henddher 2022-10-09 23:10:03 -05:00
parent f6e6621463
commit 7944ed833e
1 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

19
evennia/contrib/grid/xyzgrid/tests.py
View file

@ -4,7 +4,6 @@ Tests for the XYZgrid system.
""" """
from unittest import mock from unittest import mock
from unittest.mock import patch
from random import randint from random import randint
from parameterized import parameterized from parameterized import parameterized
@ -1419,13 +1418,16 @@ class TestBuildExampleGrid(BaseEvenniaTest):
self.assertTrue(room2b.db.desc.startswith("Sunlight sifts down")) self.assertTrue(room2b.db.desc.startswith("Sunlight sifts down"))
mock_room_callbacks = mock.MagicMock()
mock_exit_callbacks = mock.MagicMock()
class TestXyzRoom(xyzroom.XYZRoom): class TestXyzRoom(xyzroom.XYZRoom):
def at_object_creation(self): def at_object_creation(self):
pass mock_room_callbacks.at_object_creation()
class TestXyzExit(xyzroom.XYZExit): class TestXyzExit(xyzroom.XYZExit):
def at_object_creation(self): def at_object_creation(self):
pass mock_exit_callbacks.at_object_creation()
MAP_DATA = { MAP_DATA = {
"map": """ "map": """
@ -1473,7 +1475,8 @@ class TestCallbacks(BaseEvenniaTest):
# }) # })
def setUp(self): def setUp(self):
super().setUp() super().setUp()
# self.setup_grid(MAP_DATA) mock_room_callbacks.reset_mock()
mock_exit_callbacks.reset_mock()
def setup_grid(self, map_data): def setup_grid(self, map_data):
# from evennia.prototypes import prototypes as protlib # from evennia.prototypes import prototypes as protlib
@ -1505,9 +1508,7 @@ class TestCallbacks(BaseEvenniaTest):
# XYZROOM_PROTOTYPE_OVERRIDE={ # XYZROOM_PROTOTYPE_OVERRIDE={
# "typeclass": "evennia.contrib.grid.xyzgrid.tests.TestXyzRoom", # "typeclass": "evennia.contrib.grid.xyzgrid.tests.TestXyzRoom",
# }) # })
# @patch('evennia.contrib.grid.xyzgrid.tests.TestXyzRoom') def test_typeclassed_xyzroom_and_xyzexit_with_at_object_creation_are_called(self):
# def test_dummy(self, MockTestXyzRoom):
def test_dummy(self):
map_data = dict(MAP_DATA) map_data = dict(MAP_DATA)
for prototype_key, prototype_value in map_data["prototypes"].items(): for prototype_key, prototype_value in map_data["prototypes"].items():
if len(prototype_key) == 2: if len(prototype_key) == 2:
@ -1518,3 +1519,7 @@ class TestCallbacks(BaseEvenniaTest):
self.setup_grid(map_data) self.setup_grid(map_data)
self.grid.spawn() self.grid.spawn()
# Two rooms and 2 exits.
self.assertEqual(mock_room_callbacks.at_object_creation.mock_calls, [mock.call(), mock.call()])
# self.assertEqual(mock_exit_callbacks.at_object_creation.mock_calls, [mock.call(), mock.call()])