Merge branch '355132-graphql-polling' into 'master'

Add polling to security training service graphql query

See merge request gitlab-org/gitlab!82653

ee/app/assets/javascripts/vulnerabilities/components/vulnerability_training.vue

@@ -52,13 +52,14 @@ export default {
     },
     vulnerability: {
       query: securityTrainingVulnerabilityQuery,
+      pollInterval: 5000,
       update({ vulnerability }) {
         const allUrlsAreReady = vulnerability?.securityTrainingUrls?.every(
           ({ status }) => status === SECURITY_TRAINING_URL_STATUS_COMPLETED,
         );
 
         if (allUrlsAreReady) {
-          // note: once we add polling, we can call `.stopPolling` here
+          this.$apollo.queries.vulnerability.stopPolling();
           this.isUrlsLoading = false;
         }
 

ee/spec/frontend/vulnerabilities/vulnerability_training_spec.js

@@ -195,6 +195,28 @@ describe('VulnerabilityTraining component', () => {
       });
     });
 
+    describe('polling', () => {
+      beforeEach(() => {
+        createApolloProvider();
+        createComponent();
+      });
+
+      it('sets polling at 5000 ms', () => {
+        expect(wrapper.vm.$apollo.queries.vulnerability.options.pollInterval).toBe(5000);
+      });
+
+      it('stops polling when every training url status is completed', async () => {
+        jest
+          .spyOn(wrapper.vm.$apollo.queries.vulnerability, 'stopPolling')
+          .mockImplementation(jest.fn());
+
+        await waitForQueryToBeLoaded();
+        await nextTick();
+
+        expect(wrapper.vm.$apollo.queries.vulnerability.stopPolling).toHaveBeenCalled();
+      });
+    });
+
     describe('training item', () => {
       it('displays correct number of training items', async () => {
         createApolloProvider();