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"


def test_version():
    """Test version is correct."""
    assert __version__ == "1.0.0"


def test_domain_ends_with_slash():
    """Test domain ends with a 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.


def test_do_things():
    """Test do_things() works."""
    # 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 = ""
    response = client.get("/")

    # Check if response is our HTML.
    with open("templates/index.html", encoding="utf8") as our_html:
        data = our_html.read()

        # index.html has a trailing newline that we need to remove.
        if data[-1:] == "\n":
            data_without_trailing_nl = data[:-1]

    assert response.status_code == 200
    assert response.text == data_without_trailing_nl


def test_upload_file():
    """Test if we can upload files."""
    domain = os.environ["SERVE_DOMAIN"]

    # Remove trailing slash from domain
    if domain.endswith("/"):
        domain = domain[:-1]

    # Upload our video file and check if it returns the html_url.
    with open(TEST_FILE, "rb") as uploaded_file:
        response = client.post(
            url="/uploadfiles/",
            files={"file": uploaded_file},
        )
        returned_json = response.json()
        html_url = returned_json["html_url"]

    assert response.status_code == 200
    assert html_url == f"{domain}/test.mp4"