Skip to content

Commit 911341b

Browse files
committed
Correct test with distinct()
Signed-off-by: Philippe Ombredanne <pombredanne@nexb.com>
1 parent 6b006bd commit 911341b

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

vulnerabilities/tests/test_fix_models.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,9 @@ def setUp(self):
5050

5151
def test_get_vulnerable_packages(self):
5252
vuln_packages = Package.objects.vulnerable()
53-
assert vuln_packages.count() == 10
54-
vuln_purls = [pkg.purl for pkg in vuln_packages.only(*PackageURL._fields)]
53+
assert vuln_packages.count() == 20
54+
assert vuln_packages.distinct().count() == 10
55+
vuln_purls = [pkg.purl for pkg in vuln_packages.distinct().only(*PackageURL._fields)]
5556
assert vuln_purls == [
5657
"pkg:generic/nginx/test@0",
5758
"pkg:generic/nginx/test@1",

0 commit comments

Comments
 (0)