# import pytest # from ipm.__main__ import main # from unittest.mock import patch, MagicMock # # Test IDs for parametrization # HAPPY_PATH_INSTALL = "happy_install" # HAPPY_PATH_EXTRACT = "happy_extract" # HAPPY_PATH_BUILD = "happy_build" # EDGE_CASE_NO_ARGS = "edge_no_args" # ERROR_CASE_UNKNOWN_COMMAND = "error_unknown_command" # # Mock the sys.argv to simulate command line arguments # @pytest.fixture # def mock_sys_argv(monkeypatch): # def _mock_sys_argv(args): # monkeypatch.setattr("sys.argv", ["ipm"] + args) # return _mock_sys_argv # # Mock the api functions to prevent actual execution # @pytest.fixture # def mock_api_functions(monkeypatch): # install_mock = MagicMock() # extract_mock = MagicMock() # build_mock = MagicMock() # monkeypatch.setattr("ipm.__main__.install", install_mock) # monkeypatch.setattr("ipm.__main__.extract", extract_mock) # monkeypatch.setattr("ipm.__main__.build", build_mock) # return install_mock, extract_mock, build_mock # @pytest.mark.parametrize("test_id, args, expected_call", [ # # Happy path tests # (HAPPY_PATH_INSTALL, ["install", "http://ipm.hydroroll.team/package.ipk"], ("install", ["http://ipm.hydroroll.team/package.ipk", None])), # (HAPPY_PATH_EXTRACT, ["extract", "package.ipk", "--dist", "dist_folder"], ("extract", ["package.ipk", "dist_folder"])), # (HAPPY_PATH_BUILD, ["build", "source_folder"], ("build", ["source_folder"])), # # Edge case tests # (EDGE_CASE_NO_ARGS, [], ("help", [])), # # Error case tests # (ERROR_CASE_UNKNOWN_COMMAND, ["unknown", "arg"], ("error", ["unknown"])), # ]) # def test_main_commands(test_id, args, expected_call, mock_sys_argv, mock_api_functions, capsys): # mock_sys_argv(args) # install_mock, extract_mock, build_mock = mock_api_functions # # Act # with pytest.raises(SystemExit): # argparse exits the program when -h is called or on error # main() # # Assert # if expected_call[0] == "install": # install_mock.assert_called_once_with(*expected_call[1], echo=True) # elif expected_call[0] == "extract": # extract_mock.assert_called_once_with(*expected_call[1]) # elif expected_call[0] == "build": # build_mock.assert_called_once_with(*expected_call[1]) # elif expected_call[0] == "help": # captured = capsys.readouterr() # assert "Infini 包管理器" in captured.out # elif expected_call[0] == "error": # captured = capsys.readouterr() # assert "error: unrecognized arguments" in captured.err # pytest.main()