[test] fix tests related to timings

watson · Jun 8, 2021 · 5383f19 · 5383f19
1 parent e33aa21
commit 5383f19
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/ui/public/visualize/components/visualization_chart.test.js b/src/ui/public/visualize/components/visualization_chart.test.js
@@ -19,12 +19,15 @@
 
 jest.useFakeTimers();
 
+import { promisify } from 'util';
 import React from 'react';
 import { render, mount } from 'enzyme';
 import { VisualizationChart } from './visualization_chart';
 
 let renderPromise;
 
+const nextTick = promisify(process.nextTick);
+
 class VisualizationStub {
   constructor(el, vis) {
     this.el = el;
@@ -70,6 +73,7 @@ describe('<VisualizationChart/>', () => {
 
     jest.runAllTimers();
     await renderPromise;
+    await nextTick();
     expect(renderStart).toHaveBeenCalledTimes(1);
     expect(renderComplete).toHaveBeenCalledTimes(1);
 
@@ -89,12 +93,14 @@ describe('<VisualizationChart/>', () => {
     domNode.addEventListener('renderComplete', renderComplete);
     jest.runAllTimers();
     await renderPromise;
+    await nextTick();
     expect(renderComplete).toHaveBeenCalledTimes(1);
 
     vis.params.markdown = 'new text';
     wrapper.setProps({ vis });
     jest.runAllTimers();
     await renderPromise;
+    await nextTick();
 
     expect(wrapper.find('.visChart').text()).toBe('new text');
     expect(renderComplete).toHaveBeenCalledTimes(2);