"""
Test command line commands.
"""
from pathlib import Path
from subprocess import PIPE, Popen

__author__ = "Sergey Vartanov"
__email__ = "me@enzet.ru"

from xml.etree import ElementTree
from xml.etree.ElementTree import Element

from map_machine.ui.cli import COMMAND_LINES

LOG: bytes = (
    b"INFO Constructing ways...\n"
    b"INFO Constructing nodes...\n"
    b"INFO Drawing ways...\n"
    b"INFO Drawing main icons...\n"
    b"INFO Drawing extra icons...\n"
    b"INFO Drawing texts...\n"
)


def error_run(arguments: list[str], message: bytes) -> None:
    """Run command that should fail and check error message."""
    p = Popen(["map-machine"] + arguments, stderr=PIPE)
    _, error = p.communicate()
    assert p.returncode != 0
    assert error == message


def run(arguments: list[str], message: bytes) -> None:
    """Run command that should fail and check error message."""
    p = Popen(["map-machine"] + arguments, stderr=PIPE)
    _, error = p.communicate()
    assert p.returncode == 0
    assert error == message


def test_wrong_render_arguments() -> None:
    """Test `render` command with wrong arguments."""
    error_run(
        ["render", "-z", "17"],
        b"CRITICAL Specify either --input, or --boundary-box, or --coordinates "
        b"and --size.\n",
    )


def test_render() -> None:
    """Test `render` command."""
    run(
        COMMAND_LINES["render"] + ["--cache", "tests/data"],
        LOG + b"INFO Writing output SVG to out/map.svg...\n",
    )
    with Path("out/map.svg").open() as output_file:
        root: Element = ElementTree.parse(output_file).getroot()

    # 4 expected elements: `defs`, `rect` (background), `g` (outline),
    # `g` (icon).
    assert len(root) == 4
    assert len(root[3][0]) == 0
    assert root.get("width") == "186.0"
    assert root.get("height") == "198.0"


def test_render_with_tooltips() -> None:
    """Test `render` command."""
    run(
        COMMAND_LINES["render_with_tooltips"] + ["--cache", "tests/data"],
        LOG + b"INFO Writing output SVG to out/map.svg...\n",
    )
    with Path("out/map.svg").open() as output_file:
        root: Element = ElementTree.parse(output_file).getroot()

    # 4 expected elements: `defs`, `rect` (background), `g` (outline),
    # `g` (icon).
    assert len(root) == 4
    assert len(root[3][0]) == 1
    assert root[3][0][0].text == "natural: tree"
    assert root.get("width") == "186.0"
    assert root.get("height") == "198.0"


def test_icons() -> None:
    """Test `icons` command."""
    run(
        COMMAND_LINES["icons"],
        b"INFO Icon grid is written to out/icon_grid.svg.\n"
        b"INFO Icons are written to out/icons_by_name and out/icons_by_id.\n",
    )

    assert (Path("out") / "icon_grid.svg").is_file()
    assert (Path("out") / "icons_by_name").is_dir()
    assert (Path("out") / "icons_by_id").is_dir()
    assert (Path("out") / "icons_by_name" / "Röntgen apple.svg").is_file()
    assert (Path("out") / "icons_by_id" / "apple.svg").is_file()


def test_mapcss() -> None:
    """Test `mapcss` command."""
    run(
        COMMAND_LINES["mapcss"],
        b"INFO MapCSS 0.2 scheme is written to out/map_machine_mapcss.\n",
    )

    assert (Path("out") / "map_machine_mapcss").is_dir()
    assert (Path("out") / "map_machine_mapcss" / "icons").is_dir()
    assert (
        Path("out") / "map_machine_mapcss" / "icons" / "apple.svg"
    ).is_file()
    assert (Path("out") / "map_machine_mapcss" / "map_machine.mapcss").is_file()


def test_element() -> None:
    """Test `element` command."""
    run(
        COMMAND_LINES["element"],
        b"INFO Element is written to out/element.svg.\n",
    )
    assert (Path("out") / "element.svg").is_file()


def test_tile() -> None:
    """Test `tile` command."""
    run(
        COMMAND_LINES["tile"] + ["--cache", "tests/data"],
        LOG + b"INFO Tile is drawn to out/tiles/tile_18_160199_88904.svg.\n"
        b"INFO SVG file is rasterized to out/tiles/tile_18_160199_88904.png.\n",
    )

    assert (Path("out") / "tiles" / "tile_18_160199_88904.svg").is_file()
    assert (Path("out") / "tiles" / "tile_18_160199_88904.png").is_file()