1 files changed, 14 insertions, 10 deletions
diff --git a/spec/frontend/repository/components/preview/index_spec.js b/spec/frontend/repository/components/preview/index_spec.js
index e4eba65795e..d4c746b67d6 100644
--- a/spec/frontend/repository/components/preview/index_spec.js
+++ b/spec/frontend/repository/components/preview/index_spec.js
@@ -9,9 +9,14 @@ jest.mock('~/lib/utils/common_utils');
 let vm;
 let $apollo;
 
-function factory(blob) {
+function factory(blob, loading) {
   $apollo = {
-    query: jest.fn().mockReturnValue(Promise.resolve({})),
+    queries: {
+      readme: {
+        query: jest.fn().mockReturnValue(Promise.resolve({})),
+        loading,
+      },
+    },
   };
 
   vm = shallowMount(Preview, {
@@ -58,14 +63,13 @@ describe('Repository file preview component', () => {
   });
 
   it('renders loading icon', async () => {
-    factory({
-      webPath: 'http://test.com',
-      name: 'README.md',
-    });
-
-    // setData usage is discouraged. See https://gitlab.com/groups/gitlab-org/-/epics/7330 for details
-    // eslint-disable-next-line no-restricted-syntax
-    vm.setData({ loading: 1 });
+    factory(
+      {
+        webPath: 'http://test.com',
+        name: 'README.md',
+      },
+      true,
+    );
 
     await nextTick();
     expect(vm.findComponent(GlLoadingIcon).exists()).toBe(true);