@@ -704,7 +704,8 @@ void testPingMultipleCalls() {
704704
705705 @ Test
706706 void testListTasksWithNullCapabilities () {
707- McpAsyncServerExchange exchangeWithNullCapabilities = new McpAsyncServerExchange (mockSession , null , clientInfo );
707+ McpAsyncServerExchange exchangeWithNullCapabilities = new McpAsyncServerExchange ("testSessionId" , mockSession ,
708+ null , clientInfo , McpTransportContext .EMPTY );
708709
709710 StepVerifier .create (exchangeWithNullCapabilities .listTasks ("cursor" )).verifyErrorSatisfies (error -> {
710711 assertThat (error ).isInstanceOf (IllegalStateException .class )
@@ -720,8 +721,8 @@ void testListTasksWithoutTasksCapabilities() {
720721 .roots (true )
721722 .build ();
722723
723- McpAsyncServerExchange exchangeWithoutTasks = new McpAsyncServerExchange (mockSession , capabilitiesWithoutTasks ,
724- clientInfo );
724+ McpAsyncServerExchange exchangeWithoutTasks = new McpAsyncServerExchange ("testSessionId" , mockSession ,
725+ capabilitiesWithoutTasks , clientInfo , McpTransportContext . EMPTY );
725726
726727 StepVerifier .create (exchangeWithoutTasks .listTasks ("cursor" )).verifyErrorSatisfies (error -> {
727728 assertThat (error ).isInstanceOf (IllegalStateException .class )
@@ -737,8 +738,8 @@ void testListTasksWithoutListCapability() {
737738 .tasks (McpSchema .ClientCapabilities .ClientTaskCapabilities .builder ().build ())
738739 .build ();
739740
740- McpAsyncServerExchange exchangeWithoutList = new McpAsyncServerExchange (mockSession , capabilitiesWithoutList ,
741- clientInfo );
741+ McpAsyncServerExchange exchangeWithoutList = new McpAsyncServerExchange ("testSessionId" , mockSession ,
742+ capabilitiesWithoutList , clientInfo , McpTransportContext . EMPTY );
742743
743744 StepVerifier .create (exchangeWithoutList .listTasks ("cursor" )).verifyErrorSatisfies (error -> {
744745 assertThat (error ).isInstanceOf (IllegalStateException .class )
@@ -854,7 +855,8 @@ void testListTasksWithMultiplePages() {
854855
855856 @ Test
856857 void testCancelTaskWithNullCapabilities () {
857- McpAsyncServerExchange exchangeWithNullCapabilities = new McpAsyncServerExchange (mockSession , null , clientInfo );
858+ McpAsyncServerExchange exchangeWithNullCapabilities = new McpAsyncServerExchange ("testSessionId" , mockSession ,
859+ null , clientInfo , McpTransportContext .EMPTY );
858860
859861 StepVerifier
860862 .create (exchangeWithNullCapabilities
@@ -873,8 +875,8 @@ void testCancelTaskWithoutTasksCapabilities() {
873875 .roots (true )
874876 .build ();
875877
876- McpAsyncServerExchange exchangeWithoutTasks = new McpAsyncServerExchange (mockSession , capabilitiesWithoutTasks ,
877- clientInfo );
878+ McpAsyncServerExchange exchangeWithoutTasks = new McpAsyncServerExchange ("testSessionId" , mockSession ,
879+ capabilitiesWithoutTasks , clientInfo , McpTransportContext . EMPTY );
878880
879881 StepVerifier
880882 .create (exchangeWithoutTasks .cancelTask (McpSchema .CancelTaskRequest .builder ().taskId ("task-1" ).build ()))
@@ -892,8 +894,8 @@ void testCancelTaskWithoutCancelCapability() {
892894 .tasks (McpSchema .ClientCapabilities .ClientTaskCapabilities .builder ().build ())
893895 .build ();
894896
895- McpAsyncServerExchange exchangeWithoutCancel = new McpAsyncServerExchange (mockSession ,
896- capabilitiesWithoutCancel , clientInfo );
897+ McpAsyncServerExchange exchangeWithoutCancel = new McpAsyncServerExchange ("testSessionId" , mockSession ,
898+ capabilitiesWithoutCancel , clientInfo , McpTransportContext . EMPTY );
897899
898900 StepVerifier
899901 .create (exchangeWithoutCancel .cancelTask (McpSchema .CancelTaskRequest .builder ().taskId ("task-1" ).build ()))
0 commit comments