tooncher/tests/test_cli.py

import pathlib
import subprocess
from unittest.mock import patch

import pytest
import yaml

# pylint: disable=protected-access
import tooncher._cli


def test_cli_help():
    proc_info = subprocess.run(
        ["tooncher", "--help"],
        check=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
    )
    assert b"optional arguments:" in proc_info.stdout
    assert not proc_info.stderr


@patch("tooncher._cli.run")
@patch("os.environ", {})
def test_engine_path_arg(run_mock):
    with patch("sys.argv", ["", "--engine-path", "/opt/ttr/TTREngine", "username"]):
        tooncher._cli.main()
    run_mock.assert_called_once()
    args, kwargs = run_mock.call_args
    assert not args
    assert kwargs["engine_path"] == "/opt/ttr/TTREngine"


@patch("tooncher._cli.run")
@patch("os.environ", {"TOONCHER_ENGINE_PATH": "/opt/ttr/TTREnvine"})
def test_engine_path_env(run_mock):
    with patch("sys.argv", ["", "username"]):
        tooncher._cli.main()
    run_mock.assert_called_once()
    args, kwargs = run_mock.call_args
    assert not args
    assert kwargs["engine_path"] == "/opt/ttr/TTREnvine"


@patch("tooncher._cli.run")
@patch("os.environ", {"TOONCHER_ENGINE_PATH": "/opt/ttr/TTREnvine"})
def test_engine_path_arg_env(run_mock):
    with patch("sys.argv", ["", "--engine-path", "/opt/ttr/TTREngine", "username"]):
        tooncher._cli.main()
    run_mock.assert_called_once()
    args, kwargs = run_mock.call_args
    assert not args
    assert kwargs["engine_path"] == "/opt/ttr/TTREngine"


@patch("tooncher.launch")
@patch("os.environ", {})
def test_engine_path_config(launch_mock, tmpdir):
    config_file = tmpdir.join("config")
    config_file.write(
        yaml.safe_dump(
            {
                "engine_path": "/opt/conf/TTR",
                "accounts": [{"username": "someone", "password": "secret"}],
            }
        )
    )
    with patch("sys.argv", ["", "--config", config_file.strpath, "someone"]):
        tooncher._cli.main()
    launch_mock.assert_called_once()
    args, kwargs = launch_mock.call_args
    assert not args
    assert kwargs["engine_path"] == pathlib.Path("/opt/conf/TTR")


@patch("tooncher.launch")
@patch("os.environ", {"TOONCHER_ENGINE_PATH": "/opt/ttr/TTREnvine"})
def test_engine_path_env_config(launch_mock, tmpdir):
    config_file = tmpdir.join("config")
    config_file.write(
        yaml.safe_dump(
            {
                "engine_path": "/opt/conf/TTR",
                "accounts": [{"username": "someone", "password": "secret"}],
            }
        )
    )
    with patch("sys.argv", ["", "--config", config_file.strpath, "someone"]):
        tooncher._cli.main()
    launch_mock.assert_called_once()
    args, kwargs = launch_mock.call_args
    assert not args
    assert kwargs["engine_path"] == pathlib.Path("/opt/ttr/TTREnvine")


@patch("tooncher.launch")
def test_account(launch_mock, tmpdir):
    config_file = tmpdir.join("config")
    config_file.write(
        yaml.safe_dump(
            {
                "engine_path": "/opt/conf/TTR",
                "accounts": [
                    {"username": "someone", "password": "secret"},
                    {"username": "toon", "password": "town"},
                ],
            }
        )
    )
    with patch("sys.argv", ["", "--config", config_file.strpath, "toon"]):
        tooncher._cli.main()
    launch_mock.assert_called_once_with(
        engine_path=pathlib.Path("/opt/conf/TTR"),
        username="toon",
        password="town",
        validate_ssl_certs=True,
        cpu_limit_percent=None,
    )


def test_account_duplicate_username(tmpdir):
    config_file = tmpdir.join("config")
    config_file.write(
        yaml.safe_dump(
            {
                "engine_path": "/opt/conf/TTR",
                "accounts": [
                    {"username": "someone", "password": "secret"},
                    {"username": "toon", "password": "town"},
                    {"username": "toon", "password": "town2"},
                ],
            }
        )
    )
    with patch("sys.argv", ["", "--config", config_file.strpath, "toon"]):
        with pytest.raises(ValueError, match=r"multiple .* username"):
            tooncher._cli.main()


def test_account_unknown_username(tmpdir):
    config_file = tmpdir.join("config")
    config_file.write(
        yaml.safe_dump(
            {
                "engine_path": "/opt/conf/TTR",
                "accounts": [
                    {"username": "someone", "password": "secret"},
                    {"username": "toon", "password": "town"},
                ],
            }
        )
    )
    with patch("sys.argv", ["", "--config", config_file.strpath, "player"]):
        with pytest.raises(ValueError, match=r"not found"):
            tooncher._cli.main()