Updated tests

src-docs/src/views/date_picker/super_date_picker.js

@@ -26,6 +26,8 @@ function MyCustomQuickSelectPanel({ applyTime }) {
 export default class extends Component {
 
   state = {
+    refreshInterval: 1000,
+    isPaused: false,
     recentlyUsedRanges: [],
     isLoading: false,
     showUpdateButton: true,
@@ -52,7 +54,13 @@ export default class extends Component {
   }
 
   onRefresh = ({ start, end, refreshInterval }) => {
-    console.log(start, end, refreshInterval);
+    console.log('onRefresh was called');
+    return new Promise((resolve) => {
+
+      setTimeout(resolve, 100);
+    }).then(() => {
+      console.log(start, end, refreshInterval);
+    });
   }
 
   onStartInputChange = e => {

src/components/date_picker/super_date_picker/async_interval.test.js

@@ -1,31 +1,67 @@
 import { AsyncInterval } from './async_interval';
+import { times } from 'lodash';
 
-jest.useFakeTimers();
+describe('AsyncInterval',  () => {
 
-async function waitFor(instance, ms) {
-  jest.advanceTimersByTime(ms);
-  await instance.__pendingFn;
-}
+  beforeEach(() => {
+    jest.useFakeTimers();
+  });
+
+  afterEach(() => {
+    jest.useRealTimers();
+  });
+
+  async function andvanceTimerAndAwaitFn(instance, ms) {
+    const iterations = Math.floor(ms / 100);
+    const remainder = ms % 100;
+    // eslint-disable-next-line no-unused-vars
+    for (const item of times(iterations)) {
+      await instance.__pendingFn;
+      jest.advanceTimersByTime(100);
+      await instance.__pendingFn;
+    }
+    jest.advanceTimersByTime(remainder);
+    await instance.__pendingFn;
+  }
 
-describe('AsyncInterval',  () => {
   test('should call fn 3 times', async () => {
     const spy = jest.fn();
-    const instance = new AsyncInterval(spy, 1000);
-    await waitFor(instance, 2000);
-    await waitFor(instance, 2000);
-    await waitFor(instance, 2000);
-
+    const refrestInterval = 1000;
+    const instance = new AsyncInterval(spy, refrestInterval);
+    await andvanceTimerAndAwaitFn(instance, 3000);
     expect(spy).toHaveBeenCalledTimes(3);
   });
 
   test('should not call fn after stop has been called', async () => {
     const spy = jest.fn();
     const instance = new AsyncInterval(spy, 1000);
-    await waitFor(instance, 2000);
+
+    await andvanceTimerAndAwaitFn(instance, 1000);
+    expect(spy).toHaveBeenCalledTimes(1);
+
+    await andvanceTimerAndAwaitFn(instance, 1000);
+    expect(spy).toHaveBeenCalledTimes(2);
+
     instance.stop();
-    await waitFor(instance, 2000);
-    await waitFor(instance, 2000);
+    await andvanceTimerAndAwaitFn(instance, 1000);
+    expect(spy).toHaveBeenCalledTimes(2);
+  });
+
+  test('should wait invoking next refresh interval until promise has resolved', async () => {
+    const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+    const spy = jest.fn(async () => await sleep(500));
+    const instance = new AsyncInterval(spy, 1000);
 
+    await andvanceTimerAndAwaitFn(instance, 0);
+    expect(spy).toHaveBeenCalledTimes(0);
+
+    await andvanceTimerAndAwaitFn(instance, 1000);
     expect(spy).toHaveBeenCalledTimes(1);
+
+    await andvanceTimerAndAwaitFn(instance, 1500);
+    expect(spy).toHaveBeenCalledTimes(2);
+
+    await andvanceTimerAndAwaitFn(instance, 1500);
+    expect(spy).toHaveBeenCalledTimes(3);
   });
 });

src/components/date_picker/super_date_picker/super_date_picker.js

@@ -64,7 +64,9 @@ export class EuiSuperDatePicker extends Component {
     onRefreshChange: PropTypes.func,
 
     /**
-     * Callback for when the refresh interval is fired
+     * Callback for when the refresh interval is fired. Called with { start, end, refreshInterval }
+     * If a promise is returned, the next refresh interval will not start until the promise has resolved.
+     * If the promise rejects the refresh interval will stop and the error thrown
      */
     onRefresh: PropTypes.func,