37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
import imghdr
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from discord_embed import settings
|
|
from discord_embed.video import Resolution, make_thumbnail, video_resolution
|
|
|
|
TEST_FILE = "tests/test.mp4"
|
|
|
|
|
|
def test_video_resolution() -> None:
|
|
"""Test video_resolution() works."""
|
|
assert video_resolution(TEST_FILE) == Resolution(height=422, width=422)
|
|
|
|
|
|
def test_make_thumbnail() -> None:
|
|
"""Test make_thumbnail() works."""
|
|
domain: str = os.environ["SERVE_DOMAIN"]
|
|
|
|
# Remove trailing slash from domain
|
|
if domain.endswith("/"):
|
|
domain: str = domain[:-1]
|
|
|
|
# Remove thumbnail if it exists
|
|
if Path.exists(Path(f"{settings.upload_folder}/test.mp4.jpg")):
|
|
Path.unlink(Path(f"{settings.upload_folder}/test.mp4.jpg"))
|
|
|
|
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"
|
|
|
|
# Check if it returns the correct URL.
|
|
assert thumbnail == f"{domain}/test.mp4.jpg"
|