1 files changed, 36 insertions, 13 deletions

diff --git a/spec/frontend/repository/components/blob_content_viewer_spec.js b/spec/frontend/repository/components/blob_content_viewer_spec.js
index 0f7cf4e61b2..6ece72c41bb 100644
--- a/spec/frontend/repository/components/blob_content_viewer_spec.js
+++ b/spec/frontend/repository/components/blob_content_viewer_spec.js
@@ -17,7 +17,8 @@ import { loadViewer } from '~/repository/components/blob_viewers';
 import DownloadViewer from '~/repository/components/blob_viewers/download_viewer.vue';
 import EmptyViewer from '~/repository/components/blob_viewers/empty_viewer.vue';
 import SourceViewer from '~/vue_shared/components/source_viewer/source_viewer.vue';
-import blobInfoQuery from '~/repository/queries/blob_info.query.graphql';
+import blobInfoQuery from 'shared_queries/repository/blob_info.query.graphql';
+import projectInfoQuery from '~/repository/queries/project_info.query.graphql';
 import userInfoQuery from '~/repository/queries/user_info.query.graphql';
 import applicationInfoQuery from '~/repository/queries/application_info.query.graphql';
 import CodeIntelligence from '~/code_navigation/components/app.vue';
@@ -45,8 +46,9 @@ jest.mock('~/lib/utils/common_utils');
 jest.mock('~/blob/line_highlighter');
 
 let wrapper;
-let mockResolver;
+let blobInfoMockResolver;
 let userInfoMockResolver;
+let projectInfoMockResolver;
 let applicationInfoMockResolver;
 
 const mockAxios = new MockAdapter(axios);
@@ -74,22 +76,40 @@ const createComponent = async (mockData = {}, mountFn = shallowMount, mockRoute
     highlightJs = true,
   } = mockData;
 
-  const project = {
+  const blobInfo = {
     ...projectMock,
+    repository: {
+      empty,
+      blobs: { nodes: [blob] },
+    },
+  };
+
+  const projectInfo = {
+    __typename: 'Project',
+    id: '123',
     userPermissions: {
       pushCode,
       forkProject,
       downloadCode,
       createMergeRequestIn,
     },
-    repository: {
-      empty,
-      blobs: { nodes: [blob] },
+    pathLocks: {
+      nodes: [
+        {
+          id: 'test',
+          path: 'locked_file.js',
+          user: { id: '123', username: 'root' },
+        },
+      ],
     },
   };
 
-  mockResolver = jest.fn().mockResolvedValue({
-    data: { isBinary, project },
+  projectInfoMockResolver = jest.fn().mockResolvedValue({
+    data: { project: projectInfo },
+  });
+
+  blobInfoMockResolver = jest.fn().mockResolvedValue({
+    data: { isBinary, project: blobInfo },
   });
 
   userInfoMockResolver = jest.fn().mockResolvedValue({
@@ -101,8 +121,9 @@ const createComponent = async (mockData = {}, mountFn = shallowMount, mockRoute
   });
 
   const fakeApollo = createMockApollo([
-    [blobInfoQuery, mockResolver],
+    [blobInfoQuery, blobInfoMockResolver],
     [userInfoQuery, userInfoMockResolver],
+    [projectInfoQuery, projectInfoMockResolver],
     [applicationInfoQuery, applicationInfoMockResolver],
   ]);
 
@@ -129,7 +150,7 @@ const createComponent = async (mockData = {}, mountFn = shallowMount, mockRoute
 
   // setData usage is discouraged. See https://gitlab.com/groups/gitlab-org/-/epics/7330 for details
   // eslint-disable-next-line no-restricted-syntax
-  wrapper.setData({ project, isBinary });
+  wrapper.setData({ project: blobInfo, isBinary });
 
   await waitForPromises();
 };
@@ -504,14 +525,16 @@ describe('Blob content viewer component', () => {
       async ({ highlightJs, shouldFetchRawText }) => {
         await createComponent({ highlightJs });
 
-        expect(mockResolver).toHaveBeenCalledWith(expect.objectContaining({ shouldFetchRawText }));
+        expect(blobInfoMockResolver).toHaveBeenCalledWith(
+          expect.objectContaining({ shouldFetchRawText }),
+        );
       },
     );
 
     it('is called with originalBranch value if the prop has a value', async () => {
       await createComponent({ inject: { originalBranch: 'some-branch' } });
 
-      expect(mockResolver).toHaveBeenCalledWith(
+      expect(blobInfoMockResolver).toHaveBeenCalledWith(
         expect.objectContaining({
           ref: 'some-branch',
         }),
@@ -521,7 +544,7 @@ describe('Blob content viewer component', () => {
     it('is called with ref value if the originalBranch prop has no value', async () => {
       await createComponent();
 
-      expect(mockResolver).toHaveBeenCalledWith(
+      expect(blobInfoMockResolver).toHaveBeenCalledWith(
         expect.objectContaining({
           ref: 'default-ref',
         }),