34 lines
944 B
Python
34 lines
944 B
Python
import imghdr
|
|
import os
|
|
|
|
from discord_embed import settings
|
|
from discord_embed.video import Resolution, make_thumbnail, video_resolution
|
|
|
|
TEST_FILE = "tests/test.mp4"
|
|
|
|
|
|
def test_video_resolution():
|
|
"""Test video_resolution() works."""
|
|
assert video_resolution(TEST_FILE) == Resolution(height=422, width=422)
|
|
|
|
|
|
def test_make_thumbnail():
|
|
"""Test make_thumbnail() works."""
|
|
domain = os.environ["SERVE_DOMAIN"]
|
|
|
|
# Remove trailing slash from domain
|
|
if domain.endswith("/"):
|
|
domain = domain[:-1]
|
|
|
|
# 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")
|
|
|
|
# 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"
|