1 files changed, 9 insertions, 8 deletions
diff --git a/spec/frontend/batch_comments/components/publish_dropdown_spec.js b/spec/frontend/batch_comments/components/publish_dropdown_spec.js
index e89934c0192..44d7b56c14f 100644
--- a/spec/frontend/batch_comments/components/publish_dropdown_spec.js
+++ b/spec/frontend/batch_comments/components/publish_dropdown_spec.js
@@ -1,4 +1,4 @@
-import { GlDropdown, GlDropdownItem } from '@gitlab/ui';
+import { GlDisclosureDropdown } from '@gitlab/ui';
 import { shallowMount } from '@vue/test-utils';
 import Vue from 'vue';
 import Vuex from 'vuex';
@@ -12,29 +12,30 @@ Vue.use(Vuex);
 
 describe('Batch comments publish dropdown component', () => {
   let wrapper;
+  const draft = createDraft();
 
   function createComponent() {
     const store = createStore();
-    store.state.batchComments.drafts.push(createDraft(), { ...createDraft(), id: 2 });
+    store.state.batchComments.drafts.push(draft, { ...draft, id: 2 });
 
     wrapper = shallowMount(PreviewDropdown, {
       store,
+      stubs: { GlDisclosureDropdown },
     });
   }
 
-  afterEach(() => {
-    wrapper.destroy();
-  });
-
   it('renders list of drafts', () => {
     createComponent();
 
-    expect(wrapper.findAllComponents(GlDropdownItem).length).toBe(2);
+    expect(wrapper.findComponent(GlDisclosureDropdown).props('items')).toMatchObject([
+      draft,
+      { ...draft, id: 2 },
+    ]);
   });
 
   it('renders draft count in dropdown title', () => {
     createComponent();
 
-    expect(wrapper.findComponent(GlDropdown).props('headerText')).toEqual('2 pending comments');
+    expect(wrapper.findComponent(GlDisclosureDropdown).text()).toEqual('2 pending comments');
   });
 });