1 files changed, 71 insertions, 9 deletions

diff --git a/spec/frontend/reports/codequality_report/store/actions_spec.js b/spec/frontend/reports/codequality_report/store/actions_spec.js
index 9dda024bffd..1821390786b 100644
--- a/spec/frontend/reports/codequality_report/store/actions_spec.js
+++ b/spec/frontend/reports/codequality_report/store/actions_spec.js
@@ -5,8 +5,14 @@ import axios from '~/lib/utils/axios_utils';
 import createStore from '~/reports/codequality_report/store';
 import * as actions from '~/reports/codequality_report/store/actions';
 import * as types from '~/reports/codequality_report/store/mutation_types';
+import { STATUS_NOT_FOUND } from '~/reports/constants';
 import { reportIssues, parsedReportIssues } from '../mock_data';
 
+const pollInterval = 123;
+const pollIntervalHeader = {
+  'Poll-Interval': pollInterval,
+};
+
 describe('Codequality Reports actions', () => {
   let localState;
   let localStore;
@@ -19,8 +25,6 @@ describe('Codequality Reports actions', () => {
   describe('setPaths', () => {
     it('should commit SET_PATHS mutation', (done) => {
       const paths = {
-        basePath: 'basePath',
-        headPath: 'headPath',
         baseBlobPath: 'baseBlobPath',
         headBlobPath: 'headBlobPath',
         reportsPath: 'reportsPath',
@@ -39,11 +43,11 @@ describe('Codequality Reports actions', () => {
   });
 
   describe('fetchReports', () => {
+    const endpoint = `${TEST_HOST}/codequality_reports.json`;
     let mock;
 
     beforeEach(() => {
-      localState.reportsPath = `${TEST_HOST}/codequality_reports.json`;
-      localState.basePath = '/base/path';
+      localState.reportsPath = endpoint;
       mock = new MockAdapter(axios);
     });
 
@@ -53,7 +57,7 @@ describe('Codequality Reports actions', () => {
 
     describe('on success', () => {
       it('commits REQUEST_REPORTS and dispatches receiveReportsSuccess', (done) => {
-        mock.onGet(`${TEST_HOST}/codequality_reports.json`).reply(200, reportIssues);
+        mock.onGet(endpoint).reply(200, reportIssues);
 
         testAction(
           actions.fetchReports,
@@ -73,7 +77,7 @@ describe('Codequality Reports actions', () => {
 
     describe('on error', () => {
       it('commits REQUEST_REPORTS and dispatches receiveReportsError', (done) => {
-        mock.onGet(`${TEST_HOST}/codequality_reports.json`).reply(500);
+        mock.onGet(endpoint).reply(500);
 
         testAction(
           actions.fetchReports,
@@ -86,20 +90,78 @@ describe('Codequality Reports actions', () => {
       });
     });
 
-    describe('with no base path', () => {
+    describe('when base report is not found', () => {
       it('commits REQUEST_REPORTS and dispatches receiveReportsError', (done) => {
-        localState.basePath = null;
+        const data = { status: STATUS_NOT_FOUND };
+        mock.onGet(`${TEST_HOST}/codequality_reports.json`).reply(200, data);
 
         testAction(
           actions.fetchReports,
           null,
           localState,
           [{ type: types.REQUEST_REPORTS }],
-          [{ type: 'receiveReportsError' }],
+          [{ type: 'receiveReportsError', payload: data }],
           done,
         );
       });
     });
+
+    describe('while waiting for report results', () => {
+      it('continues polling until it receives data', (done) => {
+        mock
+          .onGet(endpoint)
+          .replyOnce(204, undefined, pollIntervalHeader)
+          .onGet(endpoint)
+          .reply(200, reportIssues);
+
+        Promise.all([
+          testAction(
+            actions.fetchReports,
+            null,
+            localState,
+            [{ type: types.REQUEST_REPORTS }],
+            [
+              {
+                payload: parsedReportIssues,
+                type: 'receiveReportsSuccess',
+              },
+            ],
+            done,
+          ),
+          axios
+            // wait for initial NO_CONTENT response to be fulfilled
+            .waitForAll()
+            .then(() => {
+              jest.advanceTimersByTime(pollInterval);
+            }),
+        ]).catch(done.fail);
+      });
+
+      it('continues polling until it receives an error', (done) => {
+        mock
+          .onGet(endpoint)
+          .replyOnce(204, undefined, pollIntervalHeader)
+          .onGet(endpoint)
+          .reply(500);
+
+        Promise.all([
+          testAction(
+            actions.fetchReports,
+            null,
+            localState,
+            [{ type: types.REQUEST_REPORTS }],
+            [{ type: 'receiveReportsError', payload: expect.any(Error) }],
+            done,
+          ),
+          axios
+            // wait for initial NO_CONTENT response to be fulfilled
+            .waitForAll()
+            .then(() => {
+              jest.advanceTimersByTime(pollInterval);
+            }),
+        ]).catch(done.fail);
+      });
+    });
   });
 
   describe('receiveReportsSuccess', () => {