1 files changed, 143 insertions, 38 deletions

diff --git a/spec/frontend/security_configuration/components/training_provider_list_spec.js b/spec/frontend/security_configuration/components/training_provider_list_spec.js
index 60cc36a634c..578248e696f 100644
--- a/spec/frontend/security_configuration/components/training_provider_list_spec.js
+++ b/spec/frontend/security_configuration/components/training_provider_list_spec.js
@@ -1,87 +1,192 @@
-import { GlLink, GlToggle, GlCard, GlSkeletonLoader } from '@gitlab/ui';
+import { GlAlert, GlLink, GlToggle, GlCard, GlSkeletonLoader } from '@gitlab/ui';
 import { shallowMount } from '@vue/test-utils';
 import Vue from 'vue';
 import VueApollo from 'vue-apollo';
 import createMockApollo from 'helpers/mock_apollo_helper';
 import TrainingProviderList from '~/security_configuration/components/training_provider_list.vue';
+import configureSecurityTrainingProvidersMutation from '~/security_configuration/graphql/configure_security_training_providers.mutation.graphql';
 import waitForPromises from 'helpers/wait_for_promises';
-import { securityTrainingProviders, mockResolvers } from '../mock_data';
+import {
+  securityTrainingProviders,
+  createMockResolvers,
+  testProjectPath,
+  textProviderIds,
+} from '../mock_data';
 
 Vue.use(VueApollo);
 
 describe('TrainingProviderList component', () => {
   let wrapper;
-  let mockApollo;
-  let mockSecurityTrainingProvidersData;
+  let apolloProvider;
 
-  const createComponent = () => {
-    mockApollo = createMockApollo([], mockResolvers);
+  const createApolloProvider = ({ resolvers } = {}) => {
+    apolloProvider = createMockApollo([], createMockResolvers({ resolvers }));
+  };
 
+  const createComponent = () => {
     wrapper = shallowMount(TrainingProviderList, {
-      apolloProvider: mockApollo,
+      provide: {
+        projectPath: testProjectPath,
+      },
+      apolloProvider,
     });
   };
 
   const waitForQueryToBeLoaded = () => waitForPromises();
+  const waitForMutationToBeLoaded = waitForQueryToBeLoaded;
 
   const findCards = () => wrapper.findAllComponents(GlCard);
   const findLinks = () => wrapper.findAllComponents(GlLink);
   const findToggles = () => wrapper.findAllComponents(GlToggle);
+  const findFirstToggle = () => findToggles().at(0);
   const findLoader = () => wrapper.findComponent(GlSkeletonLoader);
+  const findErrorAlert = () => wrapper.findComponent(GlAlert);
 
-  beforeEach(() => {
-    mockSecurityTrainingProvidersData = jest.fn();
-    mockSecurityTrainingProvidersData.mockResolvedValue(securityTrainingProviders);
-
-    createComponent();
-  });
+  const toggleFirstProvider = () => findFirstToggle().vm.$emit('change');
 
   afterEach(() => {
     wrapper.destroy();
-    mockApollo = null;
+    apolloProvider = null;
   });
 
-  describe('when loading', () => {
-    it('shows the loader', () => {
-      expect(findLoader().exists()).toBe(true);
+  describe('with a successful response', () => {
+    beforeEach(() => {
+      createApolloProvider();
+      createComponent();
     });
 
-    it('does not show the cards', () => {
-      expect(findCards().exists()).toBe(false);
+    describe('when loading', () => {
+      it('shows the loader', () => {
+        expect(findLoader().exists()).toBe(true);
+      });
+
+      it('does not show the cards', () => {
+        expect(findCards().exists()).toBe(false);
+      });
     });
-  });
 
-  describe('basic structure', () => {
-    beforeEach(async () => {
-      await waitForQueryToBeLoaded();
+    describe('basic structure', () => {
+      beforeEach(async () => {
+        await waitForQueryToBeLoaded();
+      });
+
+      it('renders correct amount of cards', () => {
+        expect(findCards()).toHaveLength(securityTrainingProviders.length);
+      });
+
+      securityTrainingProviders.forEach(({ name, description, url, isEnabled }, index) => {
+        it(`shows the name for card ${index}`, () => {
+          expect(findCards().at(index).text()).toContain(name);
+        });
+
+        it(`shows the description for card ${index}`, () => {
+          expect(findCards().at(index).text()).toContain(description);
+        });
+
+        it(`shows the learn more link for card ${index}`, () => {
+          expect(findLinks().at(index).attributes()).toEqual({
+            target: '_blank',
+            href: url,
+          });
+        });
+
+        it(`shows the toggle with the correct value for card ${index}`, () => {
+          expect(findToggles().at(index).props('value')).toEqual(isEnabled);
+        });
+
+        it('does not show loader when query is populated', () => {
+          expect(findLoader().exists()).toBe(false);
+        });
+      });
     });
 
-    it('renders correct amount of cards', () => {
-      expect(findCards()).toHaveLength(securityTrainingProviders.length);
+    describe('storing training provider settings', () => {
+      beforeEach(async () => {
+        jest.spyOn(apolloProvider.defaultClient, 'mutate');
+
+        await waitForMutationToBeLoaded();
+
+        toggleFirstProvider();
+      });
+
+      it.each`
+        loading  | wait     | desc
+        ${true}  | ${false} | ${'enables loading of GlToggle when mutation is called'}
+        ${false} | ${true}  | ${'disables loading of GlToggle when mutation is complete'}
+      `('$desc', async ({ loading, wait }) => {
+        if (wait) {
+          await waitForMutationToBeLoaded();
+        }
+        expect(findFirstToggle().props('isLoading')).toBe(loading);
+      });
+
+      it('calls mutation when toggle is changed', () => {
+        expect(apolloProvider.defaultClient.mutate).toHaveBeenCalledWith(
+          expect.objectContaining({
+            mutation: configureSecurityTrainingProvidersMutation,
+            variables: { input: { enabledProviders: textProviderIds, fullPath: testProjectPath } },
+          }),
+        );
+      });
     });
+  });
+
+  describe('with errors', () => {
+    const expectErrorAlertToExist = () => {
+      expect(findErrorAlert().props()).toMatchObject({
+        dismissible: false,
+        variant: 'danger',
+      });
+    };
+
+    describe('when fetching training providers', () => {
+      beforeEach(async () => {
+        createApolloProvider({
+          resolvers: {
+            Query: {
+              securityTrainingProviders: jest.fn().mockReturnValue(new Error()),
+            },
+          },
+        });
+        createComponent();
 
-    securityTrainingProviders.forEach(({ name, description, url, isEnabled }, index) => {
-      it(`shows the name for card ${index}`, () => {
-        expect(findCards().at(index).text()).toContain(name);
+        await waitForQueryToBeLoaded();
       });
 
-      it(`shows the description for card ${index}`, () => {
-        expect(findCards().at(index).text()).toContain(description);
+      it('shows an non-dismissible error alert', () => {
+        expectErrorAlertToExist();
       });
 
-      it(`shows the learn more link for card ${index}`, () => {
-        expect(findLinks().at(index).attributes()).toEqual({
-          target: '_blank',
-          href: url,
+      it('shows an error description', () => {
+        expect(findErrorAlert().text()).toBe(TrainingProviderList.i18n.providerQueryErrorMessage);
+      });
+    });
+
+    describe('when storing training provider configurations', () => {
+      beforeEach(async () => {
+        createApolloProvider({
+          resolvers: {
+            Mutation: {
+              configureSecurityTrainingProviders: () => ({
+                errors: ['something went wrong!'],
+                securityTrainingProviders: [],
+              }),
+            },
+          },
         });
+        createComponent();
+
+        await waitForQueryToBeLoaded();
+        toggleFirstProvider();
+        await waitForMutationToBeLoaded();
       });
 
-      it(`shows the toggle with the correct value for card ${index}`, () => {
-        expect(findToggles().at(index).props('value')).toEqual(isEnabled);
+      it('shows an non-dismissible error alert', () => {
+        expectErrorAlertToExist();
       });
 
-      it('does not show loader when query is populated', () => {
-        expect(findLoader().exists()).toBe(false);
+      it('shows an error description', () => {
+        expect(findErrorAlert().text()).toBe(TrainingProviderList.i18n.configMutationErrorMessage);
       });
     });
   });