1
1
Fork 0
mirror of https://github.com/pypa/pip synced 2023-12-13 21:30:23 +01:00
pip/tests/unit/test_format_control.py

73 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