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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
import sys
from unittest import mock
import pytest
from pdm.cli.commands.base import BaseCommand
from pdm.compat import importlib_metadata
from pdm.project.config import ConfigItem
from pdm.utils import cd
class HelloCommand(BaseCommand):
def add_arguments(self, parser) -> None:
parser.add_argument("-n", "--name", help="The person's name")
def handle(self, project, options) -> None:
greeting = "Hello world"
if options.name:
greeting = f"Hello, {options.name}"
print(greeting)
def new_command(core):
core.register_command(HelloCommand, "hello")
def replace_command(core):
core.register_command(HelloCommand, "info")
def add_new_config(core):
core.add_config("foo", ConfigItem("Test config", "bar"))
def make_entry_point(plugin):
ret = mock.Mock()
ret.load.return_value = plugin
return ret
def test_plugin_new_command(pdm, mocker, project, core):
mocker.patch.object(
importlib_metadata,
"entry_points",
return_value=[make_entry_point(new_command)],
)
core.init_parser()
core.load_plugins()
result = pdm(["--help"], obj=project)
assert "hello" in result.output
result = pdm(["hello"], obj=project)
assert result.output.strip() == "Hello world"
result = pdm(["hello", "-n", "Frost"], obj=project)
assert result.output.strip() == "Hello, Frost"
def test_plugin_replace_command(pdm, mocker, project, core):
mocker.patch.object(
importlib_metadata,
"entry_points",
return_value=[make_entry_point(replace_command)],
)
core.init_parser()
core.load_plugins()
result = pdm(["info"], obj=project)
assert result.output.strip() == "Hello world"
result = pdm(["info", "-n", "Frost"], obj=project)
assert result.output.strip() == "Hello, Frost"
def test_load_multiple_plugins(pdm, mocker, core):
mocker.patch.object(
importlib_metadata,
"entry_points",
return_value=[make_entry_point(new_command), make_entry_point(add_new_config)],
)
core.init_parser()
core.load_plugins()
result = pdm(["hello"])
assert result.output.strip() == "Hello world", result.outputs
result = pdm(["config", "foo"])
assert result.output.strip() == "bar"
def test_old_entry_point_compatibility(pdm, mocker, core):
def get_entry_points(group):
if group == "pdm":
return [make_entry_point(new_command)]
if group == "pdm.plugin":
return [make_entry_point(add_new_config)]
return []
mocker.patch.object(importlib_metadata, "entry_points", side_effect=get_entry_points)
core.init_parser()
core.load_plugins()
result = pdm(["hello"])
assert result.output.strip() == "Hello world"
result = pdm(["config", "foo"])
assert result.output.strip() == "bar"
@pytest.mark.usefixtures("local_finder")
def test_project_plugin_library(pdm, project, core, monkeypatch):
monkeypatch.setattr(sys, "path", sys.path[:])
project.pyproject.settings["plugins"] = ["pdm-hello"]
pdm(["install", "--plugins"], obj=project, strict=True)
assert project.root.joinpath(".pdm-plugins").exists()
assert "pdm-hello" not in project.environment.get_working_set()
with cd(project.root):
core.load_plugins()
result = pdm(["hello", "Frost"], strict=True)
assert result.stdout.strip() == "Hello, Frost!"
@pytest.mark.parametrize(
"req_str",
[
"-e file:///${PROJECT_ROOT}/plugins/test-plugin-pdm",
"-e test_plugin_pdm @ file:///${PROJECT_ROOT}/plugins/test-plugin-pdm",
],
)
@pytest.mark.usefixtures("local_finder")
def test_install_local_plugin_without_name(pdm, project, core, req_str):
import shutil
from . import FIXTURES
test_plugin_path = FIXTURES / "projects" / "test-plugin-pdm"
project.root.joinpath("plugins").mkdir(exist_ok=True)
shutil.copytree(test_plugin_path, project.root / "plugins" / "test-plugin-pdm", dirs_exist_ok=True)
project.pyproject.settings["plugins"] = [req_str]
project.pyproject.write()
with cd(project.root):
result = pdm(["install", "--plugins", "-vv"], obj=project, strict=True)
assert project.root.joinpath(".pdm-plugins").exists()
core.load_plugins()
result = pdm(["hello", "--name", "Frost"], strict=True)
assert result.stdout.strip() == "Hello, Frost"
|
