"""
Tests for specific functions in the utils package.
"""

from typing import TYPE_CHECKING

import pytest
from conftest import PLATFORM

pytestmark = pytest.mark.skipif(
    PLATFORM != "win", reason="This file contains tests that are Windows only."
)

if TYPE_CHECKING:
    from menuinst.utils import WinLex
else:
    WinLex = pytest.importorskip("menuinst.utils").WinLex


def test_quote_args_1():
    """Verify that input arguments are quoted."""
    wl = WinLex()
    test_str = [r"%SystemRoot%\system32\foo.exe", "/pt", "%1", "%2", "%3", "%4"]
    output = wl.quote_args(test_str)
    assert output == [r'"%SystemRoot%\system32\foo.exe"', "/pt", '"%1"', '"%2"', '"%3"', '"%4"']


def test_quote_args_2():
    """Verify correct quotes with blank spaces."""
    wl = WinLex()
    args = [
        r"C:\Windows\System32\notepad.exe",
        "/pt",
        r"C:\Users\Foo Bar\file.txt",
        "S p a c e s",
    ]
    output = wl.quote_args(args)
    assert output == [
        r"C:\Windows\System32\notepad.exe",
        "/pt",
        r'"C:\Users\Foo Bar\file.txt"',
        '"S p a c e s"',
    ]


def test_quote_args_3():
    """Verify special case with cmd.exe and /C."""
    args = ["cmd.exe", "/C", "echo", "Hello World"]
    wl = WinLex()
    output = wl.quote_args(args)
    assert output == ["cmd.exe", "/C", '"echo "Hello World""']


def test_quote_args_4():
    """Verify special case with cmd.exe and /K many words with spaces and percentage signs."""
    args = ["cmd.exe", "/K", "dir", "Hello World", "%1 %2 %foo% with spaces", "x", "y"]
    wl = WinLex()
    output = wl.quote_args(args)
    assert output == ["cmd.exe", "/K", '"dir "Hello World" "%1 %2 %foo% with spaces" x y"']


def test_quote_args_5():
    """Verify quotation works with mix of paths and spaces."""
    cmd = [r"C:\Program Files\App\app.exe", r"C:\Users\Me\My File.txt"]
    assert " ".join(WinLex.quote_args(cmd)) == (
        r'"C:\Program Files\App\app.exe" "C:\Users\Me\My File.txt"'
    )


def test_quote_args_6():
    """Verify command with metachars quotes as expected."""
    cmd = ["cmd", "/K", "echo", "%FOO%>", "out.txt"]
    line = " ".join(WinLex.quote_args(cmd))
    assert "cmd /K echo %FOO%> out.txt" == line


def test_quote_args_7():
    """Test with Powershell and spaced script path."""
    cmd = [
        r"C:\Windows\System32\WindowsPowerShell\v1.0\powershell.exe",
        "-WindowStyle",
        "hidden",
        r"C:\path with spaces\script.ps1",
    ]
    line = " ".join(WinLex.quote_args(cmd))
    assert r"C:\Windows\System32\WindowsPowerShell\v1.0\powershell.exe" in line
    assert r'"C:\path with spaces\script.ps1"' in line


def test_quote_args_8():
    """Verify special case with spaces, cmd.exe and /K and percentage signs."""
    args = [
        r"C:\path with spaces\cmd.exe",
        "/K",
        "dir",
        "Hello World",
        "%1 %2 %foo% with spaces",
        "x",
        "y",
    ]
    wl = WinLex()
    output = wl.quote_args(args)
    assert output == [
        r'"C:\path with spaces\cmd.exe"',
        "/K",
        '"dir "Hello World" "%1 %2 %foo% with spaces" x y"',
    ]


@pytest.mark.parametrize("special_flag", ("/C", "/K"))
def test_quote_args_special_flag(special_flag):
    """Verify special case with spaces, cmd.exe, multiple flags and percentage sign."""
    args = [
        "cmd.exe",
        "/A",
        "/B",
        special_flag,
        r"C:\Foo Bar\bin.exe",
        "%1",
    ]
    wl = WinLex()
    output = wl.quote_args(args)
    assert output == [
        "cmd.exe",
        "/A",
        "/B",
        special_flag,
        '""C:\\Foo Bar\\bin.exe" %1"',
    ]