fix: Fix nickname highlight in group chats

Fixes #11413

1 files changed, 21 insertions, 7 deletions

diff --git a/test/common/test_message_needs_highlight.py b/test/common/test_message_needs_highlight.py
index 084c44645..931bd0a9d 100644
--- a/test/common/test_message_needs_highlight.py
+++ b/test/common/test_message_needs_highlight.py
@@ -21,12 +21,19 @@ class HighlightTest(unittest.TestCase):
         t_text7 = 'Did you test this?'
         t_text8 = 'Hi romeo'
         t_text9 = 'My address is juliet@xmppserver'
-
-        f_text1 = 'RomeoDoes this work?'
-        f_text2 = ''
-        f_text3 = 'https://romeo.tld'
-        f_text4 = 'https://romeo.tld message'
-        f_text5 = 'https://test.tld/where-is-romeo'
+        t_text10 = 'Romeo, asd'
+        t_text11 = 'Romeo,'
+        t_text12 = 'Romeo,hi'
+        t_text13 = '@Romeo'
+        t_text14 = '#Romeo'
+
+        f_text1 = ''
+        f_text2 = 'RomeoDoes this work?'
+        f_text3 = 'nRomeo'
+        f_text4 = 'nRomeoa'
+        f_text_url_1 = 'https://romeo.tld'
+        f_text_url_2 = 'https://romeo.tld message'
+        f_text_url_3 = 'https://test.tld/where-is-romeo'
 
         self.assertTrue(message_needs_highlight(t_text1, NICK, JID))
         self.assertTrue(message_needs_highlight(t_text2, NICK, JID))
@@ -37,9 +44,16 @@ class HighlightTest(unittest.TestCase):
         self.assertTrue(message_needs_highlight(t_text7, NICK, JID))
         self.assertTrue(message_needs_highlight(t_text8, NICK, JID))
         self.assertTrue(message_needs_highlight(t_text9, NICK, JID))
+        self.assertTrue(message_needs_highlight(t_text10, NICK, JID))
+        self.assertTrue(message_needs_highlight(t_text11, NICK, JID))
+        self.assertTrue(message_needs_highlight(t_text12, NICK, JID))
+        self.assertTrue(message_needs_highlight(t_text13, NICK, JID))
+        self.assertTrue(message_needs_highlight(t_text14, NICK, JID))
 
         self.assertFalse(message_needs_highlight(f_text1, NICK, JID))
         self.assertFalse(message_needs_highlight(f_text2, NICK, JID))
         self.assertFalse(message_needs_highlight(f_text3, NICK, JID))
         self.assertFalse(message_needs_highlight(f_text4, NICK, JID))
-        self.assertFalse(message_needs_highlight(f_text5, NICK, JID))
+        self.assertFalse(message_needs_highlight(f_text_url_1, NICK, JID))
+        self.assertFalse(message_needs_highlight(f_text_url_2, NICK, JID))
+        self.assertFalse(message_needs_highlight(f_text_url_3, NICK, JID))