1 files changed, 78 insertions, 9 deletions

diff --git a/spec/frontend/work_items/components/work_item_links/work_item_links_form_spec.js b/spec/frontend/work_items/components/work_item_links/work_item_links_form_spec.js
index 93bf7286aa7..434c1db8a2c 100644
--- a/spec/frontend/work_items/components/work_item_links/work_item_links_form_spec.js
+++ b/spec/frontend/work_items/components/work_item_links/work_item_links_form_spec.js
@@ -1,13 +1,20 @@
 import Vue from 'vue';
-import { GlForm, GlFormCombobox } from '@gitlab/ui';
+import { GlForm, GlFormInput, GlFormCombobox } from '@gitlab/ui';
 import VueApollo from 'vue-apollo';
 import { shallowMountExtended } from 'helpers/vue_test_utils_helper';
 import createMockApollo from 'helpers/mock_apollo_helper';
 import waitForPromises from 'helpers/wait_for_promises';
 import WorkItemLinksForm from '~/work_items/components/work_item_links/work_item_links_form.vue';
 import projectWorkItemsQuery from '~/work_items/graphql/project_work_items.query.graphql';
+import projectWorkItemTypesQuery from '~/work_items/graphql/project_work_item_types.query.graphql';
+import createWorkItemMutation from '~/work_items/graphql/create_work_item.mutation.graphql';
 import updateWorkItemMutation from '~/work_items/graphql/update_work_item.mutation.graphql';
-import { availableWorkItemsResponse, updateWorkItemMutationResponse } from '../../mock_data';
+import {
+  availableWorkItemsResponse,
+  projectWorkItemTypesQueryResponse,
+  createWorkItemMutationResponse,
+  updateWorkItemMutationResponse,
+} from '../../mock_data';
 
 Vue.use(VueApollo);
 
@@ -15,14 +22,21 @@ describe('WorkItemLinksForm', () => {
   let wrapper;
 
   const updateMutationResolver = jest.fn().mockResolvedValue(updateWorkItemMutationResponse);
+  const createMutationResolver = jest.fn().mockResolvedValue(createWorkItemMutationResponse);
 
-  const createComponent = async ({ listResponse = availableWorkItemsResponse } = {}) => {
+  const createComponent = async ({
+    listResponse = availableWorkItemsResponse,
+    typesResponse = projectWorkItemTypesQueryResponse,
+    parentConfidential = false,
+  } = {}) => {
     wrapper = shallowMountExtended(WorkItemLinksForm, {
       apolloProvider: createMockApollo([
         [projectWorkItemsQuery, jest.fn().mockResolvedValue(listResponse)],
+        [projectWorkItemTypesQuery, jest.fn().mockResolvedValue(typesResponse)],
         [updateWorkItemMutation, updateMutationResolver],
+        [createWorkItemMutation, createMutationResolver],
       ]),
-      propsData: { issuableGid: 'gid://gitlab/WorkItem/1' },
+      propsData: { issuableGid: 'gid://gitlab/WorkItem/1', parentConfidential },
       provide: {
         projectPath: 'project/path',
       },
@@ -33,6 +47,7 @@ describe('WorkItemLinksForm', () => {
 
   const findForm = () => wrapper.findComponent(GlForm);
   const findCombobox = () => wrapper.findComponent(GlFormCombobox);
+  const findInput = () => wrapper.findComponent(GlFormInput);
   const findAddChildButton = () => wrapper.findByTestId('add-child-button');
 
   beforeEach(async () => {
@@ -47,19 +62,73 @@ describe('WorkItemLinksForm', () => {
     expect(findForm().exists()).toBe(true);
   });
 
-  it('passes available work items as prop when typing in combobox', async () => {
-    findCombobox().vm.$emit('input', 'Task');
+  it('creates child task in non confidential parent', async () => {
+    findInput().vm.$emit('input', 'Create task test');
+
+    findForm().vm.$emit('submit', {
+      preventDefault: jest.fn(),
+    });
     await waitForPromises();
+    expect(createMutationResolver).toHaveBeenCalledWith({
+      input: {
+        title: 'Create task test',
+        projectPath: 'project/path',
+        workItemTypeId: 'gid://gitlab/WorkItems::Type/3',
+        hierarchyWidget: {
+          parentId: 'gid://gitlab/WorkItem/1',
+        },
+        confidential: false,
+      },
+    });
+  });
+
+  it('creates child task in confidential parent', async () => {
+    await createComponent({ parentConfidential: true });
+
+    findInput().vm.$emit('input', 'Create confidential task');
 
-    expect(findCombobox().exists()).toBe(true);
-    expect(findCombobox().props('tokenList').length).toBe(2);
+    findForm().vm.$emit('submit', {
+      preventDefault: jest.fn(),
+    });
+    await waitForPromises();
+    expect(createMutationResolver).toHaveBeenCalledWith({
+      input: {
+        title: 'Create confidential task',
+        projectPath: 'project/path',
+        workItemTypeId: 'gid://gitlab/WorkItems::Type/3',
+        hierarchyWidget: {
+          parentId: 'gid://gitlab/WorkItem/1',
+        },
+        confidential: true,
+      },
+    });
   });
 
-  it('selects and add child', async () => {
+  // Follow up issue to turn this functionality back on https://gitlab.com/gitlab-org/gitlab/-/issues/368757
+  // eslint-disable-next-line jest/no-disabled-tests
+  it.skip('selects and add child', async () => {
     findCombobox().vm.$emit('input', availableWorkItemsResponse.data.workspace.workItems.edges[0]);
 
     findAddChildButton().vm.$emit('click');
     await waitForPromises();
     expect(updateMutationResolver).toHaveBeenCalled();
   });
+
+  // eslint-disable-next-line jest/no-disabled-tests
+  describe.skip('when typing in combobox', () => {
+    beforeEach(async () => {
+      findCombobox().vm.$emit('input', 'Task');
+      await waitForPromises();
+      await jest.runOnlyPendingTimers();
+    });
+
+    it('passes available work items as prop', () => {
+      expect(findCombobox().exists()).toBe(true);
+      expect(findCombobox().props('tokenList').length).toBe(2);
+    });
+
+    it('passes action to create task', () => {
+      expect(findCombobox().props('actionList').length).toBe(1);
+    });
+  });
 });