Skip to content

Commit 2326a09

Browse files
committed
Retrun list in get_package_versions
Signed-off-by: Keshav Priyadarshi <[email protected]>
1 parent 979fb06 commit 2326a09

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

vulnerabilities/improvers/valid_versions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,11 @@ def get_package_versions(
6969
Return a list of versions published before `until` for the `package_url`
7070
"""
7171
versions = package_versions.versions(str(package_url))
72-
versions_before_until = set()
72+
versions_before_until = []
7373
for version in versions or []:
7474
if until and version.release_date and version.release_date > until:
7575
continue
76-
versions_before_until.add(version.value)
76+
versions_before_until.append(version.value)
7777

7878
return versions_before_until
7979

vulnerabilities/tests/test_github.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ def test_get_package_versions(mock_response):
285285
mock_response.return_value = json.load(f)
286286

287287
improver = GitHubBasicImprover()
288-
valid_versions = {
288+
valid_versions = [
289289
"1.1.3",
290290
"1.1.4",
291291
"1.10",
@@ -300,11 +300,12 @@ def test_get_package_versions(mock_response):
300300
"1.10a1",
301301
"1.10b1",
302302
"1.10rc1",
303-
}
304-
assert (
303+
]
304+
result = sorted(
305305
improver.get_package_versions(package_url=PackageURL(type="pypi", name="django"))
306-
== valid_versions
307306
)
307+
expected = sorted(valid_versions)
308+
assert result == expected
308309

309310

310311
def test_get_cwes_from_github_advisory():

0 commit comments

Comments
 (0)