1 files changed, 169 insertions, 0 deletions
diff --git a/spec/frontend/ide/stores/modules/terminal/actions/session_status_spec.js b/spec/frontend/ide/stores/modules/terminal/actions/session_status_spec.js
new file mode 100644
index 00000000000..7909f828124
--- /dev/null
+++ b/spec/frontend/ide/stores/modules/terminal/actions/session_status_spec.js
@@ -0,0 +1,169 @@
+import MockAdapter from 'axios-mock-adapter';
+import testAction from 'helpers/vuex_action_helper';
+import { PENDING, RUNNING, STOPPING, STOPPED } from '~/ide/stores/modules/terminal/constants';
+import * as messages from '~/ide/stores/modules/terminal/messages';
+import * as mutationTypes from '~/ide/stores/modules/terminal/mutation_types';
+import * as actions from '~/ide/stores/modules/terminal/actions/session_status';
+import axios from '~/lib/utils/axios_utils';
+import createFlash from '~/flash';
+
+jest.mock('~/flash');
+
+const TEST_SESSION = {
+  id: 7,
+  status: PENDING,
+  show_path: 'path/show',
+  cancel_path: 'path/cancel',
+  retry_path: 'path/retry',
+  terminal_path: 'path/terminal',
+};
+
+describe('IDE store terminal session controls actions', () => {
+  let mock;
+  let dispatch;
+  let commit;
+
+  beforeEach(() => {
+    mock = new MockAdapter(axios);
+    dispatch = jest.fn().mockName('dispatch');
+    commit = jest.fn().mockName('commit');
+  });
+
+  afterEach(() => {
+    mock.restore();
+  });
+
+  describe('pollSessionStatus', () => {
+    it('starts interval to poll status', () => {
+      return testAction(
+        actions.pollSessionStatus,
+        null,
+        {},
+        [{ type: mutationTypes.SET_SESSION_STATUS_INTERVAL, payload: expect.any(Number) }],
+        [{ type: 'stopPollingSessionStatus' }, { type: 'fetchSessionStatus' }],
+      );
+    });
+
+    it('on interval, stops polling if no session', () => {
+      const state = {
+        session: null,
+      };
+
+      actions.pollSessionStatus({ state, dispatch, commit });
+      dispatch.mockClear();
+
+      jest.advanceTimersByTime(5001);
+
+      expect(dispatch).toHaveBeenCalledWith('stopPollingSessionStatus');
+    });
+
+    it('on interval, fetches status', () => {
+      const state = {
+        session: TEST_SESSION,
+      };
+
+      actions.pollSessionStatus({ state, dispatch, commit });
+      dispatch.mockClear();
+
+      jest.advanceTimersByTime(5001);
+
+      expect(dispatch).toHaveBeenCalledWith('fetchSessionStatus');
+    });
+  });
+
+  describe('stopPollingSessionStatus', () => {
+    it('does nothing if sessionStatusInterval is empty', () => {
+      return testAction(actions.stopPollingSessionStatus, null, {}, [], []);
+    });
+
+    it('clears interval', () => {
+      return testAction(
+        actions.stopPollingSessionStatus,
+        null,
+        { sessionStatusInterval: 7 },
+        [{ type: mutationTypes.SET_SESSION_STATUS_INTERVAL, payload: 0 }],
+        [],
+      );
+    });
+  });
+
+  describe('receiveSessionStatusSuccess', () => {
+    it('sets session status', () => {
+      return testAction(
+        actions.receiveSessionStatusSuccess,
+        { status: RUNNING },
+        {},
+        [{ type: mutationTypes.SET_SESSION_STATUS, payload: RUNNING }],
+        [],
+      );
+    });
+
+    [STOPPING, STOPPED, 'unexpected'].forEach(status => {
+      it(`kills session if status is ${status}`, () => {
+        return testAction(
+          actions.receiveSessionStatusSuccess,
+          { status },
+          {},
+          [{ type: mutationTypes.SET_SESSION_STATUS, payload: status }],
+          [{ type: 'killSession' }],
+        );
+      });
+    });
+  });
+
+  describe('receiveSessionStatusError', () => {
+    it('flashes message', () => {
+      actions.receiveSessionStatusError({ dispatch });
+
+      expect(createFlash).toHaveBeenCalledWith(messages.UNEXPECTED_ERROR_STATUS);
+    });
+
+    it('kills the session', () => {
+      return testAction(actions.receiveSessionStatusError, null, {}, [], [{ type: 'killSession' }]);
+    });
+  });
+
+  describe('fetchSessionStatus', () => {
+    let state;
+
+    beforeEach(() => {
+      state = {
+        session: {
+          showPath: TEST_SESSION.show_path,
+        },
+      };
+    });
+
+    it('does nothing if session is falsey', () => {
+      state.session = null;
+
+      actions.fetchSessionStatus({ dispatch, state });
+
+      expect(dispatch).not.toHaveBeenCalled();
+    });
+
+    it('dispatches success on success', () => {
+      mock.onGet(state.session.showPath).reply(200, TEST_SESSION);
+
+      return testAction(
+        actions.fetchSessionStatus,
+        null,
+        state,
+        [],
+        [{ type: 'receiveSessionStatusSuccess', payload: TEST_SESSION }],
+      );
+    });
+
+    it('dispatches error on error', () => {
+      mock.onGet(state.session.showPath).reply(400);
+
+      return testAction(
+        actions.fetchSessionStatus,
+        null,
+        state,
+        [],
+        [{ type: 'receiveSessionStatusError', payload: expect.any(Error) }],
+      );
+    });
+  });
+});