Misago/misago/markup/tests/test_mentions.py

from ...users.test import AuthenticatedUserTestCase
from ..mentions import add_mentions


class MockRequest:
    def __init__(self, user):
        self.user = user


class MentionsTests(AuthenticatedUserTestCase):
    def test_single_mention(self):
        """markup extension parses single mention"""
        TEST_CASES = [
            ("<p>Hello, @%s!</p>", '<p>Hello, <a href="%s">@%s</a>!</p>'),
            ("<h1>Hello, @%s!</h1>", '<h1>Hello, <a href="%s">@%s</a>!</h1>'),
            ("<div>Hello, @%s!</div>", '<div>Hello, <a href="%s">@%s</a>!</div>'),
            (
                "<h1>Hello, <strong>@%s!</strong></h1>",
                '<h1>Hello, <strong><a href="%s">@%s</a>!</strong></h1>',
            ),
            (
                "<h1>Hello, <strong>@%s</strong>!</h1>",
                '<h1>Hello, <strong><a href="%s">@%s</a></strong>!</h1>',
            ),
        ]

        for before, after in TEST_CASES:
            result = {"parsed_text": before % self.user.username, "mentions": []}

            add_mentions(MockRequest(self.user), result)

            expected_outcome = after % (
                self.user.get_absolute_url(),
                self.user.username,
            )
            self.assertEqual(result["parsed_text"], expected_outcome)
            self.assertEqual(result["mentions"], [self.user])

    def test_invalid_mentions(self):
        """markup extension leaves invalid mentions alone"""
        TEST_CASES = [
            "<p>Hello, Bob!</p>",
            "<p>Hello, @Bob!</p>",
            '<p>Hello, <a href="/">@%s</a>!</p>' % self.user.username,
            '<p>Hello, <a href="/"><b>@%s</b></a>!</p>' % self.user.username,
        ]

        for markup in TEST_CASES:
            result = {"parsed_text": markup, "mentions": []}

            add_mentions(MockRequest(self.user), result)

            self.assertEqual(result["parsed_text"], markup)
            self.assertFalse(result["mentions"])

    def test_multiple_mentions(self):
        """markup extension handles multiple mentions"""
        before = "<p>Hello @{0} and @{0}, how is it going?</p>".format(
            self.user.username
        )

        after = (
            # pylint: disable=line-too-long
            '<p>Hello <a href="{0}">@{1}</a> and <a href="{0}">@{1}</a>, how is it going?</p>'
        ).format(self.user.get_absolute_url(), self.user.username)

        result = {"parsed_text": before, "mentions": []}

        add_mentions(MockRequest(self.user), result)
        self.assertEqual(result["parsed_text"], after)
        self.assertEqual(result["mentions"], [self.user])

    def test_repeated_mention(self):
        """markup extension handles mentions across document"""
        before = "<p>Hello @{0}</p><p>@{0}, how is it going?</p>".format(
            self.user.username
        )

        after = (
            # pylint: disable=line-too-long
            '<p>Hello <a href="{0}">@{1}</a></p><p><a href="{0}">@{1}</a>, how is it going?</p>'
        ).format(self.user.get_absolute_url(), self.user.username)

        result = {"parsed_text": before, "mentions": []}

        add_mentions(MockRequest(self.user), result)
        self.assertEqual(result["parsed_text"], after)
        self.assertEqual(result["mentions"], [self.user])