Labels widget - Fix search input

Review feedback

spec/frontend/vue_shared/components/sidebar/labels_select_widget/dropdown_contents_spec.js

@@ -130,14 +130,14 @@ describe('DropdownContent', () => {
     expect(findDropdownHeader().exists()).toBe(true);
   });
 
-  it('sets searchKey on input event from header', async () => {
+  it('sets searchKey for labels view on input event from header', async () => {
     createComponent();
 
     expect(wrapper.vm.searchKey).toEqual('');
     findDropdownHeader().vm.$emit('input', '123');
     await nextTick();
 
-    expect(wrapper.vm.searchKey).toEqual('123');
+    expect(findLabelsView().props('searchKey')).toEqual('123');
   });
 
   describe('Create view', () => {

spec/frontend/vue_shared/components/sidebar/labels_select_widget/dropdown_header_spec.js

 import { GlSearchBoxByType } from '@gitlab/ui';
 import { shallowMount } from '@vue/test-utils';
+import { extendedWrapper } from 'helpers/vue_test_utils_helper';
 import DropdownHeader from '~/vue_shared/components/sidebar/labels_select_widget/dropdown_header.vue';
 
 describe('DropdownHeader', () => {
@@ -9,7 +10,8 @@ describe('DropdownHeader', () => {
     showDropdownContentsCreateView = false,
     labelsFetchInProgress = false,
   } = {}) => {
-    wrapper = shallowMount(DropdownHeader, {
+    wrapper = extendedWrapper(
+      shallowMount(DropdownHeader, {
         propsData: {
           showDropdownContentsCreateView,
           labelsFetchInProgress,
@@ -20,7 +22,8 @@ describe('DropdownHeader', () => {
         stubs: {
           GlSearchBoxByType,
         },
-    });
+      }),
+    );
   };
 
   afterEach(() => {
@@ -28,7 +31,7 @@ describe('DropdownHeader', () => {
   });
 
   const findSearchInput = () => wrapper.findComponent(GlSearchBoxByType);
-  const findGoBackButton = () => wrapper.find('[data-testid="go-back-button"]');
+  const findGoBackButton = () => wrapper.findByTestId('go-back-button');
 
   beforeEach(() => {
     createComponent();
@@ -57,13 +60,16 @@ describe('DropdownHeader', () => {
       expect(findGoBackButton().exists()).toBe(false);
     });
 
-    it('when loading labels - renders disabled search input field', async () => {
-      createComponent({ labelsFetchInProgress: true });
-      expect(findSearchInput().props('disabled')).toBe(true);
-    });
-
-    it('when labels are loaded - renders enabled search input field', async () => {
-      expect(findSearchInput().props('disabled')).toBe(false);
-    });
+    it.each`
+      labelsFetchInProgress | disabled
+      ${true}               | ${true}
+      ${false}              | ${false}
+    `(
+      'when labelsFetchInProgress is $labelsFetchInProgress, renders search input with disabled prop to $disabled',
+      ({ labelsFetchInProgress, disabled }) => {
+        createComponent({ labelsFetchInProgress });
+        expect(findSearchInput().props('disabled')).toBe(disabled);
+      },
+    );
   });
 });