Minor code improvement.

Author: henningandersen

modules/reindex/src/test/java/org/elasticsearch/index/reindex/remote/RemoteScrollableHitSourceTests.java

@@ -135,25 +135,6 @@ public void validateAllConsumed() {
         assertTrue(responseQueue.isEmpty());
     }
 
-    private <T> RejectAwareActionListener<T> wrapAsListener(Consumer<T> consumer) {
-        return new RejectAwareActionListener<>() {
-            @Override
-            public void onRejection(Exception e) {
-                throw new AssertionError(e);
-            }
-
-            @Override
-            public void onResponse(T t) {
-                consumer.accept(t);
-            }
-
-            @Override
-            public void onFailure(Exception e) {
-                throw new AssertionError(e);
-            }
-        };
-    }
-
     public void testLookupRemoteVersion() throws Exception {
         assertLookupRemoteVersion(Version.fromString("0.20.5"), "main/0_20_5.json");
         assertLookupRemoteVersion(Version.fromString("0.90.13"), "main/0_90_13.json");
@@ -610,6 +591,13 @@ private class TestRemoteScrollableHitSource extends RemoteScrollableHitSource {
         }
     }
 
+    private <T> RejectAwareActionListener<T> wrapAsListener(Consumer<T> consumer) {
+        Consumer<Exception> throwing = e -> {
+            throw new AssertionError(e);
+        };
+        return RejectAwareActionListener.wrap(consumer::accept, throwing, throwing);
+    }
+
     @SuppressWarnings("unchecked")
     private <T extends Exception, V> T expectListenerFailure(Class<T> expectedException, Consumer<RejectAwareActionListener<V>> subject) {
         AtomicReference<T> exception = new AtomicReference<>();