Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions vulnerabilities/pipes/advisory.py
Original file line number Diff line number Diff line change
Expand Up @@ -304,7 +304,7 @@ def insert_advisory_v2(
patches = get_or_create_advisory_patches(patches=advisory.patches)
weaknesses = get_or_create_advisory_weaknesses(weaknesses=advisory.weaknesses)
content_id = compute_content_id_v2(advisory_data=advisory)

created = False
try:
default_data = {
"datasource_id": pipeline_id,
Expand All @@ -314,11 +314,13 @@ def insert_advisory_v2(
"date_published": advisory.date_published,
"date_collected": datetime.now(timezone.utc),
"original_advisory_text": advisory.original_advisory_text,
"url": advisory.url,
}

advisory_obj, created = AdvisoryV2.objects.get_or_create(
advisory_id=advisory.advisory_id,
datasource_id=pipeline_id,
unique_content_id=content_id,
url=advisory.url,
defaults=default_data,
)
related_fields = {
Expand All @@ -334,7 +336,7 @@ def insert_advisory_v2(
getattr(advisory_obj, field_name).add(*values)

except Advisory.MultipleObjectsReturned:
logger.error(
logger(
f"Multiple Advisories returned: unique_content_id: {content_id}, url: {advisory.url}, advisory: {advisory!r}"
)
raise
Expand Down
47 changes: 47 additions & 0 deletions vulnerabilities/tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,12 +30,15 @@
from vulnerabilities.importer import PatchData
from vulnerabilities.importer import Reference
from vulnerabilities.importer import ReferenceV2
from vulnerabilities.importer import VulnerabilitySeverity
from vulnerabilities.models import AdvisorySeverity
from vulnerabilities.models import Alias
from vulnerabilities.models import Package
from vulnerabilities.models import Patch
from vulnerabilities.models import Vulnerability
from vulnerabilities.severity_systems import CVSSV3
from vulnerabilities.severity_systems import CVSSV4
from vulnerabilities.severity_systems import ScoringSystem
from vulnerabilities.utils import compute_content_id


Expand Down Expand Up @@ -771,3 +774,47 @@ def test_advisoryv2_to_advisory_data_patch_seralization(self):
result = models.AdvisoryV2.objects.first().to_advisory_data()

self.assertEqual(result, self.advisoryv2_data1)


class TestAdvisoryV2ModelDuplication(DjangoTestCase):
def setUp(self):
self.advisoryv2_data1 = AdvisoryDataV2(
advisory_id="CVE-2023-0401",
aliases=[],
summary="",
affected_packages=[],
severities=[
VulnerabilitySeverity(
system=CVSSV3,
value="7.5",
scoring_elements="CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:N/I:N/A:H",
)
],
patches=[],
url="https://test.com",
)

self.advisoryv2_data2 = AdvisoryDataV2(
advisory_id="CVE-2023-0662",
aliases=[],
summary="",
affected_packages=[],
severities=[
VulnerabilitySeverity(
system=CVSSV3,
value="7.5",
scoring_elements="CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:N/I:N/A:H",
)
],
patches=[],
url="https://test.com",
)

def test_advisoryv2_duplication_data(self):
from vulnerabilities.pipes.advisory import insert_advisory_v2

insert_advisory_v2(advisory=self.advisoryv2_data1, pipeline_id="test_pipeline")
insert_advisory_v2(advisory=self.advisoryv2_data2, pipeline_id="test_pipeline")
result = models.AdvisoryV2.objects.count()

self.assertEqual(result, 2)
1 change: 1 addition & 0 deletions vulnerabilities/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -689,6 +689,7 @@ def compute_content_id_v2(advisory_data):

elif isinstance(advisory_data, AdvisoryDataV2):
normalized_data = {
"advisory_id": normalize_text(advisory_data.advisory_id),
"aliases": normalize_list(advisory_data.aliases),
"summary": normalize_text(advisory_data.summary),
"affected_packages": [
Expand Down
Loading