import json
import os

import pytest

from dvc.render.image import ImageRenderer


@pytest.mark.parametrize(
    "extension, matches",
    (
        (".csv", False),
        (".json", False),
        (".tsv", False),
        (".yaml", False),
        (".jpg", True),
        (".gif", True),
        (".jpeg", True),
        (".png", True),
    ),
)
def test_matches(extension, matches):
    filename = "file" + extension
    data = {
        "HEAD": {"data": {filename: {}}},
        "v1": {"data": {filename: {}}},
    }
    assert ImageRenderer.matches(data) == matches


def test_render(tmp_dir):
    data = {"workspace": {"data": {"file.jpg": {"data": b"content"}}}}

    page_dir = os.path.join("some", "path")
    html = ImageRenderer(data).generate_html(page_dir)

    assert (tmp_dir / page_dir).is_dir()
    image_file = tmp_dir / page_dir / "static" / "workspace_file.jpg"
    assert image_file.is_file()

    with open(image_file, "rb") as fobj:
        assert fobj.read() == b"content"

    assert "<p>file.jpg</p>" in html
    assert (
        f'<img src="{os.path.join("static", "workspace_file.jpg")}">' in html
    )


def test_as_json(tmp_dir):
    data = {
        "workspace": {"data": {"file.jpg": {"data": b"content"}}},
        "HEAD": {"data": {"file.jpg": {"data": b"content2"}}},
    }

    page_dir = os.path.join("some", "path")
    json_content = json.loads(ImageRenderer(data).as_json(path=page_dir))

    img1 = os.path.join(page_dir, "workspace_file.jpg")
    img2 = os.path.join(page_dir, "HEAD_file.jpg")

    assert {
        "type": "image",
        "revisions": ["workspace"],
        "url": os.path.abspath(img1),
    } in json_content
    assert {
        "type": "image",
        "revisions": ["HEAD"],
        "url": os.path.abspath(img2),
    } in json_content

    def assert_file_with_content(path, content):
        assert os.path.isfile(path)
        with open(path, "rb") as fd:
            assert fd.read() == content

    assert_file_with_content(img1, b"content")
    assert_file_with_content(img2, b"content2")