spec/frontend/__helpers__/experimentation_helper.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

import { merge } from 'lodash';

export function withGonExperiment(experimentKey, value = true) {
  let origGon;

  beforeEach(() => {
    origGon = window.gon;
    window.gon = merge({}, window.gon || {}, { experiments: { [experimentKey]: value } });
  });

  afterEach(() => {
    window.gon = origGon;
  });
}
// This helper is for specs that use `gitlab-experiment` utilities, which have a different schema that gets pushed via Gon compared to `Experimentation Module`
export function assignGitlabExperiment(experimentKey, variant) {
  let origGon;

  beforeEach(() => {
    origGon = window.gon;
    window.gon = { experiment: { [experimentKey]: { variant } } };
  });

  afterEach(() => {
    window.gon = origGon;
  });
}