From 0b7100c3cb8a444ddb265b366f024ec6a92f93de Mon Sep 17 00:00:00 2001 From: Griatch Date: Sun, 14 Oct 2018 09:33:52 +0200 Subject: [PATCH] Correctly mock session.protocol flags --- evennia/utils/evmenu.py | 1 + evennia/utils/tests/test_evmenu.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/evennia/utils/evmenu.py b/evennia/utils/evmenu.py index 0b4e21ffa2..dbd5e88794 100644 --- a/evennia/utils/evmenu.py +++ b/evennia/utils/evmenu.py @@ -1045,6 +1045,7 @@ class EvMenu(object): node (str): The formatted node to display. """ + if self._session: screen_width = self._session.protocol_flags.get( "SCREENWIDTH", {0: _MAX_TEXT_WIDTH})[0] diff --git a/evennia/utils/tests/test_evmenu.py b/evennia/utils/tests/test_evmenu.py index a6959c0509..910e83312a 100644 --- a/evennia/utils/tests/test_evmenu.py +++ b/evennia/utils/tests/test_evmenu.py @@ -169,7 +169,11 @@ class TestEvMenu(TestCase): self.caller.msg = MagicMock() self.caller2.msg = MagicMock() self.session = MagicMock() + self.session.protocol_flags = {} self.session2 = MagicMock() + self.session2.protocol_flags = {} + self.caller.session = self.session + self.caller2.session = self.session2 self.menu = evmenu.EvMenu(self.caller, self.menutree, startnode=self.startnode, cmdset_mergetype=self.cmdset_mergetype,