1 files changed, 48 insertions, 9 deletions
diff --git a/spec/frontend/security_configuration/mock_data.js b/spec/frontend/security_configuration/mock_data.js
index 37ecce3886d..b042e870467 100644
--- a/spec/frontend/security_configuration/mock_data.js
+++ b/spec/frontend/security_configuration/mock_data.js
@@ -9,6 +9,7 @@ export const securityTrainingProviders = [
     description: 'Interactive developer security education',
     url: 'https://www.example.org/security/training',
     isEnabled: false,
+    isPrimary: false,
   },
   {
     id: textProviderIds[1],
@@ -16,24 +17,62 @@ export const securityTrainingProviders = [
     description: 'Security training with guide and learning pathways.',
     url: 'https://www.vendornametwo.com/',
     isEnabled: true,
+    isPrimary: false,
   },
 ];
 
 export const securityTrainingProvidersResponse = {
   data: {
-    securityTrainingProviders,
+    project: {
+      id: 1,
+      securityTrainingProviders,
+    },
+  },
+};
+
+export const dismissUserCalloutResponse = {
+  data: {
+    userCalloutCreate: {
+      errors: [],
+      userCallout: {
+        dismissedAt: '2022-02-02T04:36:57Z',
+        featureName: 'SECURITY_TRAINING_FEATURE_PROMOTION',
+      },
+    },
+  },
+};
+
+export const dismissUserCalloutErrorResponse = {
+  data: {
+    userCalloutCreate: {
+      errors: ['Something went wrong'],
+      userCallout: {
+        dismissedAt: '',
+        featureName: 'SECURITY_TRAINING_FEATURE_PROMOTION',
+      },
+    },
   },
 };
 
-const defaultMockResolvers = {
-  Query: {
-    securityTrainingProviders() {
-      return securityTrainingProviders;
+export const updateSecurityTrainingProvidersResponse = {
+  data: {
+    securityTrainingUpdate: {
+      errors: [],
+      training: {
+        id: 101,
+        name: 'Acme',
+        isEnabled: true,
+        isPrimary: false,
+      },
     },
   },
 };
 
-export const createMockResolvers = ({ resolvers: customMockResolvers = {} } = {}) => ({
-  ...defaultMockResolvers,
-  ...customMockResolvers,
-});
+export const updateSecurityTrainingProvidersErrorResponse = {
+  data: {
+    securityTrainingUpdate: {
+      errors: ['something went wrong!'],
+      training: null,
+    },
+  },
+};