Skip to content

Commit 7eae034

Browse files
committed
Fix advisory todo tests
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 532efd1 commit 7eae034

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

vulnerabilities/tests/pipelines/v2_improvers/test_compute_advisory_todo_v2.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,7 @@ def test_advisory_todo_missing_summary(self):
190190
)
191191
adv = AdvisoryV2.objects.first()
192192
adv.summary = ""
193+
adv._all_impacts_unfurled = True
193194
adv.save()
194195
pipeline = ComputeToDo()
195196
pipeline.execute()
@@ -200,13 +201,15 @@ def test_advisory_todo_missing_summary(self):
200201
self.assertEqual(1, todo.advisories.count())
201202

202203
def test_advisory_todo_missing_fixed(self):
203-
insert_advisory_v2(
204+
adv = insert_advisory_v2(
204205
advisory=self.advisory_data2,
205206
pipeline_id="test_pipeline1",
206207
logger=self.log.write,
207208
datasource_id="test1",
208209
)
209210
pipeline = ComputeToDo()
211+
adv._all_impacts_unfurled = True
212+
adv.save()
210213
pipeline.execute()
211214

212215
todo = AdvisoryToDoV2.objects.first()
@@ -215,12 +218,14 @@ def test_advisory_todo_missing_fixed(self):
215218
self.assertEqual(1, todo.advisories.count())
216219

217220
def test_advisory_todo_missing_affected(self):
218-
insert_advisory_v2(
221+
adv = insert_advisory_v2(
219222
advisory=self.advisory_data3,
220223
logger=self.log.write,
221224
datasource_id="test1",
222225
pipeline_id="test_pipeline1",
223226
)
227+
adv._all_impacts_unfurled = True
228+
adv.save()
224229
pipeline = ComputeToDo()
225230
pipeline.execute()
226231

0 commit comments

Comments
 (0)