1 files changed, 8 insertions, 3 deletions
diff --git a/spec/frontend/ref/components/ref_selector_spec.js b/spec/frontend/ref/components/ref_selector_spec.js
index 96601a729b2..4997c13bbb2 100644
--- a/spec/frontend/ref/components/ref_selector_spec.js
+++ b/spec/frontend/ref/components/ref_selector_spec.js
@@ -18,6 +18,8 @@ import {
   REF_TYPE_BRANCHES,
   REF_TYPE_TAGS,
   REF_TYPE_COMMITS,
+  BRANCH_REF_TYPE,
+  TAG_REF_TYPE,
 } from '~/ref/constants';
 import createStore from '~/ref/stores/';
 
@@ -34,7 +36,7 @@ describe('Ref selector component', () => {
   let commitApiCallSpy;
   let requestSpies;
 
-  const createComponent = (mountOverrides = {}) => {
+  const createComponent = (mountOverrides = {}, propsData = {}) => {
     wrapper = mount(
       RefSelector,
       merge(
@@ -42,6 +44,7 @@ describe('Ref selector component', () => {
           propsData: {
             projectId,
             value: '',
+            ...propsData,
           },
           listeners: {
             // simulate a parent component v-model binding
@@ -338,13 +341,14 @@ describe('Ref selector component', () => {
     describe('branches', () => {
       describe('when the branches search returns results', () => {
         beforeEach(() => {
-          createComponent();
+          createComponent({}, { refType: BRANCH_REF_TYPE, useSymbolicRefNames: true });
 
           return waitForRequests();
         });
 
         it('renders the branches section in the dropdown', () => {
           expect(findBranchesSection().exists()).toBe(true);
+          expect(findBranchesSection().props('shouldShowCheck')).toBe(true);
         });
 
         it('renders the "Branches" heading with a total number indicator', () => {
@@ -415,13 +419,14 @@ describe('Ref selector component', () => {
     describe('tags', () => {
       describe('when the tags search returns results', () => {
         beforeEach(() => {
-          createComponent();
+          createComponent({}, { refType: TAG_REF_TYPE, useSymbolicRefNames: true });
 
           return waitForRequests();
         });
 
         it('renders the tags section in the dropdown', () => {
           expect(findTagsSection().exists()).toBe(true);
+          expect(findTagsSection().props('shouldShowCheck')).toBe(true);
         });
 
         it('renders the "Tags" heading with a total number indicator', () => {