优化

common-openapi/src/test/java/com/baiyue/common/openapi/CorrelationHeaderCustomizerTest.java

@@ -0,0 +1,25 @@
+package com.baiyue.common.openapi;
+
+import io.swagger.v3.oas.models.Operation;
+import io.swagger.v3.oas.models.parameters.Parameter;
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+import org.springdoc.core.customizers.OperationCustomizer;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+@DisplayName("OpenAPI 请求头定制")
+class CorrelationHeaderCustomizerTest {
+
+    @Test
+    void shouldOnlyExposeCallerControlledCorrelationHeaders() {
+        OperationCustomizer customizer = new BaiyueOpenApiAutoConfiguration().correlationHeaderCustomizer();
+
+        Operation operation = customizer.customize(new Operation(), null);
+
+        assertThat(operation.getParameters())
+                .extracting(Parameter::getName)
+                .containsExactly("X-Request-Id", "X-Trace-Id")
+                .doesNotContain("X-Tenant-Id", "X-Operator-Tenant-Id", "X-App-Code");
+    }
+}