1 files changed, 15 insertions, 23 deletions

diff --git a/spec/frontend/ide/components/error_message_spec.js b/spec/frontend/ide/components/error_message_spec.js
index 2de3fa863a8..17568158131 100644
--- a/spec/frontend/ide/components/error_message_spec.js
+++ b/spec/frontend/ide/components/error_message_spec.js
@@ -1,10 +1,10 @@
 import { GlLoadingIcon } from '@gitlab/ui';
-import { mount, createLocalVue } from '@vue/test-utils';
+import { mount } from '@vue/test-utils';
+import Vue, { nextTick } from 'vue';
 import Vuex from 'vuex';
 import ErrorMessage from '~/ide/components/error_message.vue';
 
-const localVue = createLocalVue();
-localVue.use(Vuex);
+Vue.use(Vuex);
 
 describe('IDE error message component', () => {
   let wrapper;
@@ -25,7 +25,6 @@ describe('IDE error message component', () => {
         },
       },
       store: fakeStore,
-      localVue,
     });
   };
 
@@ -87,19 +86,15 @@ describe('IDE error message component', () => {
       expect(actionMock).toHaveBeenCalledWith(message.actionPayload);
     });
 
-    it('does not dispatch action when already loading', () => {
+    it('does not dispatch action when already loading', async () => {
       findActionButton().trigger('click');
       actionMock.mockReset();
-      return wrapper.vm.$nextTick(() => {
-        findActionButton().trigger('click');
-
-        return wrapper.vm.$nextTick().then(() => {
-          expect(actionMock).not.toHaveBeenCalled();
-        });
-      });
+      findActionButton().trigger('click');
+      await nextTick();
+      expect(actionMock).not.toHaveBeenCalled();
     });
 
-    it('shows loading icon when loading', () => {
+    it('shows loading icon when loading', async () => {
       let resolveAction;
       actionMock.mockImplementation(
         () =>
@@ -109,19 +104,16 @@ describe('IDE error message component', () => {
       );
       findActionButton().trigger('click');
 
-      return wrapper.vm.$nextTick(() => {
-        expect(wrapper.find(GlLoadingIcon).isVisible()).toBe(true);
-        resolveAction();
-      });
+      await nextTick();
+      expect(wrapper.find(GlLoadingIcon).isVisible()).toBe(true);
+      resolveAction();
     });
 
-    it('hides loading icon when operation finishes', () => {
+    it('hides loading icon when operation finishes', async () => {
       findActionButton().trigger('click');
-      return actionMock()
-        .then(() => wrapper.vm.$nextTick())
-        .then(() => {
-          expect(wrapper.find(GlLoadingIcon).isVisible()).toBe(false);
-        });
+      await actionMock();
+      await nextTick();
+      expect(wrapper.find(GlLoadingIcon).isVisible()).toBe(false);
     });
   });
 });