1 files changed, 118 insertions, 0 deletions
diff --git a/spec/frontend/ide/stores/modules/terminal_sync/actions_spec.js b/spec/frontend/ide/stores/modules/terminal_sync/actions_spec.js
new file mode 100644
index 00000000000..ac976300ed0
--- /dev/null
+++ b/spec/frontend/ide/stores/modules/terminal_sync/actions_spec.js
@@ -0,0 +1,118 @@
+import * as actions from '~/ide/stores/modules/terminal_sync/actions';
+import mirror, { canConnect, SERVICE_NAME } from '~/ide/lib/mirror';
+import * as types from '~/ide/stores/modules/terminal_sync/mutation_types';
+import testAction from 'helpers/vuex_action_helper';
+
+jest.mock('~/ide/lib/mirror');
+
+const TEST_SESSION = {
+  proxyWebsocketPath: 'test/path',
+  services: [SERVICE_NAME],
+};
+
+describe('ide/stores/modules/terminal_sync/actions', () => {
+  let rootState;
+
+  beforeEach(() => {
+    canConnect.mockReturnValue(true);
+    rootState = {
+      changedFiles: [],
+      terminal: {},
+    };
+  });
+
+  describe('upload', () => {
+    it('uploads to mirror and sets success', done => {
+      mirror.upload.mockReturnValue(Promise.resolve());
+
+      testAction(
+        actions.upload,
+        null,
+        rootState,
+        [{ type: types.START_LOADING }, { type: types.SET_SUCCESS }],
+        [],
+        () => {
+          expect(mirror.upload).toHaveBeenCalledWith(rootState);
+          done();
+        },
+      );
+    });
+
+    it('sets error when failed', done => {
+      const err = { message: 'it failed!' };
+      mirror.upload.mockReturnValue(Promise.reject(err));
+
+      testAction(
+        actions.upload,
+        null,
+        rootState,
+        [{ type: types.START_LOADING }, { type: types.SET_ERROR, payload: err }],
+        [],
+        done,
+      );
+    });
+  });
+
+  describe('stop', () => {
+    it('disconnects from mirror', done => {
+      testAction(actions.stop, null, rootState, [{ type: types.STOP }], [], () => {
+        expect(mirror.disconnect).toHaveBeenCalled();
+        done();
+      });
+    });
+  });
+
+  describe('start', () => {
+    it.each`
+      session                                | canConnectMock | description
+      ${null}                                | ${true}        | ${'does not exist'}
+      ${{}}                                  | ${true}        | ${'does not have proxyWebsocketPath'}
+      ${{ proxyWebsocketPath: 'test/path' }} | ${false}       | ${'can not connect service'}
+    `('rejects if session $description', ({ session, canConnectMock }) => {
+      canConnect.mockReturnValue(canConnectMock);
+
+      const result = actions.start({ rootState: { terminal: { session } } });
+
+      return expect(result).rejects.toBe(undefined);
+    });
+
+    describe('with terminal session in state', () => {
+      beforeEach(() => {
+        rootState = {
+          terminal: { session: TEST_SESSION },
+        };
+      });
+
+      it('connects to mirror and sets success', done => {
+        mirror.connect.mockReturnValue(Promise.resolve());
+
+        testAction(
+          actions.start,
+          null,
+          rootState,
+          [{ type: types.START_LOADING }, { type: types.SET_SUCCESS }],
+          [],
+          () => {
+            expect(mirror.connect).toHaveBeenCalledWith(TEST_SESSION.proxyWebsocketPath);
+            done();
+          },
+        );
+      });
+
+      it('sets error if connection fails', () => {
+        const commit = jest.fn();
+        const err = new Error('test');
+        mirror.connect.mockReturnValue(Promise.reject(err));
+
+        const result = actions.start({ rootState, commit });
+
+        return Promise.all([
+          expect(result).rejects.toEqual(err),
+          result.catch(() => {
+            expect(commit).toHaveBeenCalledWith(types.SET_ERROR, err);
+          }),
+        ]);
+      });
+    });
+  });
+});