This adds a ticker mode to ScheduledDataLoaderRegistry - testing works bitches! Found a bug in the sum code that I refactored way

Author: bbakerman

src/main/java/org/dataloader/registries/ScheduledDataLoaderRegistry.java

@@ -96,7 +96,7 @@ public int dispatchAllWithCount() {
         for (Map.Entry<String, DataLoader<?, ?>> entry : dataLoaders.entrySet()) {
             DataLoader<?, ?> dataLoader = entry.getValue();
             String key = entry.getKey();
-            dispatchOrReschedule(key, dataLoader);
+            sum += dispatchOrReschedule(key, dataLoader);
         }
         return sum;
     }
@@ -134,14 +134,16 @@ private void reschedule(String key, DataLoader<?, ?> dataLoader) {
         }
     }
 
-    private void dispatchOrReschedule(String key, DataLoader<?, ?> dataLoader) {
+    private int dispatchOrReschedule(String key, DataLoader<?, ?> dataLoader) {
+        int sum = 0;
         boolean shouldDispatch = dispatchPredicate.test(key, dataLoader);
         if (shouldDispatch) {
-            dataLoader.dispatch();
+            sum = dataLoader.dispatchWithCounts().getKeysCount();
         }
         if (tickerMode || !shouldDispatch) {
             reschedule(key, dataLoader);
         }
+        return sum;
     }
 
     /**

src/test/java/org/dataloader/registries/ScheduledDataLoaderRegistryTest.java

@@ -284,7 +284,8 @@ public void test_can_tick_after_first_dispatch_for_chain_data_loaders() {
 
         assertThat(registry.isTickerMode(), equalTo(true));
 
-        registry.dispatchAll();
+        int count = registry.dispatchAllWithCount();
+        assertThat(count,equalTo(1));
 
         await().atMost(TWO_SECONDS).untilAtomic(done, is(true));
 
@@ -312,7 +313,8 @@ public void test_chain_data_loaders_will_hang_if_not_in_ticker_mode() {
 
         assertThat(registry.isTickerMode(), equalTo(false));
 
-        registry.dispatchAll();
+        int count = registry.dispatchAllWithCount();
+        assertThat(count,equalTo(1));
 
         try {
             await().atMost(TWO_SECONDS).untilAtomic(done, is(true));