diff options
Diffstat (limited to 'tests/test_cli_main.py')
-rw-r--r-- | tests/test_cli_main.py | 75 |
1 files changed, 29 insertions, 46 deletions
diff --git a/tests/test_cli_main.py b/tests/test_cli_main.py index 9db5341..673031c 100644 --- a/tests/test_cli_main.py +++ b/tests/test_cli_main.py @@ -5,6 +5,7 @@ from __future__ import annotations import argparse from functools import wraps +from pathlib import Path from typing import Any from typing import Callable from unittest.mock import ANY @@ -18,6 +19,8 @@ from mlia.backend.errors import BackendUnavailableError from mlia.cli.main import backend_main from mlia.cli.main import CommandInfo from mlia.cli.main import main +from mlia.cli.options import add_output_directory +from mlia.core.context import ExecutionContext from mlia.core.errors import InternalError from tests.utils.logging import clear_loggers @@ -44,59 +47,16 @@ def test_option_version(capfd: pytest.CaptureFixture) -> None: assert stderr == "" -@pytest.mark.parametrize( - "is_default, expected_command_help", - [(True, "Test command [default]"), (False, "Test command")], -) -def test_command_info(is_default: bool, expected_command_help: str) -> None: +def test_command_info() -> None: """Test properties of CommandInfo object.""" def test_command() -> None: """Test command.""" - command_info = CommandInfo(test_command, ["test"], [], is_default) + command_info = CommandInfo(test_command, ["test"], []) assert command_info.command_name == "test_command" assert command_info.command_name_and_aliases == ["test_command", "test"] - assert command_info.command_help == expected_command_help - - -def test_default_command(monkeypatch: pytest.MonkeyPatch) -> None: - """Test adding default command.""" - - def mock_command(func_mock: MagicMock, name: str) -> Callable[..., None]: - """Mock cli command.""" - - def sample_cmd_1(*args: Any, **kwargs: Any) -> None: - """Sample command.""" - func_mock(*args, **kwargs) - - ret_func = sample_cmd_1 - ret_func.__name__ = name - - return ret_func - - non_default_command = MagicMock() - - def non_default_command_params(parser: argparse.ArgumentParser) -> None: - """Add parameters for non default command.""" - parser.add_argument("--param") - - monkeypatch.setattr( - "mlia.cli.main.get_commands", - MagicMock( - return_value=[ - CommandInfo( - func=mock_command(non_default_command, "non_default_command"), - aliases=["command2"], - opt_groups=[non_default_command_params], - is_default=False, - ), - ] - ), - ) - - main(["command2", "--param", "test"]) - non_default_command.assert_called_once_with(param="test") + assert command_info.command_help == "Test command" def wrap_mock_command(mock: MagicMock, command: Callable) -> Callable: @@ -293,6 +253,29 @@ def test_commands_execution( mock.assert_called_once_with(*expected_call.args, **expected_call.kwargs) +def test_passing_output_directory_parameter( + monkeypatch: pytest.MonkeyPatch, tmp_path: Path +) -> None: + """Test passing parameter --output-dir.""" + passed_context: ExecutionContext | None = None + + def sample_command(ctx: ExecutionContext) -> None: + """Sample command.""" + nonlocal passed_context + passed_context = ctx + + monkeypatch.setattr( + "mlia.cli.main.get_commands", + lambda: [CommandInfo(sample_command, [], [add_output_directory])], + ) + + output_dir = tmp_path / "output" + main(["sample_command", "--output-dir", output_dir.as_posix()]) + + assert passed_context is not None + assert passed_context.output_dir == output_dir + + @pytest.mark.parametrize( "params, expected_call", [ |