diff options
Diffstat (limited to 'spec/frontend/experimentation/utils_spec.js')
-rw-r--r-- | spec/frontend/experimentation/utils_spec.js | 101 |
1 files changed, 80 insertions, 21 deletions
diff --git a/spec/frontend/experimentation/utils_spec.js b/spec/frontend/experimentation/utils_spec.js index 87dd2d595ba..ec09bbab349 100644 --- a/spec/frontend/experimentation/utils_spec.js +++ b/spec/frontend/experimentation/utils_spec.js @@ -1,38 +1,97 @@ +import { assignGitlabExperiment } from 'helpers/experimentation_helper'; +import { DEFAULT_VARIANT, CANDIDATE_VARIANT } from '~/experimentation/constants'; import * as experimentUtils from '~/experimentation/utils'; -const TEST_KEY = 'abc'; - describe('experiment Utilities', () => { - const oldGon = window.gon; - - afterEach(() => { - window.gon = oldGon; - }); + const TEST_KEY = 'abc'; describe('getExperimentData', () => { - it.each` - gon | input | output - ${{ experiment: { [TEST_KEY]: '_data_' } }} | ${[TEST_KEY]} | ${'_data_'} - ${{}} | ${[TEST_KEY]} | ${undefined} - `('with input=$input and gon=$gon, returns $output', ({ gon, input, output }) => { - window.gon = gon; + describe.each` + gon | input | output + ${[TEST_KEY, '_data_']} | ${[TEST_KEY]} | ${{ variant: '_data_' }} + ${[]} | ${[TEST_KEY]} | ${undefined} + `('with input=$input and gon=$gon', ({ gon, input, output }) => { + assignGitlabExperiment(...gon); - expect(experimentUtils.getExperimentData(...input)).toEqual(output); + it(`returns ${output}`, () => { + expect(experimentUtils.getExperimentData(...input)).toEqual(output); + }); }); }); describe('isExperimentVariant', () => { + describe.each` + gon | input | output + ${[TEST_KEY, DEFAULT_VARIANT]} | ${[TEST_KEY, DEFAULT_VARIANT]} | ${true} + ${[TEST_KEY, '_variant_name']} | ${[TEST_KEY, '_variant_name']} | ${true} + ${[TEST_KEY, '_variant_name']} | ${[TEST_KEY, '_bogus_name']} | ${false} + ${[TEST_KEY, '_variant_name']} | ${['boguskey', '_variant_name']} | ${false} + ${[]} | ${[TEST_KEY, '_variant_name']} | ${false} + `('with input=$input and gon=$gon', ({ gon, input, output }) => { + assignGitlabExperiment(...gon); + + it(`returns ${output}`, () => { + expect(experimentUtils.isExperimentVariant(...input)).toEqual(output); + }); + }); + }); + + describe('experiment', () => { + const controlSpy = jest.fn(); + const candidateSpy = jest.fn(); + const getUpStandUpSpy = jest.fn(); + + const variants = { + use: controlSpy, + try: candidateSpy, + get_up_stand_up: getUpStandUpSpy, + }; + + describe('when there is no experiment data', () => { + it('calls control variant', () => { + experimentUtils.experiment('marley', variants); + expect(controlSpy).toHaveBeenCalled(); + }); + }); + + describe('when experiment variant is "control"', () => { + assignGitlabExperiment('marley', DEFAULT_VARIANT); + + it('calls the control variant', () => { + experimentUtils.experiment('marley', variants); + expect(controlSpy).toHaveBeenCalled(); + }); + }); + + describe('when experiment variant is "candidate"', () => { + assignGitlabExperiment('marley', CANDIDATE_VARIANT); + + it('calls the candidate variant', () => { + experimentUtils.experiment('marley', variants); + expect(candidateSpy).toHaveBeenCalled(); + }); + }); + + describe('when experiment variant is "get_up_stand_up"', () => { + assignGitlabExperiment('marley', 'get_up_stand_up'); + + it('calls the get-up-stand-up variant', () => { + experimentUtils.experiment('marley', variants); + expect(getUpStandUpSpy).toHaveBeenCalled(); + }); + }); + }); + + describe('getExperimentVariant', () => { it.each` - gon | input | output - ${{ experiment: { [TEST_KEY]: { variant: 'control' } } }} | ${[TEST_KEY, 'control']} | ${true} - ${{ experiment: { [TEST_KEY]: { variant: '_variant_name' } } }} | ${[TEST_KEY, '_variant_name']} | ${true} - ${{ experiment: { [TEST_KEY]: { variant: '_variant_name' } } }} | ${[TEST_KEY, '_bogus_name']} | ${false} - ${{ experiment: { [TEST_KEY]: { variant: '_variant_name' } } }} | ${['boguskey', '_variant_name']} | ${false} - ${{}} | ${[TEST_KEY, '_variant_name']} | ${false} + gon | input | output + ${{ experiment: { [TEST_KEY]: { variant: DEFAULT_VARIANT } } }} | ${[TEST_KEY]} | ${DEFAULT_VARIANT} + ${{ experiment: { [TEST_KEY]: { variant: CANDIDATE_VARIANT } } }} | ${[TEST_KEY]} | ${CANDIDATE_VARIANT} + ${{}} | ${[TEST_KEY]} | ${DEFAULT_VARIANT} `('with input=$input and gon=$gon, returns $output', ({ gon, input, output }) => { window.gon = gon; - expect(experimentUtils.isExperimentVariant(...input)).toEqual(output); + expect(experimentUtils.getExperimentVariant(...input)).toEqual(output); }); }); }); |