1 files changed, 11 insertions, 14 deletions

diff --git a/spec/frontend/ide/components/terminal/session_spec.js b/spec/frontend/ide/components/terminal/session_spec.js
index 5659a7d15da..6a70ddb46a8 100644
--- a/spec/frontend/ide/components/terminal/session_spec.js
+++ b/spec/frontend/ide/components/terminal/session_spec.js
@@ -1,5 +1,6 @@
 import { GlButton } from '@gitlab/ui';
-import { createLocalVue, shallowMount } from '@vue/test-utils';
+import { shallowMount } from '@vue/test-utils';
+import Vue, { nextTick } from 'vue';
 import Vuex from 'vuex';
 import TerminalSession from '~/ide/components/terminal/session.vue';
 import Terminal from '~/ide/components/terminal/terminal.vue';
@@ -13,8 +14,7 @@ import {
 
 const TEST_TERMINAL_PATH = 'terminal/path';
 
-const localVue = createLocalVue();
-localVue.use(Vuex);
+Vue.use(Vuex);
 
 describe('IDE TerminalSession', () => {
   let wrapper;
@@ -33,7 +33,6 @@ describe('IDE TerminalSession', () => {
     });
 
     wrapper = shallowMount(TerminalSession, {
-      localVue,
       store,
       ...options,
     });
@@ -68,32 +67,30 @@ describe('IDE TerminalSession', () => {
   });
 
   [STARTING, PENDING, RUNNING].forEach((status) => {
-    it(`show stop button when status is ${status}`, () => {
+    it(`show stop button when status is ${status}`, async () => {
       state.session = { status };
       factory();
 
       const button = findButton();
       button.vm.$emit('click');
 
-      return wrapper.vm.$nextTick().then(() => {
-        expect(button.text()).toEqual('Stop Terminal');
-        expect(actions.stopSession).toHaveBeenCalled();
-      });
+      await nextTick();
+      expect(button.text()).toEqual('Stop Terminal');
+      expect(actions.stopSession).toHaveBeenCalled();
     });
   });
 
   [STOPPING, STOPPED].forEach((status) => {
-    it(`show stop button when status is ${status}`, () => {
+    it(`show stop button when status is ${status}`, async () => {
       state.session = { status };
       factory();
 
       const button = findButton();
       button.vm.$emit('click');
 
-      return wrapper.vm.$nextTick().then(() => {
-        expect(button.text()).toEqual('Restart Terminal');
-        expect(actions.restartSession).toHaveBeenCalled();
-      });
+      await nextTick();
+      expect(button.text()).toEqual('Restart Terminal');
+      expect(actions.restartSession).toHaveBeenCalled();
     });
   });
 });