@@ -221,13 +221,11 @@ public void testSendResponse() throws IOException {
221221
222222 AtomicLong requestIdRef = new AtomicLong ();
223223 AtomicReference <String > actionRef = new AtomicReference <>();
224- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
225224 handler .setMessageListener (new TransportMessageListener () {
226225 @ Override
227- public void onResponseSent (long requestId , String action , TransportResponse response ) {
226+ public void onResponseSent (long requestId , String action ) {
228227 requestIdRef .set (requestId );
229228 actionRef .set (action );
230- responseRef .set (response );
231229 }
232230 });
233231 if (compress ) {
@@ -245,7 +243,6 @@ public void onResponseSent(long requestId, String action, TransportResponse resp
245243 }
246244 assertEquals (requestId , requestIdRef .get ());
247245 assertEquals (action , actionRef .get ());
248- assertEquals (response , responseRef .get ());
249246
250247 pipeline .handleBytes (channel , new ReleasableBytesReference (reference , () -> {}));
251248 final Tuple <Header , BytesReference > tuple = message .get ();
@@ -339,18 +336,15 @@ public void writeTo(StreamOutput out) {
339336
340337 AtomicLong requestIdRef = new AtomicLong ();
341338 AtomicReference <String > actionRef = new AtomicReference <>();
342- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
343339 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
344340 handler .setMessageListener (new TransportMessageListener () {
345341 @ Override
346- public void onResponseSent (long requestId , String action , TransportResponse response ) {
342+ public void onResponseSent (long requestId , String action ) {
347343 assertNull (channel .getMessageCaptor ().get ());
348344 assertThat (requestIdRef .get (), equalTo (0L ));
349345 requestIdRef .set (requestId );
350346 assertNull (actionRef .get ());
351347 actionRef .set (action );
352- assertNull (responseRef .get ());
353- responseRef .set (response );
354348 }
355349
356350 @ Override
@@ -375,7 +369,6 @@ public void onResponseSent(long requestId, String action, Exception error) {
375369 }
376370 assertEquals (requestId , requestIdRef .get ());
377371 assertEquals (action , actionRef .get ());
378- assertEquals (response , responseRef .get ());
379372 assertThat (exceptionRef .get ().getMessage (), equalTo ("simulated cbe" ));
380373 assertTrue (response .released .get ());
381374 BytesReference reference = channel .getMessageCaptor ().get ();
@@ -401,18 +394,15 @@ public void sendMessage(BytesReference reference, ActionListener<Void> listener)
401394 long requestId = randomLongBetween (0 , 300 );
402395 AtomicLong requestIdRef = new AtomicLong ();
403396 AtomicReference <String > actionRef = new AtomicReference <>();
404- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
405397 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
406398 handler .setMessageListener (new TransportMessageListener () {
407399 @ Override
408- public void onResponseSent (long requestId , String action , TransportResponse response ) {
400+ public void onResponseSent (long requestId , String action ) {
409401 assertNull (channel .getMessageCaptor ().get ());
410402 assertThat (requestIdRef .get (), equalTo (0L ));
411403 requestIdRef .set (requestId );
412404 assertNull (actionRef .get ());
413405 actionRef .set (action );
414- assertNull (responseRef .get ());
415- responseRef .set (response );
416406 }
417407
418408 @ Override
@@ -437,7 +427,6 @@ public void writeTo(StreamOutput out) {
437427 }
438428 assertEquals (requestId , requestIdRef .get ());
439429 assertEquals (action , actionRef .get ());
440- assertEquals (response , responseRef .get ());
441430 assertThat (exceptionRef .get ().getMessage (), equalTo ("simulated cbe" ));
442431 assertTrue (response .released .get ());
443432 assertNull (channel .getMessageCaptor ().get ());
@@ -458,17 +447,14 @@ public void writeTo(StreamOutput out) {
458447
459448 AtomicLong requestIdRef = new AtomicLong ();
460449 AtomicReference <String > actionRef = new AtomicReference <>();
461- AtomicReference <TransportResponse > responseRef = new AtomicReference <>();
462450 handler .setMessageListener (new TransportMessageListener () {
463451 @ Override
464- public void onResponseSent (long requestId , String action , TransportResponse response ) {
452+ public void onResponseSent (long requestId , String action ) {
465453 assertNull (channel .getMessageCaptor ().get ());
466454 assertThat (requestIdRef .get (), equalTo (0L ));
467455 requestIdRef .set (requestId );
468456 assertNull (actionRef .get ());
469457 actionRef .set (action );
470- assertNull (responseRef .get ());
471- responseRef .set (response );
472458 }
473459
474460 @ Override
@@ -486,7 +472,6 @@ public void onResponseSent(long requestId, String action, Exception error) {
486472 assertNull (channel .getListenerCaptor ().get ());
487473 assertEquals (requestId , requestIdRef .get ());
488474 assertEquals (action , actionRef .get ());
489- assertEquals (response , responseRef .get ());
490475 assertTrue (response .released .get ());
491476 assertFalse (channel .isOpen ());
492477 }
@@ -508,7 +493,7 @@ public void sendMessage(BytesReference reference, ActionListener<Void> listener)
508493 AtomicReference <Exception > exceptionRef = new AtomicReference <>();
509494 handler .setMessageListener (new TransportMessageListener () {
510495 @ Override
511- public void onResponseSent (long requestId , String action , TransportResponse response ) {
496+ public void onResponseSent (long requestId , String action ) {
512497 throw new AssertionError ("must not be called" );
513498 }
514499
0 commit comments