Update vuln test cases

Author: bitterpanda63

aikido_zen/vulnerabilities/init_test.py

@@ -198,8 +198,13 @@ def test_ssrf_vulnerability_scan_bypassed_ip(get_context):
     run_vulnerability_scan(kind="ssrf", op="test", args=(dns_results, hostname, port))
     assert get_cache().stats.get_record()["requests"]["attacksDetected"]["total"] == 0
 
-    # Verify that hostnames.add was not called due to bypassed IP
-    assert get_cache().hostnames.as_array() == []
+    assert get_cache().hostnames.as_array() == [
+        {
+            "hits": 1,
+            "hostname": "example.com",
+            "port": 80,
+        },
+    ]
 
 
 def test_ssrf_vulnerability_scan_protection_gets_forced_off(get_context):
@@ -211,7 +216,7 @@ def test_ssrf_vulnerability_scan_protection_gets_forced_off(get_context):
     port = 80
     assert get_context.should_skip_attack_scan is None
     run_vulnerability_scan(kind="ssrf", op="test", args=(dns_results, hostname, port))
-    assert get_context.should_skip_attack_scan is False
+    assert get_context.should_skip_attack_scan is True  # Bypassed IP
 
 
 def test_sql_injection_with_protection_forced_off(caplog, get_context, monkeypatch):