1 files changed, 126 insertions, 0 deletions
diff --git a/spec/frontend/jobs/components/table/cells/actions_cell_spec.js b/spec/frontend/jobs/components/table/cells/actions_cell_spec.js
new file mode 100644
index 00000000000..1b1e2d4df8f
--- /dev/null
+++ b/spec/frontend/jobs/components/table/cells/actions_cell_spec.js
@@ -0,0 +1,126 @@
+import { GlModal } from '@gitlab/ui';
+import { nextTick } from 'vue';
+import { shallowMountExtended } from 'helpers/vue_test_utils_helper';
+import ActionsCell from '~/jobs/components/table/cells/actions_cell.vue';
+import JobPlayMutation from '~/jobs/components/table/graphql/mutations/job_play.mutation.graphql';
+import JobRetryMutation from '~/jobs/components/table/graphql/mutations/job_retry.mutation.graphql';
+import JobUnscheduleMutation from '~/jobs/components/table/graphql/mutations/job_unschedule.mutation.graphql';
+import { playableJob, retryableJob, scheduledJob } from '../../../mock_data';
+
+describe('Job actions cell', () => {
+  let wrapper;
+  let mutate;
+
+  const findRetryButton = () => wrapper.findByTestId('retry');
+  const findPlayButton = () => wrapper.findByTestId('play');
+  const findDownloadArtifactsButton = () => wrapper.findByTestId('download-artifacts');
+  const findCountdownButton = () => wrapper.findByTestId('countdown');
+  const findPlayScheduledJobButton = () => wrapper.findByTestId('play-scheduled');
+  const findUnscheduleButton = () => wrapper.findByTestId('unschedule');
+
+  const findModal = () => wrapper.findComponent(GlModal);
+
+  const MUTATION_SUCCESS = { data: { JobRetryMutation: { jobId: retryableJob.id } } };
+  const MUTATION_SUCCESS_UNSCHEDULE = {
+    data: { JobUnscheduleMutation: { jobId: scheduledJob.id } },
+  };
+  const MUTATION_SUCCESS_PLAY = { data: { JobPlayMutation: { jobId: playableJob.id } } };
+
+  const $toast = {
+    show: jest.fn(),
+  };
+
+  const createComponent = (jobType, mutationType = MUTATION_SUCCESS, props = {}) => {
+    mutate = jest.fn().mockResolvedValue(mutationType);
+
+    wrapper = shallowMountExtended(ActionsCell, {
+      propsData: {
+        job: jobType,
+        ...props,
+      },
+      mocks: {
+        $apollo: {
+          mutate,
+        },
+        $toast,
+      },
+    });
+  };
+
+  afterEach(() => {
+    wrapper.destroy();
+  });
+
+  it('does not display an artifacts download button', () => {
+    createComponent(retryableJob);
+
+    expect(findDownloadArtifactsButton().exists()).toBe(false);
+  });
+
+  it.each`
+    button                         | action                  | jobType
+    ${findPlayButton}              | ${'play'}               | ${playableJob}
+    ${findRetryButton}             | ${'retry'}              | ${retryableJob}
+    ${findDownloadArtifactsButton} | ${'download artifacts'} | ${playableJob}
+  `('displays the $action button', ({ button, jobType }) => {
+    createComponent(jobType);
+
+    expect(button().exists()).toBe(true);
+  });
+
+  it.each`
+    button             | mutationResult           | action     | jobType         | mutationFile
+    ${findPlayButton}  | ${MUTATION_SUCCESS_PLAY} | ${'play'}  | ${playableJob}  | ${JobPlayMutation}
+    ${findRetryButton} | ${MUTATION_SUCCESS}      | ${'retry'} | ${retryableJob} | ${JobRetryMutation}
+  `('performs the $action mutation', ({ button, mutationResult, jobType, mutationFile }) => {
+    createComponent(jobType, mutationResult);
+
+    button().vm.$emit('click');
+
+    expect(mutate).toHaveBeenCalledWith({
+      mutation: mutationFile,
+      variables: {
+        id: jobType.id,
+      },
+    });
+  });
+
+  describe('Scheduled Jobs', () => {
+    const today = () => new Date('2021-08-31');
+
+    beforeEach(() => {
+      jest.spyOn(Date, 'now').mockImplementation(today);
+    });
+
+    it('displays the countdown, play and unschedule buttons', () => {
+      createComponent(scheduledJob);
+
+      expect(findCountdownButton().exists()).toBe(true);
+      expect(findPlayScheduledJobButton().exists()).toBe(true);
+      expect(findUnscheduleButton().exists()).toBe(true);
+    });
+
+    it('unschedules a job', () => {
+      createComponent(scheduledJob, MUTATION_SUCCESS_UNSCHEDULE);
+
+      findUnscheduleButton().vm.$emit('click');
+
+      expect(mutate).toHaveBeenCalledWith({
+        mutation: JobUnscheduleMutation,
+        variables: {
+          id: scheduledJob.id,
+        },
+      });
+    });
+
+    it('shows the play job confirmation modal', async () => {
+      createComponent(scheduledJob, MUTATION_SUCCESS);
+
+      findPlayScheduledJobButton().vm.$emit('click');
+
+      await nextTick();
+
+      expect(findModal().exists()).toBe(true);
+    });
+  });
+});