@@ -220,13 +220,11 @@ public void testSendResponse() throws IOException {
220220
221221 AtomicLong requestIdRef = new AtomicLong ();
222222 AtomicReference <String > actionRef = new AtomicReference <>();
223- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
224223 handler .setMessageListener (new TransportMessageListener () {
225224 @ Override
226- public void onResponseSent (long requestId , String action , TransportResponse response ) {
225+ public void onResponseSent (long requestId , String action ) {
227226 requestIdRef .set (requestId );
228227 actionRef .set (action );
229- responseRef .set (response );
230228 }
231229 });
232230 if (compress ) {
@@ -244,7 +242,6 @@ public void onResponseSent(long requestId, String action, TransportResponse resp
244242 }
245243 assertEquals (requestId , requestIdRef .get ());
246244 assertEquals (action , actionRef .get ());
247- assertEquals (response , responseRef .get ());
248245
249246 pipeline .handleBytes (channel , new ReleasableBytesReference (reference , () -> {}));
250247 final Tuple <Header , BytesReference > tuple = message .get ();
@@ -338,18 +335,15 @@ public void writeTo(StreamOutput out) {
338335
339336 AtomicLong requestIdRef = new AtomicLong ();
340337 AtomicReference <String > actionRef = new AtomicReference <>();
341- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
342338 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
343339 handler .setMessageListener (new TransportMessageListener () {
344340 @ Override
345- public void onResponseSent (long requestId , String action , TransportResponse response ) {
341+ public void onResponseSent (long requestId , String action ) {
346342 assertNull (channel .getMessageCaptor ().get ());
347343 assertThat (requestIdRef .get (), equalTo (0L ));
348344 requestIdRef .set (requestId );
349345 assertNull (actionRef .get ());
350346 actionRef .set (action );
351- assertNull (responseRef .get ());
352- responseRef .set (response );
353347 }
354348
355349 @ Override
@@ -374,7 +368,6 @@ public void onResponseSent(long requestId, String action, Exception error) {
374368 }
375369 assertEquals (requestId , requestIdRef .get ());
376370 assertEquals (action , actionRef .get ());
377- assertEquals (response , responseRef .get ());
378371 assertThat (exceptionRef .get ().getMessage (), equalTo ("simulated cbe" ));
379372 assertTrue (response .released .get ());
380373 BytesReference reference = channel .getMessageCaptor ().get ();
@@ -400,18 +393,15 @@ public void sendMessage(BytesReference reference, ActionListener<Void> listener)
400393 long requestId = randomLongBetween (0 , 300 );
401394 AtomicLong requestIdRef = new AtomicLong ();
402395 AtomicReference <String > actionRef = new AtomicReference <>();
403- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
404396 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
405397 handler .setMessageListener (new TransportMessageListener () {
406398 @ Override
407- public void onResponseSent (long requestId , String action , TransportResponse response ) {
399+ public void onResponseSent (long requestId , String action ) {
408400 assertNull (channel .getMessageCaptor ().get ());
409401 assertThat (requestIdRef .get (), equalTo (0L ));
410402 requestIdRef .set (requestId );
411403 assertNull (actionRef .get ());
412404 actionRef .set (action );
413- assertNull (responseRef .get ());
414- responseRef .set (response );
415405 }
416406
417407 @ Override
@@ -436,7 +426,6 @@ public void writeTo(StreamOutput out) {
436426 }
437427 assertEquals (requestId , requestIdRef .get ());
438428 assertEquals (action , actionRef .get ());
439- assertEquals (response , responseRef .get ());
440429 assertThat (exceptionRef .get ().getMessage (), equalTo ("simulated cbe" ));
441430 assertTrue (response .released .get ());
442431 assertNull (channel .getMessageCaptor ().get ());
@@ -457,17 +446,14 @@ public void writeTo(StreamOutput out) {
457446
458447 AtomicLong requestIdRef = new AtomicLong ();
459448 AtomicReference <String > actionRef = new AtomicReference <>();
460- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
461449 handler .setMessageListener (new TransportMessageListener () {
462450 @ Override
463- public void onResponseSent (long requestId , String action , TransportResponse response ) {
451+ public void onResponseSent (long requestId , String action ) {
464452 assertNull (channel .getMessageCaptor ().get ());
465453 assertThat (requestIdRef .get (), equalTo (0L ));
466454 requestIdRef .set (requestId );
467455 assertNull (actionRef .get ());
468456 actionRef .set (action );
469- assertNull (responseRef .get ());
470- responseRef .set (response );
471457 }
472458
473459 @ Override
@@ -485,7 +471,6 @@ public void onResponseSent(long requestId, String action, Exception error) {
485471 assertNull (channel .getListenerCaptor ().get ());
486472 assertEquals (requestId , requestIdRef .get ());
487473 assertEquals (action , actionRef .get ());
488- assertEquals (response , responseRef .get ());
489474 assertTrue (response .released .get ());
490475 assertFalse (channel .isOpen ());
491476 }
@@ -507,7 +492,7 @@ public void sendMessage(BytesReference reference, ActionListener<Void> listener)
507492 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
508493 handler .setMessageListener (new TransportMessageListener () {
509494 @ Override
510- public void onResponseSent (long requestId , String action , TransportResponse response ) {
495+ public void onResponseSent (long requestId , String action ) {
511496 throw new AssertionError ("must not be called" );
512497 }
513498
0 commit comments