JUnit enhancement

Author: Allen Wang

suro-elasticsearch/src/main/java/com/netflix/suro/sink/elasticsearch/ElasticSearchSink.java

@@ -402,4 +402,14 @@ protected void innerClose() {
     RestClient getClient() {
         return client;
     }
+
+    @VisibleForTesting
+    int getSleepOverClientException() {
+        return sleepOverClientException;
+    }
+
+    @VisibleForTesting
+    boolean getReenqueueOnException() {
+        return reEnqueueOnException;
+    }
 }

suro-elasticsearch/src/test/java/com/netflix/suro/sink/elasticsearch/TestElasticSearchSink.java

@@ -182,6 +182,8 @@ public Object findInjectableValue(
         IClientConfig config = ((BaseLoadBalancer) client.getLoadBalancer()).getClientConfig();
         assertTrue(config.get(CommonClientConfigKey.OkToRetryOnAllOperations));
         assertEquals(2, config.get(CommonClientConfigKey.MaxAutoRetriesNextServer).intValue());
+        assertEquals(0, esSink.getSleepOverClientException());
+        assertFalse(esSink.getReenqueueOnException());
     }
 
     @Test