mirror of https://github.com/pypa/pip
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
from optparse import Values
|
|
from typing import FrozenSet, List, Set
|
|
|
|
import pytest
|
|
|
|
from pip._internal.cli import cmdoptions
|
|
from pip._internal.cli.base_command import Command
|
|
from pip._internal.cli.status_codes import SUCCESS
|
|
from pip._internal.models.format_control import FormatControl
|
|
|
|
|
|
class SimpleCommand(Command):
|
|
def __init__(self) -> None:
|
|
super().__init__("fake", "fake summary")
|
|
|
|
def add_options(self) -> None:
|
|
self.cmd_opts.add_option(cmdoptions.no_binary())
|
|
self.cmd_opts.add_option(cmdoptions.only_binary())
|
|
|
|
def run(self, options: Values, args: List[str]) -> int:
|
|
self.options = options
|
|
return SUCCESS
|
|
|
|
|
|
def test_no_binary_overrides() -> None:
|
|
cmd = SimpleCommand()
|
|
cmd.main(["fake", "--only-binary=:all:", "--no-binary=fred"])
|
|
format_control = FormatControl({"fred"}, {":all:"})
|
|
assert cmd.options.format_control == format_control
|
|
|
|
|
|
def test_only_binary_overrides() -> None:
|
|
cmd = SimpleCommand()
|
|
cmd.main(["fake", "--no-binary=:all:", "--only-binary=fred"])
|
|
format_control = FormatControl({":all:"}, {"fred"})
|
|
assert cmd.options.format_control == format_control
|
|
|
|
|
|
def test_none_resets() -> None:
|
|
cmd = SimpleCommand()
|
|
cmd.main(["fake", "--no-binary=:all:", "--no-binary=:none:"])
|
|
format_control = FormatControl(set(), set())
|
|
assert cmd.options.format_control == format_control
|
|
|
|
|
|
def test_none_preserves_other_side() -> None:
|
|
cmd = SimpleCommand()
|
|
cmd.main(["fake", "--no-binary=:all:", "--only-binary=fred", "--no-binary=:none:"])
|
|
format_control = FormatControl(set(), {"fred"})
|
|
assert cmd.options.format_control == format_control
|
|
|
|
|
|
def test_comma_separated_values() -> None:
|
|
cmd = SimpleCommand()
|
|
cmd.main(["fake", "--no-binary=1,2,3"])
|
|
format_control = FormatControl({"1", "2", "3"}, set())
|
|
assert cmd.options.format_control == format_control
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"no_binary,only_binary,argument,expected",
|
|
[
|
|
({"fred"}, set(), "fred", frozenset(["source"])),
|
|
({"fred"}, {":all:"}, "fred", frozenset(["source"])),
|
|
(set(), {"fred"}, "fred", frozenset(["binary"])),
|
|
({":all:"}, {"fred"}, "fred", frozenset(["binary"])),
|
|
],
|
|
)
|
|
def test_fmt_ctl_matches(
|
|
no_binary: Set[str], only_binary: Set[str], argument: str, expected: FrozenSet[str]
|
|
) -> None:
|
|
fmt = FormatControl(no_binary, only_binary)
|
|
assert fmt.get_allowed_formats(argument) == expected
|