Skip to content

Commit 3b5ca88

Browse files
committed
Fix tests
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent ab667dd commit 3b5ca88

4 files changed

Lines changed: 12 additions & 0 deletions

File tree

vulnerabilities/tests/pipelines/v2_improvers/test_enhance_with_exploitdb_v2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ def test_exploit_db_improver(mock_get):
4646
unique_content_id="i3giu",
4747
url="https://test.com",
4848
date_collected=datetime.now(),
49+
_all_impacts_unfurled_at=datetime.now(),
50+
is_latest=True,
4951
)
5052

5153
alias = AdvisoryAlias.objects.create(alias="CVE-2009-3699")

vulnerabilities/tests/pipelines/v2_improvers/test_enhance_with_github_poc.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,8 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
4242
url="https://test.com",
4343
date_collected=datetime.now(),
4444
pipeline_id="ghsa_pipeline_v2",
45+
is_latest=True,
46+
_all_impacts_unfurled_at=datetime.now(),
4547
)
4648
adv2 = AdvisoryV2.objects.create(
4749
advisory_id="VCIO-123-1002",
@@ -51,6 +53,8 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
5153
url="https://test.com",
5254
date_collected=datetime.now(),
5355
pipeline_id="ghsa_pipeline_v2",
56+
is_latest=True,
57+
_all_impacts_unfurled_at=datetime.now(),
5458
)
5559
adv3 = AdvisoryV2.objects.create(
5660
advisory_id="VCIO-123-1003",
@@ -60,6 +64,8 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
6064
url="https://test.com",
6165
date_collected=datetime.now(),
6266
pipeline_id="ghsa_pipeline_v2",
67+
is_latest=True,
68+
_all_impacts_unfurled_at=datetime.now(),
6369
)
6470

6571
alias1 = AdvisoryAlias.objects.create(alias="CVE-2022-0236")

vulnerabilities/tests/pipelines/v2_improvers/test_enhance_with_kev_v2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ def test_kev_improver(mock_get):
4646
unique_content_id="i3giu",
4747
url="https://test.com",
4848
date_collected=datetime.now(),
49+
is_latest=True,
50+
_all_impacts_unfurled_at=datetime.now(),
4951
)
5052
adv1.save()
5153

vulnerabilities/tests/pipelines/v2_improvers/test_enhance_with_metasploit_v2.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@ def test_metasploit_improver(mock_get):
4949
unique_content_id="i3giu",
5050
url="https://test.com",
5151
date_collected=datetime.now(),
52+
_all_impacts_unfurled_at=datetime.now(),
53+
is_latest=True,
5254
)
5355
alias = AdvisoryAlias.objects.create(alias="CVE-2007-4387")
5456

0 commit comments

Comments
 (0)