Archipelago/test/webhost/test_markdown.py at crosscode-dev · CodeTriangle/Archipelago · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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("![Image](image.png)")
        self.assertIn(f'src="{self.base_url}/image.png"', html)

    def test_absolute_img_no_rewrite(self) -> None:
        html = self.markdown("![Image](/image.png)")
        self.assertIn(f'src="/image.png"', html)
        self.assertNotIn(self.base_url, html)

    def test_remote_img_no_rewrite(self) -> None:
        html = self.markdown("![Image](https://example.com/image.png)")
        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("![Image](../../generic/docs/image.png)")
        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("![Image](image.png)".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("![Image](image.png)".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)