Set gon.ee in Jestwinh-jest-gon

2 files changed, 10 insertions, 9 deletions

diff --git a/jest.config.js b/jest.config.js
index 4cdc16adc16..c7518be9e96 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 cb128c7d880..1067a53906a 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,
+    };
   }
 }