Skip to content

Commit c6c8803

Browse files
ruff
1 parent 3325f7b commit c6c8803

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

unittests/test_reimport_prefetch.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def test_reimport_duplicate_findings_in_same_report_endpoints(self):
154154
"""
155155
# This previously raised:
156156
# AttributeError: 'Finding' object has no attribute 'status_finding_non_special'
157-
test, _, len_new, len_closed, _, _, _ = self._reimport_with_overridden_hashcode()
157+
_test, _, len_new, len_closed, _, _, _ = self._reimport_with_overridden_hashcode()
158158

159159
# The first finding is new (empty test, no candidates). The second finding
160160
# matches the first (same hash_code) so it is not counted as new.
@@ -185,11 +185,13 @@ def test_reimport_duplicate_findings_in_same_report_endpoints(self):
185185
# gets mitigated. This is arguably wrong for batch-created findings — both
186186
# endpoints came from the same report — but it is the current behavior.
187187
self.assertEqual(
188-
endpoint_statuses.filter(mitigated=False).count(), 1,
188+
endpoint_statuses.filter(mitigated=False).count(),
189+
1,
189190
"One endpoint status should be active (/app/dashboard from the matched finding)",
190191
)
191192
self.assertEqual(
192-
endpoint_statuses.filter(mitigated=True).count(), 1,
193+
endpoint_statuses.filter(mitigated=True).count(),
194+
1,
193195
"One endpoint status should be mitigated (/app/login — mitigated by update_endpoint_status "
194196
"because it is not in the second finding's endpoint list)",
195197
)
@@ -217,7 +219,7 @@ def test_reimport_duplicate_findings_in_same_report_locations(self):
217219
first finding is created as new and added to candidates. The second finding
218220
matches against it (a batch-created finding).
219221
"""
220-
test, _, len_new, len_closed, _, _, _ = self._reimport_with_overridden_hashcode()
222+
_test, _, len_new, len_closed, _, _, _ = self._reimport_with_overridden_hashcode()
221223

222224
# The first finding is new (empty test, no candidates). The second finding
223225
# matches the first (same hash_code) so it is not counted as new.
@@ -245,11 +247,13 @@ def test_reimport_duplicate_findings_in_same_report_locations(self):
245247
# This is arguably wrong for batch-created findings — both locations came from
246248
# the same report — but it is the current behavior.
247249
self.assertEqual(
248-
location_refs.filter(status=FindingLocationStatus.Active).count(), 1,
250+
location_refs.filter(status=FindingLocationStatus.Active).count(),
251+
1,
249252
"One location ref should be active (/app/dashboard from the matched finding)",
250253
)
251254
self.assertEqual(
252-
location_refs.filter(status=FindingLocationStatus.Mitigated).count(), 1,
255+
location_refs.filter(status=FindingLocationStatus.Mitigated).count(),
256+
1,
253257
"One location ref should be mitigated (/app/login — mitigated by update_location_status "
254258
"because it is not in the second finding's location list)",
255259
)

0 commit comments

Comments
 (0)