1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
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.ipm"], ("install", ["http://ipm.hydroroll.team/package.ipm", None])),
(HAPPY_PATH_EXTRACT, ["extract", "package.ipm", "--dist", "dist_folder"], ("extract", ["package.ipm", "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()
|