diff --git a/jest.config.js b/jest.config.js index 4cdc16adc16475025e55e180cfa638b8be2c4b1a..c7518be9e96ebac69e325b7e65280fbdafb00b08 100644 --- a/jest.config.js +++ b/jest.config.js @@ -1,12 +1,4 @@ - -const fs = require('fs'); -const path = require('path'); - -const ROOT_PATH = __dirname; -const IS_EE = - process.env.EE !== undefined - ? JSON.parse(process.env.EE) - : fs.existsSync(path.join(ROOT_PATH, 'ee')); +const IS_EE = require('./config/helpers/is_ee_env'); const reporters = ['default']; @@ -46,4 +38,7 @@ module.exports = { transformIgnorePatterns: ['node_modules/(?!(@gitlab/ui)/)'], timers: 'fake', testEnvironment: '<rootDir>/spec/frontend/environment.js', + testEnvironmentOptions: { + IS_EE, + }, }; diff --git a/spec/frontend/environment.js b/spec/frontend/environment.js index cb128c7d88098edba5e88ed81674ea12623f6678..1067a53906ace205b1f428718a9a9de8c35e0c34 100644 --- a/spec/frontend/environment.js +++ b/spec/frontend/environment.js @@ -6,6 +6,7 @@ const JSDOMEnvironment = require('jest-environment-jsdom'); class CustomEnvironment extends JSDOMEnvironment { constructor(config, context) { super(config, context); + Object.assign(context.console, { error(...args) { throw new ErrorWithStack( @@ -21,6 +22,11 @@ class CustomEnvironment extends JSDOMEnvironment { ); }, }); + + const { testEnvironmentOptions } = config; + this.global.gon = { + ee: testEnvironmentOptions.IS_EE, + }; } }