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

Add type hints

Browse Source
This commit is contained in:
Joakim Hellsén
2023-01-09 04:25:20 +01:00
parent fc424f7263
commit ab477c0ab4
10 changed files with 69 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tests/test_discord_embed.py
View File

@ -1,67 +1,68 @@
import os
from fastapi import Response
from fastapi.testclient import TestClient
from discord_embed import __version__, settings
from discord_embed.main import app
client = TestClient(app)
TEST_FILE = "tests/test.mp4"
client: TestClient = TestClient(app)
TEST_FILE: str = "tests/test.mp4"
def test_version():
def test_version() -> None:
"""Test version is correct."""
assert __version__ == "1.0.0"
def test_domain_ends_with_slash():
def test_domain_ends_with_slash() -> None:
"""Test domain ends with a slash."""
assert not settings.serve_domain.endswith("/")
def test_save_to_disk():
def test_save_to_disk() -> None:
"""Test save_to_disk() works."""
# TODO: Implement this test. I need to mock the UploadFile object.
def test_do_things():
def test_do_things() -> None:
"""Test do_things() works."""
# TODO: Implement this test. I need to mock the UploadFile object.
def test_main():
def test_main() -> None:
"""Test main() works."""
data_without_trailing_nl = ""
response = client.get("/")
response: Response = client.get("/")
# Check if response is our HTML.
with open("templates/index.html", encoding="utf8") as our_html:
data = our_html.read()
data: str = our_html.read()
# index.html has a trailing newline that we need to remove.
if data[-1:] == "\n":
data_without_trailing_nl = data[:-1]
data_without_trailing_nl: str = data[:-1] # type: ignore
assert response.status_code == 200
assert response.text == data_without_trailing_nl
def test_upload_file():
def test_upload_file() -> None:
"""Test if we can upload files."""
domain = os.environ["SERVE_DOMAIN"]
# Remove trailing slash from domain
if domain.endswith("/"):
domain = domain[:-1]
domain: str = domain[:-1] # type: ignore
# Upload our video file and check if it returns the html_url.
with open(TEST_FILE, "rb") as uploaded_file:
response = client.post(
response: Response = client.post(
url="/uploadfiles/",
files={"file": uploaded_file},
)
returned_json = response.json()
html_url = returned_json["html_url"]
html_url: str = returned_json["html_url"]
assert response.status_code == 200
assert html_url == f"{domain}/test.mp4"

43
tests/test_generate_html.py
View File

@ -3,19 +3,19 @@ import os
from discord_embed.generate_html import generate_html_for_videos
def test_generate_html_for_videos():
def test_generate_html_for_videos() -> None:
"""Test generate_html_for_videos() works."""
domain = os.environ["SERVE_DOMAIN"]
domain: str = 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")
if os.path.exists("Uploads/test_video.mp4.html"):
os.remove("Uploads/test_video.mp4.html")
generated_html = generate_html_for_videos(
generated_html: str = generate_html_for_videos(
url="https://www.youtube.com/watch?v=dQw4w9WgXcQ",
width=1920,
height=1080,
@ -27,7 +27,7 @@ def test_generate_html_for_videos():
# 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()
generated_html_lines: list[str] = generated_html_file.readlines()
"""
<!DOCTYPE html>
<html>
@ -46,32 +46,35 @@ def test_generate_html_for_videos():
for line, html in enumerate(generated_html_lines):
# Strip spaces and newlines
html = html.strip()
stripped_html: str = html.strip()
rick: str = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
# Check each line
if line == 1:
assert html == "<!DOCTYPE html>"
assert stripped_html == "<!DOCTYPE html>"
elif line == 2:
assert html == "<html>"
assert stripped_html == "<html>"
elif line == 3:
assert html.startswith("<!-- Generated at ")
assert stripped_html.startswith("<!-- Generated at ")
elif line == 4:
assert html == "<head>"
assert stripped_html == "<head>"
elif line == 5:
assert html == '<meta property="og:type" content="video.other">'
assert stripped_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">'
assert stripped_html == f'<meta property="twitter:player" content="{rick}">'
elif line == 7:
assert html == '<meta property="og:video:type" content="text/html">'
assert stripped_html == '<meta property="og:video:type" content="text/html">'
elif line == 8:
assert html == '<meta property="og:video:width" content="1920">'
assert stripped_html == '<meta property="og:video:width" content="1920">'
elif line == 9:
assert html == '<meta property="og:video:height" content="1080">'
assert stripped_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">'
thumb: str = "https://i.ytimg.com/vi/dQw4w9WgXcQ/hqdefault.jpg"
assert stripped_html == f'<meta name="twitter:image" content="{thumb}">'
elif line == 11:
assert html == '<meta http-equiv="refresh" content="0;url=https://www.youtube.com/watch?v=dQw4w9WgXcQ">'
assert stripped_html == f'<meta http-equiv="refresh" content="0;url={rick}">'
elif line == 12:
assert html == "</head>"
assert stripped_html == "</head>"
elif line == 13:
assert html == "</html>"
assert stripped_html == "</html>"

10
tests/test_video.py
View File

@ -7,24 +7,24 @@ from discord_embed.video import Resolution, make_thumbnail, video_resolution
TEST_FILE = "tests/test.mp4"
def test_video_resolution():
def test_video_resolution() -> None:
"""Test video_resolution() works."""
assert video_resolution(TEST_FILE) == Resolution(height=422, width=422)
def test_make_thumbnail():
def test_make_thumbnail() -> None:
"""Test make_thumbnail() works."""
domain = os.environ["SERVE_DOMAIN"]
domain: str = os.environ["SERVE_DOMAIN"]
# Remove trailing slash from domain
if domain.endswith("/"):
domain = domain[:-1]
domain: str = domain[:-1] # type: ignore
# Remove thumbnail if it exists
if os.path.exists(f"{settings.upload_folder}/test.mp4.jpg"):
os.remove(f"{settings.upload_folder}/test.mp4.jpg")
thumbnail = make_thumbnail(TEST_FILE, "test.mp4")
thumbnail: str = make_thumbnail(TEST_FILE, "test.mp4")
# Check if thumbnail is a jpeg.
assert imghdr.what(f"{settings.upload_folder}/test.mp4.jpg") == "jpeg"

2
tests/test_webhook.py
View File

@ -1,6 +1,6 @@
from discord_embed.webhook import send_webhook
def test_send_webhook():
def test_send_webhook() -> None:
"""Test send_webhook() works."""
send_webhook("Running Pytest")