Correctly mock session.protocol flags

This commit is contained in:
Griatch 2018-10-14 09:33:52 +02:00
parent e05e9cda2c
commit 0b7100c3cb
2 changed files with 5 additions and 0 deletions

View file

@ -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]

View file

@ -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,