from unittest import TestCase, main
from lonk import HtmlToGmi


def fn_media_url(mime, img_url):
    return img_url


class TestHtmlToGmi(TestCase):
    def test_br(self):
        html = '<p>head<br>tail</p>'
        self.assertEqual(HtmlToGmi("https://localhost/api", fn_media_url).feed(html), """\
head

tail
""")

    def test_img_realtive(self):
        html = '<img src="/d/xxxx.jpg" title="yyy">'
        self.assertEqual(HtmlToGmi("https://localhost/api", fn_media_url).feed(html), "=> https://localhost/d/xxxx.jpg yyy")

    def test_link_realtive(self):
        html = '<p>head <a href="/sub1/1">https link</a> tail</p>'
        self.assertEqual(HtmlToGmi("https://localhost/api", fn_media_url).feed(html), """\
head ↳ https link tail
=> https://localhost/sub1/1 https link
""")

    def test_links_in_paragraph(self):
        html = '<p>head <a href="https://127.0.0.1">https link</a> <a href="gemini://127.0.0.1">gemini link</a> tail</p>'
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), """\
head ↳ https link ↳ gemini link tail
=> https://127.0.0.1 https link
=> gemini://127.0.0.1 gemini link
""")

    def test_in_text_tag(self):
        html = "<p><b>bold</b> text</p>"
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), "bold text\n")

    def test_img_emu(self):
        html = "aa <img class=\"emu\" title=\":blobcatgooglyshrug:\" src=\"/d/6ytBYw515CvqFJZ8N2.png\"> bb</p>"
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), "aa :blobcatgooglyshrug: bb\n")

    def test_html2gmi_header(self):
        html = """\
            <h1>Header 1</h1>
            <p>Paragraph 1</p>
            <h2>Header 1.1</h2>
            <p>Paragraph 1.1</p>
        """
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), """\
# Header 1

Paragraph 1

## Header 1.1

Paragraph 1.1
""")

    def test_html2gmi_pre(self):
        html = """\
<pre>
def fib(n):
    if n == 1 or n == 2:
        return 1
    return fib(n - 1) + fib(n - 1)
</pre>
    """
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), """\
```

def fib(n):
    if n == 1 or n == 2:
        return 1
    return fib(n - 1) + fib(n - 1)

```
""")

    def test_html2gmi_description_list(self):
        html = """
            <dl>
                <dt>Coffee</dt>
                <dd>Black hot drink</dd>
                <dt>Milk</dt>
                <dd>White cold drink</dd>
            </dl>
        """
        self.assertEqual(HtmlToGmi("", fn_media_url).feed(html), """\
* Coffee

Black hot drink

* Milk

White cold drink
""")


if __name__ == '__main__':
    main()