1 files changed, 13 insertions, 13 deletions

diff --git a/spec/frontend/vue_mr_widget/components/states/mr_widget_wip_spec.js b/spec/frontend/vue_mr_widget/components/states/mr_widget_wip_spec.js
index 4070ca8d8dc..4998147c6b6 100644
--- a/spec/frontend/vue_mr_widget/components/states/mr_widget_wip_spec.js
+++ b/spec/frontend/vue_mr_widget/components/states/mr_widget_wip_spec.js
@@ -1,4 +1,5 @@
-import Vue from 'vue';
+import Vue, { nextTick } from 'vue';
+import waitForPromises from 'helpers/wait_for_promises';
 import WorkInProgress from '~/vue_merge_request_widget/components/states/work_in_progress.vue';
 import toast from '~/vue_shared/plugins/global_toast';
 import eventHub from '~/vue_merge_request_widget/event_hub';
@@ -47,7 +48,7 @@ describe('Wip', () => {
     };
 
     describe('handleRemoveDraft', () => {
-      it('should make a request to service and handle response', (done) => {
+      it('should make a request to service and handle response', async () => {
         const vm = createComponent();
 
         jest.spyOn(eventHub, '$emit').mockImplementation(() => {});
@@ -60,12 +61,12 @@ describe('Wip', () => {
         );
 
         vm.handleRemoveDraft();
-        setImmediate(() => {
-          expect(vm.isMakingRequest).toBeTruthy();
-          expect(eventHub.$emit).toHaveBeenCalledWith('UpdateWidgetData', mrObj);
-          expect(toast).toHaveBeenCalledWith('Marked as ready. Merging is now allowed.');
-          done();
-        });
+
+        await waitForPromises();
+
+        expect(vm.isMakingRequest).toBeTruthy();
+        expect(eventHub.$emit).toHaveBeenCalledWith('UpdateWidgetData', mrObj);
+        expect(toast).toHaveBeenCalledWith('Marked as ready. Merging is now allowed.');
       });
     });
   });
@@ -91,13 +92,12 @@ describe('Wip', () => {
       );
     });
 
-    it('should not show removeWIP button is user cannot update MR', (done) => {
+    it('should not show removeWIP button is user cannot update MR', async () => {
       vm.mr.removeWIPPath = '';
 
-      Vue.nextTick(() => {
-        expect(el.querySelector('.js-remove-draft')).toEqual(null);
-        done();
-      });
+      await nextTick();
+
+      expect(el.querySelector('.js-remove-draft')).toEqual(null);
     });
   });
 });