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()
"""
"""
for line, html in enumerate(generated_html_lines):
# Strip spaces and newlines
html = html.strip()
# Check each line
if line == 1:
assert html == ""
elif line == 2:
assert html == ""
elif line == 3:
assert html.startswith("