Update HTML to markdown converter

This commit is contained in:
2023-01-28 23:01:17 +01:00
parent 596238420a
commit 4d8a0e9ded
2 changed files with 33 additions and 42 deletions

View File

@ -1,7 +1,7 @@
from discord_rss_bot.markdown import convert_html_to_md
def test_convert_to_md():
def test_convert_to_md() -> None:
# Test bold
assert convert_html_to_md("<b>bold</b>") == "**bold**"
@ -28,22 +28,16 @@ def test_convert_to_md():
# Test multiple tags
assert (
convert_html_to_md(
'<b>bold</b> <i>italic</i> <a href="https://example.com">link</a> <code>code</code> <s>strikethrough</s>'
)
convert_html_to_md('<b>bold</b> <i>italic</i> <a href="https://example.com">link</a> <code>code</code> <s>strikethrough</s>') # noqa: E501
== "**bold** *italic* [link](https://example.com) `code` ~~strikethrough~~"
)
# Test removing all other tags
assert convert_html_to_md("<p>paragraph</p>") == "paragraph"
assert convert_html_to_md("<p>paragraph</p><p>paragraph</p>") == "paragraphparagraph"
assert convert_html_to_md("<p>paragraph</p><p>paragraph</p>") == "paragraph\nparagraph"
# Test <br> tags
assert (
convert_html_to_md("<p>paragraph<br>paragraph</p>")
== """paragraph
paragraph"""
)
assert convert_html_to_md("<p>paragraph<br>paragraph</p>") == "paragraph\nparagraph"
# Test removing trailing newline
assert convert_html_to_md("paragraph ") == "paragraph"
@ -52,11 +46,22 @@ paragraph"""
assert convert_html_to_md(" paragraph ") == "paragraph"
# Test removing leading and trailing whitespace and trailing newline
assert (
convert_html_to_md(
""" paragraph
""" # noqa: W293
)
== "paragraph"
assert convert_html_to_md(" paragraph\n \n") == "paragraph"
# Test real entry
nvidia_entry: str = (
'<p><a href="https://www.nvidia.com/en-us/geforce/news/jan-2023-nvidia-broadcast-update/">'
"NVIDIA Broadcast 1.4 Adds Eye Contact and Vignette Effects With Virtual Background Enhancements</a></p>"
'<div class="field field-name-field-short-description field-type-text-long field-label-hidden">'
'<div class="field-items"><div class="field-item even">Plus new options to mirror your camera and take a selfie.</div>' # noqa: E501
'</div></div><div class="field field-name-field-thumbnail-image field-type-image field-label-hidden">'
'<div class="field-items"><div class="field-item even"><a href="https://www.nvidia.com/en-us/geforce/news/jan-2023-nvidia-broadcast-update/">' # noqa: E501
'<img width="210" src="https://www.nvidia.com/content/dam/en-zz/Solutions/geforce/news/jan-2023-nvidia-broadcast-update/broadcast-owned-asset-625x330-newsfeed.png"' # noqa: E501
' title="NVIDIA Broadcast 1.4 Adds Eye Contact and Vignette Effects With Virtual Background Enhancements" '
'alt="NVIDIA Broadcast 1.4 Adds Eye Contact and Vignette Effects With Virtual Background Enhancements"></a></div></div></div>' # noqa: E501
)
assert (
convert_html_to_md(nvidia_entry)
== "[NVIDIA Broadcast 1.4 Adds Eye Contact and Vignette Effects With Virtual Background Enhancements](https://www.nvidia.com/en-us/geforce/news/jan-2023-nvidia-broadcast-update/)\n" # noqa: E501
"Plus new options to mirror your camera and take a selfie.[https://www.nvidia.com/en-us/geforce/news/jan-2023-nvidia-broadcast-update/](https://www.nvidia.com/en-us/geforce/news/jan-2023-nvidia-broadcast-update/)" # noqa: E501
)