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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
from types import SimpleNamespace
from unittest.mock import ANY, Mock
import pytest
from pdm.cli.commands import self_cmd
def mock_distribution(metadata, entry_points=()):
entry_points = (SimpleNamespace(group=ep) for ep in entry_points)
return SimpleNamespace(metadata=metadata, entry_points=entry_points)
DISTRIBUTIONS = [
mock_distribution(
{"Name": "foo", "Version": "1.0.0", "Summary": "Foo package"}, ["pdm.plugin"]
),
mock_distribution(
{"Name": "bar", "Version": "2.0.0", "Summary": "Bar package"}, ["pdm"]
),
mock_distribution({"Name": "baz", "Version": "3.0.0", "Summary": "Baz package"}),
]
@pytest.fixture()
def mock_pip(monkeypatch):
mocked = Mock()
monkeypatch.setattr(self_cmd, "run_pip", mocked)
return mocked
@pytest.fixture()
def mock_all_distributions(monkeypatch):
monkeypatch.setattr(
self_cmd, "_get_distributions", Mock(return_value=DISTRIBUTIONS)
)
@pytest.fixture()
def mock_latest_pdm_version(mocker):
return mocker.patch(
"pdm.cli.commands.self_cmd.get_latest_pdm_version_from_pypi",
)
@pytest.mark.usefixtures("mock_all_distributions")
def test_self_list(invoke):
result = invoke(["self", "list"])
assert result.exit_code == 0, result.stderr
packages = [line.split()[0] for line in result.stdout.splitlines()]
assert packages == ["bar", "baz", "foo"]
@pytest.mark.usefixtures("mock_all_distributions")
def test_self_list_plugins(invoke):
result = invoke(["self", "list", "--plugins"])
assert result.exit_code == 0, result.stderr
packages = [line.split()[0] for line in result.stdout.splitlines()]
assert packages == ["bar", "foo"]
def test_self_add(invoke, mock_pip):
result = invoke(["self", "add", "foo"])
assert result.exit_code == 0, result.stderr
mock_pip.assert_called_with(ANY, ["install", "foo"])
result = invoke(["self", "add", "--pip-args", "--force-reinstall --upgrade", "foo"])
assert result.exit_code == 0, result.stderr
mock_pip.assert_called_with(
ANY, ["install", "--force-reinstall", "--upgrade", "foo"]
)
def test_self_remove(invoke, mock_pip, monkeypatch):
def _mock_resolve(self, packages):
return ["demo", "pytz"] if "demo" in packages else packages
monkeypatch.setattr(
self_cmd.RemoveCommand,
"_resolve_dependencies_to_remove",
_mock_resolve,
)
result = invoke(["self", "remove", "foo"])
assert result.exit_code != 0
result = invoke(["self", "remove", "-y", "demo"])
assert result.exit_code == 0, result.stderr
mock_pip.assert_called_with(ANY, ["uninstall", "-y", "demo", "pytz"])
@pytest.mark.parametrize(
"args,expected",
[
(["self", "update"], ["install", "--upgrade", "pdm==99.0.0"]),
(["self", "update", "--pre"], ["install", "--upgrade", "pdm==99.0.1b1"]),
(
["self", "update", "--head"],
["install", "--upgrade", f"pdm @ git+{self_cmd.PDM_REPO}@main"],
),
],
)
def test_self_update(invoke, mock_pip, mock_latest_pdm_version, args, expected):
def mocked_latest_version(project, pre):
return "99.0.1b1" if pre else "99.0.0"
mock_latest_pdm_version.side_effect = mocked_latest_version
result = invoke(args)
assert result.exit_code == 0, result.stderr
mock_pip.assert_called_with(ANY, expected)
def test_self_update_already_latest(invoke, mock_pip, mock_latest_pdm_version):
mock_latest_pdm_version.return_value = "0.0.0"
result = invoke(["self", "update"])
assert result.exit_code == 0, result.stderr
assert "Already up-to-date" in result.stdout
mock_pip.assert_not_called()
|
