diff options
Diffstat (limited to 'app/assets/javascripts/experimentation/utils.js')
-rw-r--r-- | app/assets/javascripts/experimentation/utils.js | 25 |
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) { |