79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
|
import os
|
||
|
|
import unittest
|
||
|
|
from tempfile import NamedTemporaryFile
|
||
|
|
|
||
|
|
from mistune import HTMLRenderer, Markdown
|
||
|
|
|
||
|
|
from WebHostLib.markdown import ImgUrlRewriteInlineParser, render_markdown
|
||
|
|
|
||
|
|
|
||
|
|
class ImgUrlRewriteTest(unittest.TestCase):
|
||
|
|
markdown: Markdown
|
||
|
|
base_url = "/static/generated/docs/some_game"
|
||
|
|
|
||
|
|
def setUp(self) -> None:
|
||
|
|
self.markdown = Markdown(
|
||
|
|
renderer=HTMLRenderer(escape=False),
|
||
|
|
inline=ImgUrlRewriteInlineParser(self.base_url),
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_relative_img_rewrite(self) -> None:
|
||
|
|
html = self.markdown("")
|
||
|
|
self.assertIn(f'src="{self.base_url}/image.png"', html)
|
||
|
|
|
||
|
|
def test_absolute_img_no_rewrite(self) -> None:
|
||
|
|
html = self.markdown("")
|
||
|
|
self.assertIn(f'src="/image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
|
||
|
|
def test_remote_img_no_rewrite(self) -> None:
|
||
|
|
html = self.markdown("")
|
||
|
|
self.assertIn(f'src="https://example.com/image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
|
||
|
|
def test_relative_link_no_rewrite(self) -> None:
|
||
|
|
# The parser is only supposed to update images, not links.
|
||
|
|
html = self.markdown("[Link](image.png)")
|
||
|
|
self.assertIn(f'href="image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
|
||
|
|
def test_absolute_link_no_rewrite(self) -> None:
|
||
|
|
html = self.markdown("[Link](/image.png)")
|
||
|
|
self.assertIn(f'href="/image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
|
||
|
|
def test_auto_link_no_rewrite(self) -> None:
|
||
|
|
html = self.markdown("<https://example.com/image.png>")
|
||
|
|
self.assertIn(f'href="https://example.com/image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
|
||
|
|
def test_relative_img_to_other_game(self) -> None:
|
||
|
|
html = self.markdown("")
|
||
|
|
self.assertIn(f'src="{self.base_url}/../Archipelago/image.png"', html)
|
||
|
|
|
||
|
|
|
||
|
|
class RenderMarkdownTest(unittest.TestCase):
|
||
|
|
"""Tests that render_markdown does the right thing."""
|
||
|
|
base_url = "/static/generated/docs/some_game"
|
||
|
|
|
||
|
|
def test_relative_img_rewrite(self) -> None:
|
||
|
|
f = NamedTemporaryFile(delete=False)
|
||
|
|
try:
|
||
|
|
f.write("".encode("utf-8"))
|
||
|
|
f.close()
|
||
|
|
html = render_markdown(f.name, self.base_url)
|
||
|
|
self.assertIn(f'src="{self.base_url}/image.png"', html)
|
||
|
|
finally:
|
||
|
|
os.unlink(f.name)
|
||
|
|
|
||
|
|
def test_no_img_rewrite(self) -> None:
|
||
|
|
f = NamedTemporaryFile(delete=False)
|
||
|
|
try:
|
||
|
|
f.write("".encode("utf-8"))
|
||
|
|
f.close()
|
||
|
|
html = render_markdown(f.name)
|
||
|
|
self.assertIn(f'src="image.png"', html)
|
||
|
|
self.assertNotIn(self.base_url, html)
|
||
|
|
finally:
|
||
|
|
os.unlink(f.name)
|