No need to wrap the listener

elastic · Apr 15, 2019 · 1350c0f · 1350c0f
1 parent fb1f7eb
commit 1350c0f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...st/java/org/elasticsearch/action/support/replication/TransportReplicationActionTests.java b/...st/java/org/elasticsearch/action/support/replication/TransportReplicationActionTests.java
@@ -659,12 +659,12 @@ protected void shardOperationOnPrimary(Request shardRequest, IndexShard primary,
                 assertFalse(executed.getAndSet(true));
                 super.shardOperationOnPrimary(shardRequest, primary, listener);
             }
-        }.new AsyncPrimaryAction(primaryRequest, ActionListener.wrap(listener::onResponse, throwable -> {
-            throw new RuntimeException(throwable);
-        }), task).run();
+        }.new AsyncPrimaryAction(primaryRequest, listener, task).run();
         assertThat(executed.get(), equalTo(true));
         assertPhase(task, "finished");
         assertFalse(request.isRetrySet.get());
+        assertTrue(listener.isDone());
+        listener.actionGet(); // throws no exception
     }
 
     public void testPrimaryReference() {