1 files changed, 17 insertions, 8 deletions
diff --git a/app/assets/javascripts/experimentation/utils.js b/app/assets/javascripts/experimentation/utils.js
index e572280a62c..9079c238169 100644
--- a/app/assets/javascripts/experimentation/utils.js
+++ b/app/assets/javascripts/experimentation/utils.js
@@ -1,18 +1,27 @@
 // This file only applies to use of experiments through https://gitlab.com/gitlab-org/gitlab-experiment
-import { get } from 'lodash';
+import { get, pick } from 'lodash';
 import { DEFAULT_VARIANT, CANDIDATE_VARIANT, TRACKING_CONTEXT_SCHEMA } from './constants';
 
+function getExperimentsData() {
+  return get(window, ['gon', 'experiment'], {});
+}
+
+function convertExperimentDataToExperimentContext(experimentData) {
+  return { schema: TRACKING_CONTEXT_SCHEMA, data: experimentData };
+}
+
 export function getExperimentData(experimentName) {
-  return get(window, ['gon', 'experiment', experimentName]);
+  return getExperimentsData()[experimentName];
 }
 
 export function getExperimentContexts(...experimentNames) {
-  return experimentNames
-    .map((name) => {
-      const data = getExperimentData(name);
-      return data && { schema: TRACKING_CONTEXT_SCHEMA, data };
-    })
-    .filter((context) => context);
+  return Object.values(pick(getExperimentsData(), experimentNames)).map(
+    convertExperimentDataToExperimentContext,
+  );
+}
+
+export function getAllExperimentContexts() {
+  return Object.values(getExperimentsData()).map(convertExperimentDataToExperimentContext);
 }
 
 export function isExperimentVariant(experimentName, variantName) {