mirror of
https://github.com/jointakahe/takahe.git
synced 2024-11-26 09:11:00 +00:00
025fd5cf07
Emojis are now prefetched from the post, and if not, looked up individually by shortcode, to prevent loading hundreds.
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from core.html import ContentRenderer, html_to_plaintext, sanitize_html
|
|
|
|
|
|
def test_html_to_plaintext():
|
|
|
|
assert html_to_plaintext("<p>Hi!</p>") == "Hi!"
|
|
assert html_to_plaintext("<p>Hi!<br>There</p>") == "Hi!\nThere"
|
|
assert (
|
|
html_to_plaintext("<p>Hi!</p>\n\n<p>How are you?</p>") == "Hi!\n\nHow are you?"
|
|
)
|
|
|
|
assert (
|
|
html_to_plaintext("<p>Hi!</p>\n\n<p>How are<br> you?</p><p>today</p>")
|
|
== "Hi!\n\nHow are\n you?\n\ntoday"
|
|
)
|
|
|
|
|
|
def test_sanitize_post():
|
|
|
|
assert sanitize_html("<p>Hello!</p>") == "<p>Hello!</p>"
|
|
assert sanitize_html("<p>It's great</p>") == "<p>It's great</p>"
|
|
|
|
# Note that we only want to linkify things with protocol prefixes to prevent
|
|
# too many false positives.
|
|
assert sanitize_html("<p>test.com</p>") == "<p>test.com</p>"
|
|
assert (
|
|
sanitize_html("<p>https://test.com</p>")
|
|
== '<p><a href="https://test.com" rel="nofollow">https://test.com</a></p>'
|
|
)
|
|
assert (
|
|
sanitize_html("<p>@someone@subdomain.some-domain.com</p>")
|
|
== "<p>@someone@subdomain.some-domain.com</p>"
|
|
)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_link_preservation():
|
|
"""
|
|
We want to:
|
|
- Preserve incoming links from other servers
|
|
- Linkify mentions and hashtags
|
|
- Not have these all step on each other!
|
|
"""
|
|
renderer = ContentRenderer(local=True)
|
|
fake_mention = Mock()
|
|
fake_mention.username = "andrew"
|
|
fake_mention.domain_id = "aeracode.org"
|
|
fake_mention.urls.view = "/@andrew@aeracode.org/"
|
|
fake_post = Mock()
|
|
fake_post.mentions.all.return_value = [fake_mention]
|
|
fake_post.author.domain.uri_domain = "example.com"
|
|
fake_post.emojis.all.return_value = []
|
|
|
|
assert (
|
|
renderer.render_post(
|
|
'Hello @andrew, I want to link to this <span>#</span>hashtag: <a href="http://example.com/@andrew/#notahashtag">here</a> and rewrite <a href="https://example.com/tags/thishashtag/">#thishashtag</a>',
|
|
fake_post,
|
|
)
|
|
== 'Hello <a href="/@andrew@aeracode.org/">@andrew</a>, I want to link to this <a href="/tags/hashtag/" class="hashtag">#hashtag</a>: <a href="http://example.com/@andrew/#notahashtag" rel="nofollow">here</a> and rewrite <a href="/tags/thishashtag/" class="hashtag">#thishashtag</a>'
|
|
)
|