TheLovinator/discord-embed
1
Fork 0
Code Issues Pull Requests Packages Releases Activity

Add more tests

Browse Source
This commit is contained in:
Joakim Hellsén
2022-08-08 08:45:42 +02:00
parent 72cfd729a7
commit 74f0e8d22c
4 changed files with 86 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/test_discord_embed.py
View File

@@ -3,9 +3,7 @@ import os
from fastapi.testclient import TestClient
from discord_embed import __version__, settings
from discord_embed.generate_html import generate_html_for_videos
from discord_embed.main import app
from discord_embed.webhook import send_webhook
client = TestClient(app)
TEST_FILE = "tests/test.mp4"
@@ -21,24 +19,6 @@ def test_domain_ends_with_slash():
assert not settings.serve_domain.endswith("/")
def test_generate_html_for_videos():
"""Test generate_html_for_videos() works."""
domain = os.environ["SERVE_DOMAIN"]
# Remove trailing slash from domain
if domain.endswith("/"):
domain = domain[:-1]
generated_html = generate_html_for_videos(
url="https://www.youtube.com/watch?v=dQw4w9WgXcQ",
width=1920,
height=1080,
screenshot="https://i.ytimg.com/vi/dQw4w9WgXcQ/hqdefault.jpg",
filename="test_video.mp4",
)
assert generated_html == f"{domain}/test_video.mp4"
def test_save_to_disk():
"""Test save_to_disk() works."""
# TODO: Implement this test. I need to mock the UploadFile object.
@@ -49,11 +29,6 @@ def test_do_things():
# TODO: Implement this test. I need to mock the UploadFile object.
def test_send_webhook():
"""Test send_webhook() works."""
send_webhook("Running Pytest")
def test_main():
"""Test main() works."""
data_without_trailing_nl = ""

77
tests/test_generate_html.py Normal file
View File

@@ -0,0 +1,77 @@
import os
from discord_embed.generate_html import generate_html_for_videos
def test_generate_html_for_videos():
"""Test generate_html_for_videos() works."""
domain = os.environ["SERVE_DOMAIN"]
# Remove trailing slash from domain
if domain.endswith("/"):
domain = domain[:-1]
# Delete the old HTML file if it exists
if os.path.exists(f"Uploads/test_video.mp4.html"):
os.remove(f"Uploads/test_video.mp4.html")
generated_html = generate_html_for_videos(
url="https://www.youtube.com/watch?v=dQw4w9WgXcQ",
width=1920,
height=1080,
screenshot="https://i.ytimg.com/vi/dQw4w9WgXcQ/hqdefault.jpg",
filename="test_video.mp4",
)
assert generated_html == f"{domain}/test_video.mp4"
# Open the generated HTML and check if it contains the correct URL, width, height, and screenshot.
with open("Uploads/test_video.mp4.html", "r") as generated_html_file:
generated_html_lines = generated_html_file.readlines()
"""
<!DOCTYPE html>
<html>
<!-- Generated at 2022-08-08 08:16:53 -->
<head>
<meta property="og:type" content="video.other">
<meta property="twitter:player" content="https://www.youtube.com/watch?v=dQw4w9WgXcQ">
<meta property="og:video:type" content="text/html">
<meta property="og:video:width" content="1920">
<meta property="og:video:height" content="1080">
<meta name="twitter:image" content="https://i.ytimg.com/vi/dQw4w9WgXcQ/hqdefault.jpg">
<meta http-equiv="refresh" content="0;url=https://www.youtube.com/watch?v=dQw4w9WgXcQ">
</head>
</html>
"""
for line, html in enumerate(generated_html_lines):
# Strip spaces and newlines
html = html.strip()
# Check each line
if line == 1:
assert html == "<!DOCTYPE html>"
elif line == 2:
assert html == "<html>"
elif line == 3:
assert html.startswith("<!-- Generated at ")
elif line == 4:
assert html == '<head>'
elif line == 5:
assert html == '<meta property="og:type" content="video.other">'
elif line == 6:
assert html == '<meta property="twitter:player" content="https://www.youtube.com/watch?v=dQw4w9WgXcQ">'
elif line == 7:
assert html == '<meta property="og:video:type" content="text/html">'
elif line == 8:
assert html == '<meta property="og:video:width" content="1920">'
elif line == 9:
assert html == '<meta property="og:video:height" content="1080">'
elif line == 10:
assert html == '<meta name="twitter:image" content="https://i.ytimg.com/vi/dQw4w9WgXcQ/hqdefault.jpg">'
elif line == 11:
assert html == '<meta http-equiv="refresh" content="0;url=https://www.youtube.com/watch?v=dQw4w9WgXcQ">'
elif line == 12:
assert html == "</head>"
elif line == 13:
assert html == "</html>"

4
tests/test_video.py
View File

@@ -1,8 +1,8 @@
import imghdr
import os
import settings
from video import make_thumbnail, video_resolution
from discord_embed import settings
from discord_embed.video import make_thumbnail, video_resolution
TEST_FILE = "tests/test.mp4"

7
tests/test_webhook.py Normal file
View File

@@ -0,0 +1,7 @@
"""Test webhook.py, the only thing we have right now is send_webhook()."""
from discord_embed.webhook import send_webhook
def test_send_webhook():
"""Test send_webhook() works."""
send_webhook("Running Pytest")