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.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()
|