1 files changed, 5 insertions, 8 deletions
diff --git a/spec/frontend/diffs/components/diff_discussions_spec.js b/spec/frontend/diffs/components/diff_discussions_spec.js
index bd6f4cd2545..2da68adddf6 100644
--- a/spec/frontend/diffs/components/diff_discussions_spec.js
+++ b/spec/frontend/diffs/components/diff_discussions_spec.js
@@ -1,5 +1,5 @@
 import { GlIcon } from '@gitlab/ui';
-import { mount, createLocalVue } from '@vue/test-utils';
+import { mount } from '@vue/test-utils';
 import DiffDiscussions from '~/diffs/components/diff_discussions.vue';
 import { createStore } from '~/mr_notes/stores';
 import DiscussionNotes from '~/notes/components/discussion_notes.vue';
@@ -8,8 +8,6 @@ import TimelineEntryItem from '~/vue_shared/components/notes/timeline_entry_item
 import '~/behaviors/markdown/render_gfm';
 import discussionsMockData from '../mock_data/diff_discussions';
 
-const localVue = createLocalVue();
-
 describe('DiffDiscussions', () => {
   let store;
   let wrapper;
@@ -17,13 +15,12 @@ describe('DiffDiscussions', () => {
 
   const createComponent = (props) => {
     store = createStore();
-    wrapper = mount(localVue.extend(DiffDiscussions), {
+    wrapper = mount(DiffDiscussions, {
       store,
       propsData: {
         discussions: getDiscussionsMockData(),
         ...props,
       },
-      localVue,
     });
   };
 
@@ -74,7 +71,7 @@ describe('DiffDiscussions', () => {
 
       expect(diffNotesToggle.text().trim()).toBe('1');
       expect(diffNotesToggle.classes()).toEqual(
-        expect.arrayContaining(['btn-transparent', 'badge', 'badge-pill']),
+        expect.arrayContaining(['js-diff-notes-toggle', 'gl-translate-x-n50', 'design-note-pin']),
       );
     });
 
@@ -90,8 +87,8 @@ describe('DiffDiscussions', () => {
       createComponent({ renderAvatarBadge: true });
       const noteableDiscussion = wrapper.find(NoteableDiscussion);
 
-      expect(noteableDiscussion.find('.badge-pill').exists()).toBe(true);
-      expect(noteableDiscussion.find('.badge-pill').text().trim()).toBe('1');
+      expect(noteableDiscussion.find('.design-note-pin').exists()).toBe(true);
+      expect(noteableDiscussion.find('.design-note-pin').text().trim()).toBe('1');
     });
   });
 });