1 files changed, 25 insertions, 84 deletions
diff --git a/spec/frontend/vue_merge_request_widget/components/states/mr_widget_wip_spec.js b/spec/frontend/vue_merge_request_widget/components/states/mr_widget_wip_spec.js
index 7259f210b6e..82aeac1a47d 100644
--- a/spec/frontend/vue_merge_request_widget/components/states/mr_widget_wip_spec.js
+++ b/spec/frontend/vue_merge_request_widget/components/states/mr_widget_wip_spec.js
@@ -1,101 +1,42 @@
-import Vue, { nextTick } from 'vue';
-import waitForPromises from 'helpers/wait_for_promises';
+import { mount } from '@vue/test-utils';
 import WorkInProgress from '~/vue_merge_request_widget/components/states/work_in_progress.vue';
-import toast from '~/vue_shared/plugins/global_toast';
-import eventHub from '~/vue_merge_request_widget/event_hub';
 
-jest.mock('~/vue_shared/plugins/global_toast');
-
-const createComponent = () => {
-  const Component = Vue.extend(WorkInProgress);
-  const mr = {
-    title: 'The best MR ever',
-    removeWIPPath: '/path/to/remove/wip',
-  };
-  const service = {
-    removeWIP() {},
-  };
-  return new Component({
-    el: document.createElement('div'),
-    propsData: { mr, service },
+let wrapper;
+
+const createComponent = (updateMergeRequest = true) => {
+  wrapper = mount(WorkInProgress, {
+    propsData: {
+      mr: {},
+    },
+    data() {
+      return {
+        userPermissions: {
+          updateMergeRequest,
+        },
+      };
+    },
   });
 };
 
-describe('Wip', () => {
-  describe('props', () => {
-    it('should have props', () => {
-      const { mr, service } = WorkInProgress.props;
-
-      expect(mr.type instanceof Object).toBe(true);
-      expect(mr.required).toBe(true);
-
-      expect(service.type instanceof Object).toBe(true);
-      expect(service.required).toBe(true);
-    });
-  });
-
-  describe('data', () => {
-    it('should have default data', () => {
-      const vm = createComponent();
-
-      expect(vm.isMakingRequest).toBe(false);
-    });
-  });
-
-  describe('methods', () => {
-    const mrObj = {
-      is_new_mr_data: true,
-    };
-
-    describe('handleRemoveDraft', () => {
-      it('should make a request to service and handle response', async () => {
-        const vm = createComponent();
-
-        jest.spyOn(eventHub, '$emit').mockImplementation(() => {});
-        jest.spyOn(vm.service, 'removeWIP').mockReturnValue(
-          new Promise((resolve) => {
-            resolve({
-              data: mrObj,
-            });
-          }),
-        );
-
-        vm.handleRemoveDraft();
-
-        await waitForPromises();
-
-        expect(vm.isMakingRequest).toBe(true);
-        expect(eventHub.$emit).toHaveBeenCalledWith('UpdateWidgetData', mrObj);
-        expect(toast).toHaveBeenCalledWith('Marked as ready. Merging is now allowed.');
-      });
-    });
+describe('Merge request widget draft state component', () => {
+  afterEach(() => {
+    wrapper.destroy();
   });
 
   describe('template', () => {
-    let vm;
-    let el;
-
-    beforeEach(() => {
-      vm = createComponent();
-      el = vm.$el;
-    });
-
     it('should have correct elements', () => {
-      expect(el.classList.contains('mr-widget-body')).toBe(true);
-      expect(el.innerText).toContain(
+      createComponent(true);
+
+      expect(wrapper.text()).toContain(
         "Merge blocked: merge request must be marked as ready. It's still marked as draft.",
       );
-      expect(el.querySelector('.js-remove-draft').innerText.replace(/\s\s+/g, ' ')).toContain(
-        'Mark as ready',
-      );
+      expect(wrapper.find('[data-testid="removeWipButton"]').text()).toContain('Mark as ready');
     });
 
-    it('should not show removeWIP button is user cannot update MR', async () => {
-      vm.mr.removeWIPPath = '';
-
-      await nextTick();
+    it('should not show removeWIP button is user cannot update MR', () => {
+      createComponent(false);
 
-      expect(el.querySelector('.js-remove-draft')).toBeNull();
+      expect(wrapper.find('[data-testid="removeWipButton"]').exists()).toBe(false);
     });
   });
 });