@@ -125,7 +125,7 @@ def test_mask_sensitive_data_ignores_empty_normalized_keys():
125125 assert masked ["password" ] == "******"
126126
127127
128- def test_mask_sensitive_data_uses_rust_sidecar_when_enabled (monkeypatch ):
128+ def test_mask_sensitive_data_uses_pyo3_module_when_enabled (monkeypatch ):
129129 rust_module = MagicMock ()
130130 rust_module .mask_sensitive_data .return_value = {"password" : "******" , "username" : "user" } # pragma: allowlist secret
131131 monkeypatch .setattr ("mcpgateway.middleware.request_logging_middleware.settings.experimental_rust_request_logging_masking_enabled" , True , raising = False )
@@ -137,14 +137,6 @@ def test_mask_sensitive_data_uses_rust_sidecar_when_enabled(monkeypatch):
137137 rust_module .mask_sensitive_data .assert_called_once_with ({"password" : "secret" , "username" : "user" }, 10 ) # pragma: allowlist secret
138138
139139
140- def test_mask_sensitive_headers_missing_sidecar_is_hard_failure_when_enabled (monkeypatch ):
141- monkeypatch .setattr ("mcpgateway.middleware.request_logging_middleware.settings.experimental_rust_request_logging_masking_enabled" , True , raising = False )
142-
143- with patch ("mcpgateway.middleware.request_logging_middleware._load_rust_request_logging_module" , side_effect = ModuleNotFoundError ("missing sidecar" )):
144- with pytest .raises (ModuleNotFoundError , match = "missing sidecar" ):
145- mask_sensitive_headers ({"Authorization" : "Bearer abc" })
146-
147-
148140# --- mask_jwt_in_cookies tests ---
149141
150142
@@ -204,6 +196,18 @@ def test_mask_sensitive_headers_respects_non_sensitive_suffixes():
204196 assert masked ["X-JWT_Status_Count" ] == "7"
205197
206198
199+ def test_mask_sensitive_headers_uses_pyo3_module_when_enabled (monkeypatch ):
200+ rust_module = MagicMock ()
201+ rust_module .mask_sensitive_headers .return_value = {"Authorization" : "******" }
202+ monkeypatch .setattr ("mcpgateway.middleware.request_logging_middleware.settings.experimental_rust_request_logging_masking_enabled" , True , raising = False )
203+
204+ with patch ("mcpgateway.middleware.request_logging_middleware._load_rust_request_logging_module" , return_value = rust_module ):
205+ masked = mask_sensitive_headers ({"Authorization" : "Bearer abc" })
206+
207+ assert masked == {"Authorization" : "******" }
208+ rust_module .mask_sensitive_headers .assert_called_once_with ({"Authorization" : "Bearer abc" })
209+
210+
207211# --- RequestLoggingMiddleware tests ---
208212
209213
0 commit comments