diff --git a/tests/test_markov_ircplugin.py b/tests/test_markov_ircplugin.py index d40daba..63994f6 100644 --- a/tests/test_markov_ircplugin.py +++ b/tests/test_markov_ircplugin.py @@ -73,6 +73,23 @@ class MarkovTestCase(TestCase): self.assertEqual(mock_learn_line.call_args.args[0], 'hello this is a test message') + def test_learn_bridge_no_edit(self): + """Test that if a message is from the bridge, we learn what looks like a speaker's nick. + + This is primarily for if someone on IRC is pasting a log --- the " whatever" should + be retained in full, rather than learned as "whatever". + """ + mock_event = mock.MagicMock() + mock_event.arguments = [' hello this is a test message'] + mock_event.target = '#test' + mock_event.recursing = False + mock_event.source = 'not-bridge!not-bridge@localhost' + + with mock.patch('markov.lib.learn_line') as mock_learn_line: + self.plugin.handle_chatter(self.mock_connection, mock_event) + + self.assertEqual(mock_learn_line.call_args.args[0], ' hello this is a test message') + def test_learn_bridge_and_self_edit(self): """Test that we don't learn our own name when learning something addressed to us, discord style.""" mock_event = mock.MagicMock()