Mikaela
/
Limnoria
mirror of https://github.com/Mikaela/Limnoria.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

ChannelLogger: Add tests for non-relayed messages when rewriteRelayed is True.

This commit is contained in:
Valentin Lorentz 2020-08-30 14:52:45 +02:00
parent 01dc289cd6
commit 77682cc50b
1 changed files with 34 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
plugins/ChannelLogger/test.py
View File

@ -103,23 +103,33 @@ class ChannelLoggerTestCase(ChannelPluginTestCase):
) )
@patch_open @patch_open
def testLogRewriteRelayedEmulatedEcho(self, mock_open): def _testLogRewriteRelayedEmulatedEcho(self, mock_open, relayed):
with conf.supybot.plugins.ChannelLogger.rewriteRelayed.context(True): with conf.supybot.plugins.ChannelLogger.rewriteRelayed.context(True):
log_file = io.StringIO() log_file = io.StringIO()
mock_open.return_value = log_file mock_open.return_value = log_file
msg = ircmsgs.privmsg('#foo', '<someone> test message') msg = ircmsgs.privmsg(
'#foo', '<someone> test message', prefix='foo!bar@baz'
)
if relayed:
msg.tag('relayedMsg') msg.tag('relayedMsg')
self.irc.getCallback('ChannelLogger').outFilter(self.irc, msg) self.irc.getCallback('ChannelLogger').outFilter(self.irc, msg)
self.irc.feedMsg(msg) self.irc.feedMsg(msg)
self.assertRegex( if relayed:
log_file.getvalue(), content = '<someone> test message\n'
timestamp_re + '<someone> test message\n' else:
) content = '<foo> <someone> test message\n'
self.assertRegex(log_file.getvalue(), timestamp_re + content)
def testLogRewriteRelayedEmulatedEcho(self):
self._testLogRewriteRelayedEmulatedEcho(relayed=True)
def testLogRewriteRelayedEmulatedEchoNotRelayed(self):
self._testLogRewriteRelayedEmulatedEcho(relayed=False)
@patch_open @patch_open
def testLogRewriteRelayedRealEcho(self, mock_open): def _testLogRewriteRelayedRealEcho(self, mock_open, relayed):
with conf.supybot.plugins.ChannelLogger.rewriteRelayed.context(True): with conf.supybot.plugins.ChannelLogger.rewriteRelayed.context(True):
log_file = io.StringIO() log_file = io.StringIO()
mock_open.return_value = log_file mock_open.return_value = log_file
@ -131,19 +141,29 @@ class ChannelLoggerTestCase(ChannelPluginTestCase):
try: try:
out_msg = ircmsgs.privmsg('#foo', '<someone> test message') out_msg = ircmsgs.privmsg('#foo', '<someone> test message')
if relayed:
out_msg.tag('relayedMsg') out_msg.tag('relayedMsg')
self.irc.getCallback('ChannelLogger').outFilter(self.irc, out_msg) self.irc.getCallback('ChannelLogger').outFilter(self.irc, out_msg)
finally: finally:
self.irc.state.capabilities_ack = original_caps self.irc.state.capabilities_ack = original_caps
msg = ircmsgs.privmsg('#foo', '<someone> test message') msg = ircmsgs.privmsg(
'#foo', '<someone> test message', prefix='foo!bar@baz'
)
msg.server_tags['label'] = out_msg.server_tags['label'] msg.server_tags['label'] = out_msg.server_tags['label']
self.irc.feedMsg(msg) self.irc.feedMsg(msg)
self.assertRegex( if relayed:
log_file.getvalue(), content = '<someone> test message\n'
timestamp_re + '<someone> test message\n' else:
) content = '<foo> <someone> test message\n'
self.assertRegex(log_file.getvalue(), timestamp_re + content)
def testLogRewriteRelayedRealEcho(self):
self._testLogRewriteRelayedRealEcho(relayed=True)
def testLogRewriteRelayedRealEchoNotRelayed(self):
self._testLogRewriteRelayedRealEcho(relayed=False)
@patch_open @patch_open
def testLogNotice(self, mock_open): def testLogNotice(self, mock_open):