Allow only specific packages to be skipped during startup dependency installation (#82758)

This commit is contained in:
puddly
2022-11-30 02:38:52 -05:00
committed by GitHub
parent fcf60a3b53
commit 8c8994352d
9 changed files with 78 additions and 2 deletions

View File

@@ -61,3 +61,27 @@ def test_validate_python(mock_exit):
assert mock_exit.called is False
mock_exit.reset_mock()
@patch("sys.exit")
def test_skip_pip_mutually_exclusive(mock_exit):
"""Test --skip-pip and --skip-pip-package are mutually exclusive."""
def parse_args(*args):
with patch("sys.argv", ["python"] + list(args)):
return main.get_arguments()
args = parse_args("--skip-pip")
assert args.skip_pip is True
args = parse_args("--skip-pip-packages", "foo")
assert args.skip_pip is False
assert args.skip_pip_packages == ["foo"]
args = parse_args("--skip-pip-packages", "foo-asd,bar-xyz")
assert args.skip_pip is False
assert args.skip_pip_packages == ["foo-asd", "bar-xyz"]
assert mock_exit.called is False
args = parse_args("--skip-pip", "--skip-pip-packages", "foo")
assert mock_exit.called is True