Fix mock assertion to include status_code and run black formatter

Author: bitterpanda63

aikido_zen/storage/attack_wave_detector_store_test.py

@@ -446,7 +446,7 @@ def test_mock_detector_integration(mock_detector_class):
     store = AttackWaveDetectorStore()
     context = test_utils.generate_context()
 
-    # Should use the mocked detector
+    # Should use the mocked detector (default status_code=404)
     result = store.is_attack_wave(context)
     assert result is True
-    mock_detector.is_attack_wave.assert_called_once_with(context)
+    mock_detector.is_attack_wave.assert_called_once_with(context, 404)

aikido_zen/vulnerabilities/attack_wave_detection/is_web_scanner_test.py

@@ -29,7 +29,9 @@ def test_is_web_scanner():
     assert is_web_scanner(get_test_context("/.aws/config", "GET"), 404)
     assert is_web_scanner(get_test_context("/../secret", "GET"), 404)
     assert is_web_scanner(get_test_context("/", "BADMETHOD"), 404)
-    assert is_web_scanner(get_test_context("/", "GET", {"test": "SELECT * FROM admin"}), 404)
+    assert is_web_scanner(
+        get_test_context("/", "GET", {"test": "SELECT * FROM admin"}), 404
+    )
     assert is_web_scanner(get_test_context("/", "GET", {"test": "../etc/passwd"}), 404)