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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
import pytest
from pip._internal.models.direct_url import (
ArchiveInfo,
DirectUrl,
DirectUrlValidationError,
DirInfo,
VcsInfo,
)
def test_from_json() -> None:
json = '{"url": "file:///home/user/project", "dir_info": {}}'
direct_url = DirectUrl.from_json(json)
assert direct_url.url == "file:///home/user/project"
assert isinstance(direct_url.info, DirInfo)
assert direct_url.info.editable is False
def test_to_json() -> None:
direct_url = DirectUrl(
url="file:///home/user/archive.tgz",
info=ArchiveInfo(),
)
direct_url.validate()
assert direct_url.to_json() == (
'{"archive_info": {}, "url": "file:///home/user/archive.tgz"}'
)
def test_archive_info() -> None:
direct_url_dict = {
"url": "file:///home/user/archive.tgz",
"archive_info": {"hash": "sha1=1b8c5bc61a86f377fea47b4276c8c8a5842d2220"},
}
direct_url = DirectUrl.from_dict(direct_url_dict)
assert isinstance(direct_url.info, ArchiveInfo)
assert direct_url.url == direct_url_dict["url"]
assert (
direct_url.info.hash == direct_url_dict["archive_info"]["hash"] # type: ignore
)
# test we add the hashes key automatically
direct_url_dict["archive_info"]["hashes"] = { # type: ignore
"sha1": "1b8c5bc61a86f377fea47b4276c8c8a5842d2220"
}
assert direct_url.to_dict() == direct_url_dict
def test_dir_info() -> None:
direct_url_dict = {
"url": "file:///home/user/project",
"dir_info": {"editable": True},
}
direct_url = DirectUrl.from_dict(direct_url_dict)
assert isinstance(direct_url.info, DirInfo)
assert direct_url.url == direct_url_dict["url"]
assert direct_url.info.editable is True
assert direct_url.to_dict() == direct_url_dict
# test editable default to False
direct_url_dict = {"url": "file:///home/user/project", "dir_info": {}}
direct_url = DirectUrl.from_dict(direct_url_dict)
assert isinstance(direct_url.info, DirInfo)
assert direct_url.info.editable is False
def test_vcs_info() -> None:
direct_url_dict = {
"url": "https:///g.c/u/p.git",
"vcs_info": {
"vcs": "git",
"requested_revision": "master",
"commit_id": "1b8c5bc61a86f377fea47b4276c8c8a5842d2220",
},
}
direct_url = DirectUrl.from_dict(direct_url_dict)
assert isinstance(direct_url.info, VcsInfo)
assert direct_url.url == direct_url_dict["url"]
assert direct_url.info.vcs == "git"
assert direct_url.info.requested_revision == "master"
assert direct_url.info.commit_id == "1b8c5bc61a86f377fea47b4276c8c8a5842d2220"
assert direct_url.to_dict() == direct_url_dict
def test_parsing_validation() -> None:
with pytest.raises(DirectUrlValidationError, match="url must have a value"):
DirectUrl.from_dict({"dir_info": {}})
with pytest.raises(
DirectUrlValidationError,
match="missing one of archive_info, dir_info, vcs_info",
):
DirectUrl.from_dict({"url": "http://..."})
with pytest.raises(DirectUrlValidationError, match="unexpected type for editable"):
DirectUrl.from_dict({"url": "http://...", "dir_info": {"editable": "false"}})
with pytest.raises(DirectUrlValidationError, match="unexpected type for hash"):
DirectUrl.from_dict({"url": "http://...", "archive_info": {"hash": 1}})
with pytest.raises(DirectUrlValidationError, match="unexpected type for vcs"):
DirectUrl.from_dict({"url": "http://...", "vcs_info": {"vcs": None}})
with pytest.raises(DirectUrlValidationError, match="commit_id must have a value"):
DirectUrl.from_dict({"url": "http://...", "vcs_info": {"vcs": "git"}})
with pytest.raises(
DirectUrlValidationError,
match="more than one of archive_info, dir_info, vcs_info",
):
DirectUrl.from_dict({"url": "http://...", "dir_info": {}, "archive_info": {}})
with pytest.raises(
DirectUrlValidationError,
match="invalid archive_info.hash format",
):
DirectUrl.from_dict(
{"url": "http://...", "archive_info": {"hash": "sha256:aaa"}}
)
def test_redact_url() -> None:
def _redact_git(url: str) -> str:
direct_url = DirectUrl(
url=url,
info=VcsInfo(vcs="git", commit_id="1"),
)
return direct_url.redacted_url
def _redact_archive(url: str) -> str:
direct_url = DirectUrl(
url=url,
info=ArchiveInfo(),
)
return direct_url.redacted_url
assert (
_redact_git("https://user:password@g.c/u/p.git@branch#egg=pkg")
== "https://g.c/u/p.git@branch#egg=pkg"
)
assert _redact_git("https://${USER}:password@g.c/u/p.git") == "https://g.c/u/p.git"
assert (
_redact_archive("file://${U}:${PIP_PASSWORD}@g.c/u/p.tgz")
== "file://${U}:${PIP_PASSWORD}@g.c/u/p.tgz"
)
assert (
_redact_git("https://${PIP_TOKEN}@g.c/u/p.git")
== "https://${PIP_TOKEN}@g.c/u/p.git"
)
assert _redact_git("ssh://git@g.c/u/p.git") == "ssh://git@g.c/u/p.git"
def test_hash_to_hashes() -> None:
direct_url = DirectUrl(url="https://e.c/archive.tar.gz", info=ArchiveInfo())
assert isinstance(direct_url.info, ArchiveInfo)
direct_url.info.hash = "sha256=abcdef"
assert direct_url.info.hashes == {"sha256": "abcdef"}
def test_hash_to_hashes_constructor() -> None:
direct_url = DirectUrl(
url="https://e.c/archive.tar.gz", info=ArchiveInfo(hash="sha256=abcdef")
)
assert isinstance(direct_url.info, ArchiveInfo)
assert direct_url.info.hashes == {"sha256": "abcdef"}
direct_url = DirectUrl(
url="https://e.c/archive.tar.gz",
info=ArchiveInfo(hash="sha256=abcdef", hashes={"sha512": "123456"}),
)
assert isinstance(direct_url.info, ArchiveInfo)
assert direct_url.info.hashes == {"sha256": "abcdef", "sha512": "123456"}
# In case of conflict between hash and hashes, hashes wins.
direct_url = DirectUrl(
url="https://e.c/archive.tar.gz",
info=ArchiveInfo(
hash="sha256=abcdef", hashes={"sha256": "012345", "sha512": "123456"}
),
)
assert isinstance(direct_url.info, ArchiveInfo)
assert direct_url.info.hashes == {"sha256": "012345", "sha512": "123456"}
|
