From e1bfd9a4769fb2cc78bb15efdd15295a8f602232 Mon Sep 17 00:00:00 2001 From: Cody Maffucci <46459665+Maffooch@users.noreply.github.com> Date: Tue, 21 May 2024 13:08:18 -0500 Subject: [PATCH] Update tests --- unittests/test_import_reimport.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/unittests/test_import_reimport.py b/unittests/test_import_reimport.py index f16bcb0dc57..c2e3aadf568 100644 --- a/unittests/test_import_reimport.py +++ b/unittests/test_import_reimport.py @@ -1449,15 +1449,12 @@ def test_import_reimport_vulnerability_ids(self): self.assertEqual('CVE-2021-44420', findings[3].vulnerability_ids[1]) def test_import_history_reactivated_and_untouched_findings_do_not_mix(self): - with assertTestImportModelsCreated(self, imports=1, affected_findings=4, created=4): - import0 = self.import_scan_with_params(self.generic_import_1, scan_type=self.scan_type_generic) - test_id = import0['test'] + import0 = self.import_scan_with_params(self.generic_import_1, scan_type=self.scan_type_generic) + test_id = import0['test'] # reimport the second report - with assertTestImportModelsCreated(self, reimports=1, affected_findings=4, closed=3, reactivated=1): - self.reimport_scan_with_params(test_id, self.generic_import_2, scan_type=self.scan_type_generic) + self.reimport_scan_with_params(test_id, self.generic_import_2, scan_type=self.scan_type_generic) # reimport the first report again - with assertTestImportModelsCreated(self, reimports=1, affected_findings=2, closed=1, reactivated=1): - self.reimport_scan_with_params(test_id, self.generic_import_1, scan_type=self.scan_type_generic) + self.reimport_scan_with_params(test_id, self.generic_import_1, scan_type=self.scan_type_generic) # Passing this test means an exception does not occur