@@ -333,66 +333,6 @@ void testCustomizeClient() {
333333 customizedTransport .closeGracefully ().block ();
334334 }
335335
336- @ Test
337- void testCustomizeRequest () {
338- // Create an atomic boolean to verify the customizer was called
339- AtomicBoolean customizerCalled = new AtomicBoolean (false );
340-
341- // Create a reference to store the custom header value
342- AtomicReference <String > headerName = new AtomicReference <>();
343- AtomicReference <String > headerValue = new AtomicReference <>();
344-
345- // Create a transport with the customizer
346- HttpClientSseClientTransport customizedTransport = HttpClientSseClientTransport .builder (host )
347- // Create a request customizer that adds a custom header
348- .customizeRequest (builder -> {
349- builder .header ("X-Custom-Header" , "test-value" );
350- customizerCalled .set (true );
351-
352- // Create a new request to verify the header was set
353- HttpRequest request = builder .uri (URI .create ("http://example.com" )).build ();
354- headerName .set ("X-Custom-Header" );
355- headerValue .set (request .headers ().firstValue ("X-Custom-Header" ).orElse (null ));
356- })
357- .build ();
358-
359- // Verify the customizer was called
360- assertThat (customizerCalled .get ()).isTrue ();
361-
362- // Verify the header was set correctly
363- assertThat (headerName .get ()).isEqualTo ("X-Custom-Header" );
364- assertThat (headerValue .get ()).isEqualTo ("test-value" );
365-
366- // Clean up
367- customizedTransport .closeGracefully ().block ();
368- }
369-
370- @ Test
371- void testChainedCustomizations () {
372- // Create atomic booleans to verify both customizers were called
373- AtomicBoolean clientCustomizerCalled = new AtomicBoolean (false );
374- AtomicBoolean requestCustomizerCalled = new AtomicBoolean (false );
375-
376- // Create a transport with both customizers chained
377- HttpClientSseClientTransport customizedTransport = HttpClientSseClientTransport .builder (host )
378- .customizeClient (builder -> {
379- builder .connectTimeout (Duration .ofSeconds (30 ));
380- clientCustomizerCalled .set (true );
381- })
382- .customizeRequest (builder -> {
383- builder .header ("X-Api-Key" , "test-api-key" );
384- requestCustomizerCalled .set (true );
385- })
386- .build ();
387-
388- // Verify both customizers were called
389- assertThat (clientCustomizerCalled .get ()).isTrue ();
390- assertThat (requestCustomizerCalled .get ()).isTrue ();
391-
392- // Clean up
393- customizedTransport .closeGracefully ().block ();
394- }
395-
396336 @ Test
397337 void testRequestCustomizer () {
398338 var mockCustomizer = mock (McpSyncHttpClientRequestCustomizer .class );
0 commit comments